Update app.py

app.py

@@ -4,6 +4,11 @@ import shutil
 from pathlib import Path
 import pandas as pd
 from openpyxl import load_workbook
+from fastapi import FastAPI, UploadFile, File, HTTPException
+from fastapi.responses import FileResponse
+from typing import List
+import uvicorn
+import os
 
 """
 Real Estate Financial Model Pipeline
@@ -1600,6 +1605,71 @@ class RealEstateModelPipeline:
         
         return output_excel
 
+# ============= FastAPI Setup =============
+app = FastAPI(title="Real Estate Financial Model API")
+
+# Hardcoded API Key
+GEMINI_API_KEY = "AIzaSyCy6GoBR724Hj9VyuW3hKM4N0P6liBOlDo"
+
+@app.post("/api/generate-model")
+async def generate_model(files: List[UploadFile] = File(...)):
+    """
+    API endpoint to process uploaded files and return Excel model
+    
+    Parameters:
+    - files: List of PDF/XLSX files to process
+    
+    Returns:
+    - Excel file with financial model
+    """
+    if not files:
+        raise HTTPException(status_code=400, detail="No files uploaded")
+    
+    try:
+        # Create temporary directory
+        temp_dir = tempfile.mkdtemp()
+        
+        # Save uploaded files
+        saved_files = []
+        for upload_file in files:
+            file_path = Path(temp_dir) / upload_file.filename
+            
+            # Write file content
+            with open(file_path, "wb") as f:
+                content = await upload_file.read()
+                f.write(content)
+            
+            saved_files.append(str(file_path))
+        
+        # Initialize pipeline
+        pipeline = RealEstateModelPipeline(GEMINI_API_KEY)
+        
+        # Create output file
+        output_file = Path(temp_dir) / "Real_Estate_Financial_Model.xlsx"
+        
+        # Run pipeline
+        pipeline.run_full_pipeline(temp_dir, str(output_file))
+        
+        # Return Excel file
+        return FileResponse(
+            path=str(output_file),
+            filename="Real_Estate_Financial_Model.xlsx",
+            media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+        )
+        
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Processing error: {str(e)}")
+    
+    finally:
+        # Cleanup will happen when temp directory is garbage collected
+        pass
+
+
+@app.get("/api/health")
+async def health_check():
+    """Health check endpoint"""
+    return {"status": "healthy", "service": "Real Estate Financial Model API"}
+    
 
 if __name__ == "__main__":
     
@@ -1715,4 +1785,11 @@ if __name__ == "__main__":
         """)
     
     # Launch the app
-    demo.launch(share=False)
+    # demo.launch(share=False)
+    app = gr.mount_gradio_app(app, demo, path="/")
+    uvicorn.run(
+        app, 
+        host="0.0.0.0", 
+        port=7860,
+        log_level="info"
+    )