Update app.py

app.py

@@ -1987,125 +1987,159 @@ async def analyze_only(files: List[UploadFile] = File(...)):
         raise HTTPException(status_code=500, detail=str(e))
 
 
-if __name__ == "__main__":
-    
-    # Hardcoded API Key
-    GEMINI_API_KEY = "AIzaSyCy6GoBR724Hj9VyuW3hKM4N0P6liBOlDo"
+def process_pdfs(pdf_files):
+    """Process uploaded PDFs and return Excel file"""
+    if not pdf_files:
+        return None, "⚠️ Please upload at least one PDF/XLSX file"
     
-    def process_pdfs(pdf_files):
-        """Process uploaded PDFs and return Excel file"""
-        if not pdf_files:
-            return None, "Please upload at least one PDF file"
+    try:
+        # Create temporary directory for PDFs
+        temp_dir = tempfile.mkdtemp()
         
-        try:
-            # Create temporary directory for PDFs
-            temp_dir = tempfile.mkdtemp()
-            
-            # Save uploaded PDFs to temp directory
-            for pdf_file in pdf_files:
-                shutil.copy(pdf_file.name, temp_dir)
-            
-            # Initialize pipeline with hardcoded API key
-            pipeline = RealEstateModelPipeline(GEMINI_API_KEY)
-            
-            # Create output file in temp directory
-            output_file = Path(temp_dir) / "Real_Estate_Financial_Model.xlsx"
-            
-            # Run pipeline
-            result = pipeline.run_full_pipeline(temp_dir, str(output_file))
-            
-            # Generate summary text
-            summary = f"""
-            ✅ Processing Complete!
-            
-            Key Metrics:
-            • Total Project Cost: ${pipeline.formula_results.get('TOTAL_FINANCING_CONTINGENCY_AND_RESERVES', 0):,.0f}
-            • Total Debt: ${pipeline.formula_results.get('TOTAL_DEBT', 0):,.0f}
-            • Total Equity: ${pipeline.formula_results.get('TOTAL_EQUITY', 0):,.0f}
-            • NOI: ${pipeline.formula_results.get('NET_OPERATING_INCOME', 0):,.0f}
-            • Yield on Cost: {pipeline.formula_results.get('YIELD_ON_COST_PERCENTAGE', 0):.2%}
-            • LP IRR: {float(pipeline.formula_results.get('IRR_TO_LP', 0).real if isinstance(pipeline.formula_results.get('IRR_TO_LP', 0), complex) else pipeline.formula_results.get('IRR_TO_LP', 0)):.2f}%
-            
-            Download your Excel file below ⬇️
-            """
-            
-            return str(output_file), summary
-            
-        except Exception as e:
-            return None, f"❌ Error: {str(e)}"
+        # Save uploaded PDFs to temp directory
+        for pdf_file in pdf_files:
+            if pdf_file is None:
+                continue
+            dest_path = Path(temp_dir) / Path(pdf_file.name).name
+            shutil.copy(pdf_file.name, dest_path)
+        
+        # Initialize pipeline with hardcoded API key
+        pipeline = RealEstateModelPipeline(GEMINI_API_KEY)
+        
+        # Create output file in temp directory
+        output_file = Path(temp_dir) / "Real_Estate_Financial_Model.xlsx"
+        
+        # Run pipeline
+        result = pipeline.run_full_pipeline(temp_dir, str(output_file))
+        
+        # Generate summary text
+        r = pipeline.formula_results
+        irr_val = r.get('IRR_TO_LP', 0)
+        if isinstance(irr_val, complex):
+            irr_val = irr_val.real
+        
+        summary = f"""✅ **Processing Complete!**
+
+📊 **Key Metrics:**
+• Total Project Cost: ${r.get('TOTAL_FINANCING_CONTINGENCY_AND_RESERVES', 0):,.0f}
+• Total Debt: ${r.get('TOTAL_DEBT', 0):,.0f}
+• Total Equity: ${r.get('TOTAL_EQUITY', 0):,.0f}
+• NOI: ${r.get('NET_OPERATING_INCOME', 0):,.0f}
+• Yield on Cost: {r.get('YIELD_ON_COST_PERCENTAGE', 0):.2%}
+• LP IRR: {float(irr_val):.2f}%
+
+✨ Download your Excel file below ⬇️
+"""
+        
+        return str(output_file), summary
+        
+    except Exception as e:
+        import traceback
+        error_msg = f"❌ **Error occurred:**\n\n{str(e)}\n\n**Details:**\n{traceback.format_exc()}"
+        return None, error_msg
+
+# Create Gradio interface with better styling
+with gr.Blocks(
+    title="Real Estate Financial Model Generator",
+    theme=gr.themes.Soft(),
+    css="""
+        .gradio-container {
+            max-width: 1200px !important;
+        }
+        .output-text {
+            font-family: monospace;
+        }
+    """
+) as demo:
     
-    # Create Gradio interface
-    with gr.Blocks(title="Real Estate Financial Model Generator", theme=gr.themes.Soft()) as demo:
-        
-        gr.Markdown("""
-        # Real Estate Financial Model Generator
-        Upload your PDF documents and generate a comprehensive financial model in Excel format.
-        """)
-        
-        with gr.Row():
-            with gr.Column(scale=2):
-                pdf_input = gr.File(
-                    label="Upload PDF/XLSX Files",
-                    file_count="multiple",
-                    file_types=[".pdf", ".xlsx", ".xls"],  # Added .xlsx and .xls
-                    type="filepath"
-                )
-                
-                process_btn = gr.Button("Generate Financial Model", variant="primary", size="lg")
-            
-            with gr.Column(scale=1):
-                gr.Markdown("""
-                ### 📋 Supported Formats
-                - **PDF**: Offering Memorandum, Reports
-                - **XLSX/XLS**: Financial statements, data tables
-                
-                ### 📋 Required Documents
-                - Offering Memorandum (PDF/XLSX)
-                - Operating Expenses Summary (PDF/XLSX)
-                - Sales Comps (PDF/XLSX)
-                - Rent Comps (PDF/XLSX)
-                - Market Report (PDF/XLSX)
-                - Demographics Overview (PDF/XLSX)
-                
-                ### ⚡ Features
-                - Automated data extraction
-                - Formula calculations
-                - Professional Excel output
-                - Multiple analysis sheets
-                """)
-        
-        with gr.Row():
-            output_text = gr.Textbox(
-                label="Processing Results",
-                lines=12,
-                interactive=False
+    gr.Markdown("""
+    # 🏢 Real Estate Financial Model Generator
+    
+    Upload your PDF/XLSX documents and generate a comprehensive financial model in Excel format.
+    """)
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            pdf_input = gr.File(
+                label="📁 Upload PDF/XLSX Files",
+                file_count="multiple",
+                file_types=[".pdf", ".xlsx", ".xls"],
+                type="filepath"
             )
-        
-        with gr.Row():
-            excel_output = gr.File(
-                label="📊 Download Excel File"
+            
+            process_btn = gr.Button(
+                "🚀 Generate Financial Model",
+                variant="primary",
+                size="lg"
             )
         
-        process_btn.click(
-            fn=process_pdfs,
-            inputs=[pdf_input],
-            outputs=[excel_output, output_text]
+        with gr.Column(scale=1):
+            gr.Markdown("""
+            ### 📋 Supported Formats
+            - **PDF**: Offering Memorandum, Reports
+            - **XLSX/XLS**: Financial statements, data tables
+            
+            ### 📋 Required Documents
+            - Offering Memorandum
+            - Operating Expenses Summary
+            - Sales Comps
+            - Rent Comps
+            - Market Report
+            - Demographics Overview
+            
+            ### ⚡ Features
+            - ✨ Automated data extraction
+            - 🧮 Formula calculations
+            - 📊 Professional Excel output
+            - 📈 Multiple analysis sheets
+            """)
+    
+    with gr.Row():
+        output_text = gr.Textbox(
+            label="📋 Processing Results",
+            lines=15,
+            interactive=False,
+            elem_classes=["output-text"]
         )
-        
-        gr.Markdown("""
-        ---
-        ### 💡 Tips
-        - Ensure PDF files are readable and not scanned images
-        - Use descriptive filenames (e.g., "Offering_Memorandum.pdf")
-        - Processing may take 30-60 seconds depending on file sizes
-        """)
     
-    # Launch the app
-    # demo.launch(share=False)
+    with gr.Row():
+        excel_output = gr.File(
+            label="📊 Download Excel File",
+            interactive=False
+        )
+    
+    # Connect the button
+    process_btn.click(
+        fn=process_pdfs,
+        inputs=[pdf_input],
+        outputs=[excel_output, output_text],
+        api_name="process"
+    )
+    
+    gr.Markdown("""
+    ---
+    ### 💡 Tips
+    - Ensure PDF files are readable and not scanned images
+    - Use descriptive filenames (e.g., "Offering_Memorandum.pdf")
+    - Processing may take 30-60 seconds depending on file sizes
+    - Check the **Processing Results** section for detailed feedback
+    
+    ### 🔧 API Endpoints
+    - `POST /api/generate-model` - Generate Excel only
+    - `POST /api/generate-model-with-summary` - Generate Excel + JSON summary
+    - `POST /api/analyze-only` - Generate JSON summary only
+    - `GET /api/health` - Health check
+    """)
+
+# ============= Mount and Launch =============
+if __name__ == "__main__":
+    # Mount Gradio to FastAPI
     app = gr.mount_gradio_app(app, demo, path="/")
+    
+    # Launch with proper configuration
     uvicorn.run(
-        app, 
-        host="0.0.0.0", 
+        app,
+        host="0.0.0.0",
         port=7860,
         log_level="info"
     )