Update app.py

app.py

@@ -1,13 +1,12 @@
 from fastapi import FastAPI, UploadFile, File, Form, HTTPException, Request
 from fastapi.staticfiles import StaticFiles
-from fastapi.responses import RedirectResponse, JSONResponse, HTMLResponse
+from fastapi.responses import RedirectResponse, JSONResponse
 from transformers import pipeline, ViltProcessor, ViltForQuestionAnswering, M2M100ForConditionalGeneration, M2M100Tokenizer
-from typing import Optional, Dict, Any, List
+from typing import Optional, Dict, Any
 import logging
 import time
 import os
 import io
-import json
 import re
 from PIL import Image
 from docx import Document
@@ -149,7 +148,10 @@ def load_model(task: str, model_name: str = None):
         )
         
     except Exception as e:
-        logger.error(f"Model load failed: {str(e)}")
+        logger.error(f"Model load failed: {task} - {str(e)}")
+        if task == "file-qa":
+            logger.warning("Falling back to Gemini for file-qa due to model load failure")
+            return None
         raise HTTPException(status_code=500, detail=f"Model loading failed: {task} - {str(e)}")
 
 def get_gemini_response(user_input: str, is_generation: bool = False):
@@ -216,7 +218,7 @@ def translate_text(text: str, target_language: str):
         return translated_text
         
     except Exception as e:
-        logger.error(f"Translation error: {str(e)}", exc_info=True)
+        logger.error(f"Translation error: {str(e)}")
         return f"Translation error: {str(e)}"
 
 def detect_intent(text: str = None, file: UploadFile = None) -> tuple[str, str]:
@@ -585,10 +587,9 @@ async def process_input(
             if not content.strip():
                 raise HTTPException(status_code=400, detail="No text could be extracted from the file")
             
-            try:
-                qa_pipeline = load_model("file-qa")
-            except Exception as e:
-                logger.warning(f"File-QA model failed: {str(e)}. Falling back to Gemini.")
+            qa_pipeline = load_model("file-qa")
+            if qa_pipeline is None:
+                logger.info("Using Gemini fallback for file-qa")
                 question = text.strip()
                 if not question.endswith('?'):
                     question += '?'
@@ -648,7 +649,7 @@ async def process_input(
             return {"response": response, "type": "chat"}
             
     except Exception as e:
-        logger.error(f"Processing error: {str(e)}", exc_info=True)
+        logger.error(f"Processing error: {str(e)}")
         raise HTTPException(status_code=500, detail=str(e))
     finally:
         process_time = time.time() - start_time
@@ -695,7 +696,7 @@ async def extract_text_from_file(file: UploadFile) -> str:
             raise HTTPException(status_code=400, detail=f"Unsupported file format: {filename}")
 
     except Exception as e:
-        logger.error(f"File extraction error: {str(e)}", exc_info=True)
+        logger.error(f"File extraction error: {str(e)}")
         raise HTTPException(
             status_code=500,
             detail=f"Error extracting text: {str(e)}. Supported formats: PDF, DOCX, TXT, RTF"
@@ -807,17 +808,17 @@ plt.tight_layout()
 plt.savefig('dashboard.png')
 plt.show()"""
 
-@app.get("/", include_in_schema=False)
+@app.get("/")
 async def home():
     """Redirect to the static index.html file"""
     return RedirectResponse(url="/static/index.html")
 
-@app.get("/health", include_in_schema=True)
+@app.get("/health")
 async def health_check():
     """Health check endpoint"""
     return {"status": "healthy", "version": "2.0.0"}
 
-@app.get("/models", include_in_schema=True)
+@app.get("/models")
 async def list_models():
     """List available models"""
     return {"models": MODELS}