Update app.py

app.py

@@ -92,31 +92,61 @@ def extract_pptx(path):
     return text
 
 
+# def mode_load(path):
+#     choice = ""
+#     file_type = path.split(".")[-1]
+#     print(file_type)
+#     if file_type in ["pdf", "txt", "py", "docx", "pptx", "json", "cpp", "md"]:
+#         if file_type.endswith("pdf"):
+#             content = extract_pdf(path)
+#         elif file_type.endswith("docx"):
+#             content = extract_docx(path)
+#         elif file_type.endswith("pptx"):
+#             content = extract_pptx(path)
+#         else:
+#             content = extract_text(path)
+#         choice = "doc"
+#         print(content[:100])
+#         return choice, content[:5000]
+
+
+#     elif file_type in ["png", "jpg", "jpeg", "bmp", "tiff", "webp"]:
+#         content = Image.open(path).convert('RGB')
+#         choice = "image"
+#         return choice, content
+
+#     else:
+#         raise gr.Error("Oops, unsupported files.")
+
+
 def mode_load(path):
     choice = ""
-    file_type = path.split(".")[-1]
+    file_type = path.split(".")[-1].lower()
     print(file_type)
+    
     if file_type in ["pdf", "txt", "py", "docx", "pptx", "json", "cpp", "md"]:
-        if file_type.endswith("pdf"):
+        if file_type == "pdf":
             content = extract_pdf(path)
-        elif file_type.endswith("docx"):
+        elif file_type == "docx":
             content = extract_docx(path)
-        elif file_type.endswith("pptx"):
+        elif file_type == "pptx":
             content = extract_pptx(path)
         else:
             content = extract_text(path)
         choice = "doc"
         print(content[:100])
         return choice, content[:5000]
-
-
+    
     elif file_type in ["png", "jpg", "jpeg", "bmp", "tiff", "webp"]:
-        content = Image.open(path).convert('RGB')
-        choice = "image"
-        return choice, content
-
+        try:
+            content = Image.open(path).convert('RGB')
+            choice = "image"
+            return choice, content
+        except Exception as e:
+            raise ValueError(f"Error processing image file: {e}")
+    
     else:
-        raise gr.Error("Oops, unsupported files.")
+        raise ValueError("Oops, unsupported file type.")
 
 
 @spaces.GPU()