Update app.py

app.py

@@ -23,6 +23,7 @@ app = FastAPI(
 )
 
 # Mount static files
+os.makedirs("static", exist_ok=True)
 app.mount("/static", StaticFiles(directory="static"), name="static")
 
 # Load AI models
@@ -31,7 +32,8 @@ try:
     image_pipeline = pipeline(
         "image-to-text", 
         model="Salesforce/blip-image-captioning-base",
-        device="cpu"
+        device="cpu",
+        use_fast=False  # Explicitly set to avoid warning
     )
     text_pipeline = pipeline(
         "text2text-generation", 
@@ -68,6 +70,8 @@ async def home():
     try:
         with open("static/index.html") as f:
             return f.read()
+    except FileNotFoundError:
+        return "<h1>Welcome to AI Web Services</h1><p>Frontend not found</p>"
     except Exception as e:
         logger.error(f"Failed to load frontend: {e}")
         raise HTTPException(500, "Frontend loading failed")
@@ -77,11 +81,7 @@ async def summarize(
     file: Optional[UploadFile] = File(None),
     text: Optional[str] = Form(None)
 ):
-    """
-    Summarize text or document
-    Accepts: PDF, DOCX or raw text
-    Returns: {'summary': str}
-    """
+    """Summarize text or document"""
     try:
         if file:
             text = await extract_text(file)
@@ -89,7 +89,7 @@ async def summarize(
             raise HTTPException(400, "No text provided")
         
         result = text_pipeline(f"summarize: {text}", max_length=150)
-        return JSONResponse({"summary": result[0]['generated_text']})
+        return {"summary": result[0]['generated_text']}
     except HTTPException:
         raise
     except Exception as e:
@@ -98,15 +98,11 @@ async def summarize(
 
 @app.post("/api/caption")
 async def caption_image(file: UploadFile = File(...)):
-    """
-    Generate caption for image
-    Accepts: JPEG, PNG
-    Returns: {'caption': str}
-    """
+    """Generate caption for image"""
     try:
         image = Image.open(io.BytesIO(await file.read()))
         result = image_pipeline(image)
-        return JSONResponse({"caption": result[0]['generated_text']})
+        return {"caption": result[0]['generated_text']}
     except Exception as e:
         logger.error(f"Captioning error: {e}")
         raise HTTPException(500, "Image captioning failed")
@@ -117,11 +113,7 @@ async def answer_question(
     text: Optional[str] = Form(None),
     question: str = Form(...)
 ):
-    """
-    Answer questions about text/document
-    Accepts: PDF, DOCX or raw text + question
-    Returns: {'answer': str}
-    """
+    """Answer questions about text/document"""
     try:
         if file:
             text = await extract_text(file)
@@ -129,7 +121,7 @@ async def answer_question(
             raise HTTPException(400, "No text provided")
         
         result = text_pipeline(f"question: {question} context: {text}")
-        return JSONResponse({"answer": result[0]['generated_text']})
+        return {"answer": result[0]['generated_text']}
     except HTTPException:
         raise
     except Exception as e:
@@ -141,11 +133,7 @@ async def generate_visualization(
     file: UploadFile = File(...),
     chart_type: str = Form("bar")
 ):
-    """
-    Generate visualization code for Excel data
-    Accepts: XLSX, CSV
-    Returns: {'code': str, 'columns': list}
-    """
+    """Generate visualization code for Excel data"""
     try:
         df = pd.read_excel(io.BytesIO(await file.read()))
         
@@ -160,10 +148,10 @@ sns.pairplot(df)
 plt.title('Data Distribution')
 plt.show()"""
             
-        return JSONResponse({
+        return {
             "code": code,
             "columns": list(df.columns)
-        })
+        }
     except Exception as e:
         logger.error(f"Visualization error: {e}")
         raise HTTPException(500, "Visualization code generation failed")
@@ -171,21 +159,21 @@ plt.show()"""
 @app.get("/health")
 async def health_check():
     """Health check endpoint"""
-    return JSONResponse({
+    return {
         "status": "healthy",
         "models": {
             "image_captioning": "loaded",
             "text_generation": "loaded"
         }
-    })
+    }
 
 # Server initialization
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(
-        app,
+        "app:app",  # Changed to string format for proper reload
         host="0.0.0.0",
         port=8000,
         log_level="info",
-        reload=True
+        reload=False  # Disabled reload for direct execution
     )