Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 from fastapi import FastAPI, UploadFile, File, HTTPException, Form
 from fastapi.staticfiles import StaticFiles
-from fastapi.responses import RedirectResponse
+from fastapi.responses import RedirectResponse, JSONResponse
 from transformers import pipeline
 import logging
 from PIL import Image
@@ -206,24 +206,39 @@ plt.show()
 
         return {"code": code, "filename": code_filename}
     except Exception as e:
-        logger.error(f"Error during visualization codeS generation: {e}")
+        logger.error(f"Error during visualization code generation: {e}")
         raise HTTPException(status_code=500, detail=str(e))
 
 async def extract_text_from_file(file: UploadFile):
     try:
         file_content = await file.read()
+        if not file_content:
+            raise ValueError("Uploaded file is empty.")
+
         if file.filename.endswith(".pdf"):
-            doc = fitz.open(stream=file_content, filetype="pdf")
-            text = ""
-            for page in doc:
-                text += page.get_text()
-            return text
+            try:
+                doc = fitz.open(stream=file_content, filetype="pdf")
+                text = ""
+                for page in doc:
+                    text += page.get_text()
+                return text
+            except Exception as e:
+                logger.error(f"Error reading PDF file: {e}")
+                raise ValueError("Failed to read PDF file. It might be corrupted or not a valid PDF.")
         elif file.filename.endswith(".docx"):
-            doc = Document(io.BytesIO(file_content))
-            text = "\n".join([para.text for para in doc.paragraphs])
-            return text
+            try:
+                doc = Document(io.BytesIO(file_content))
+                text = "\n".join([para.text for para in doc.paragraphs])
+                return text
+            except Exception as e:
+                logger.error(f"Error reading DOCX file: {e}")
+                raise ValueError("Failed to read DOCX file. It might be corrupted or not a valid DOCX.")
         elif file.filename.endswith(".txt"):
-            return file_content.decode("utf-8")
+            try:
+                return file_content.decode("utf-8")
+            except Exception as e:
+                logger.error(f"Error reading TXT file: {e}")
+                raise ValueError("Failed to read TXT file. It might be corrupted or not a valid TXT.")
         else:
             raise ValueError("Unsupported file format. Please upload a PDF, DOCX, or TXT file.")
     except Exception as e: