Update main.py

main.py

@@ -723,6 +723,16 @@ def generate_dynamic_visualization_code(df: pd.DataFrame, request: Visualization
 
 
 
+SUPPORTED_EXTENSIONS: Dict[str, str] = {
+    '.txt': 'text/plain',
+    '.md': 'text/markdown',
+    '.docx': 'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
+    '.pptx': 'application/vnd.openxmlformats-officedocument.presentationml.presentation',
+    '.pdf': 'application/pdf',
+    '.rtf': 'application/rtf',
+    '.odt': 'application/vnd.oasis.opendocument.text'
+}
+
 SUPPORTED_EXTENSIONS: Dict[str, str] = {
     '.txt': 'text/plain',
     '.md': 'text/markdown',
@@ -791,9 +801,8 @@ async def extract_text_from_file(file: UploadFile, file_ext: str) -> str:
             return "\n".join(text)
         elif file_ext == '.pdf':
             content = await file.read()
-            return extract_text(content, 'pdf')  # Your existing PDF extraction
+            return extract_text(content, 'pdf')
         elif file_ext == '.odt':
-            # For ODT files, we'll use a simple text extraction
             content = await file.read()
             try:
                 return content.decode('utf-8')
@@ -829,8 +838,6 @@ async def generate_summary(text: str, chunk_size: int = 1000) -> str:
     return " ".join(summaries)
 
 
-
-
 @app.post("/qa")
 @limiter.limit("5/minute")
 async def question_answering(