Update app.py

app.py

@@ -1,48 +1,147 @@
 #Gradio app
 # app.py
 # app.py - Original approach with profile object
+# app.py
 import gradio as gr
 import requests
+import json
 
-MODAL_URL = "https://devsam2898--personal-investment-strategist-9-web.modal.run/strategy"
+MODAL_URL = "https://devsam2898--personal-investment-strategist-10-web.modal.run/strategy"
 
 def get_investment_strategy(age_group, income, expenses, risk_profile, goal, timeframe):
-    """Get investment strategy from Modal backend"""
+    """
+    Get investment strategy from Modal backend
+    """
+    # Input validation
+    if not all([age_group, income, expenses, risk_profile, goal, timeframe]):
+        return "❌ Please fill in all fields to get a personalized strategy."
+    
+    # Convert income and expenses to numbers if they're strings
+    try:
+        income_val = float(income.replace('$', '').replace(',', '')) if isinstance(income, str) else float(income)
+        expenses_val = float(expenses.replace('$', '').replace(',', '')) if isinstance(expenses, str) else float(expenses)
+    except ValueError:
+        return "❌ Please enter valid numbers for income and expenses."
+    
     payload = {
         "profile": {
             "age_group": age_group,
-            "income": income,
-            "expenses": expenses,
+            "income": income_val,
+            "expenses": expenses_val,
             "risk_profile": risk_profile,
             "goal": goal,
             "timeframe": timeframe
         }
     }
+    
     try:
-        response = requests.post(MODAL_URL, json=payload, timeout=60)
+        print(f"Sending request to: {MODAL_URL}")
+        print(f"Payload: {json.dumps(payload, indent=2)}")
+        
+        response = requests.post(
+            MODAL_URL, 
+            json=payload,
+            headers={
+                'Content-Type': 'application/json',
+                'Accept': 'application/json'
+            },
+            timeout=30  # Add timeout
+        )
+        
+        print(f"Response status: {response.status_code}")
+        print(f"Response headers: {dict(response.headers)}")
+        
         if response.status_code == 200:
-            return response.json().get("strategy", "No strategy returned.")
+            result = response.json()
+            strategy = result.get("strategy", "No strategy returned.")
+            return f"## 📊 Your Personalized Investment Strategy\n\n{strategy}"
         else:
-            return f"❌ Error {response.status_code}: {response.text}"
+            error_msg = f"❌ **Error {response.status_code}**\n\n"
+            try:
+                error_detail = response.json()
+                error_msg += f"Details: {error_detail}"
+            except:
+                error_msg += f"Response: {response.text}"
+            return error_msg
+            
+    except requests.exceptions.Timeout:
+        return "❌ **Request Timeout**: The service is taking too long to respond. Please try again."
+    except requests.exceptions.ConnectionError:
+        return "❌ **Connection Error**: Unable to connect to the backend service. Please check if the Modal service is running."
+    except requests.exceptions.RequestException as e:
+        return f"❌ **Network Error**: {str(e)}"
     except Exception as e:
-        return f"❌ Network error: {str(e)}"
+        return f"❌ **Unexpected Error**: {str(e)}"
+
+# Test function to verify the function works
+def test_function():
+    return "✅ Function is working correctly!"
 
-demo = gr.Interface(
-    fn=get_investment_strategy,
-    inputs=[
-        gr.Dropdown(["20s", "30s", "40s", "50s+"], label="Age Group", value="30s"),
-        gr.Textbox(label="Monthly Income ($)", value="6000"),
-        gr.Textbox(label="Monthly Expenses ($)", value="4000"),
-        gr.Radio(["Conservative", "Moderate", "Aggressive"], label="Risk Tolerance", value="Moderate"),
-        gr.Textbox(label="Financial Goal (e.g., buy a house)", value="retirement planning"),
-        gr.Textbox(label="Timeframe (e.g., 5 years)", value="10 years")
-    ],
-    outputs=gr.Markdown(label="Investment Strategy"),
-    title="📈 Personal Finance Strategist",
-    description="Powered by LlamaIndex + Nebius Qwen3 | Get personalized strategies based on life stage and goals.",
-    theme="soft",
-    allow_flagging="never"
-)
+# Create the interface
+with gr.Blocks(theme="soft", title="📈 Personal Finance Strategist") as interface:
+    gr.Markdown(
+        """
+        # 📈 Personal Finance Strategist
+        **Powered by LlamaIndex + Nebius Qwen3**
+        
+        Get personalized investment strategies based on your life stage and financial goals.
+        """
+    )
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            age_group = gr.Dropdown(
+                choices=["20s", "30s", "40s", "50s+"], 
+                label="Age Group",
+                value="30s"
+            )
+            income = gr.Textbox(
+                label="Monthly Income ($)", 
+                value="6000",
+                placeholder="e.g., 6000"
+            )
+            expenses = gr.Textbox(
+                label="Monthly Expenses ($)", 
+                value="4000",
+                placeholder="e.g., 4000"
+            )
+            
+        with gr.Column(scale=1):
+            risk_profile = gr.Radio(
+                choices=["Conservative", "Moderate", "Aggressive"], 
+                label="Risk Tolerance",
+                value="Moderate"
+            )
+            goal = gr.Textbox(
+                label="Financial Goal", 
+                placeholder="e.g., buy a house, retirement, emergency fund"
+            )
+            timeframe = gr.Textbox(
+                label="Timeframe", 
+                placeholder="e.g., 5 years, 10 years"
+            )
+    
+    with gr.Row():
+        submit_btn = gr.Button("Get Investment Strategy", variant="primary")
+        test_btn = gr.Button("Test Connection", variant="secondary")
+    
+    output = gr.Markdown(label="Investment Strategy")
+    
+    # Event handlers
+    submit_btn.click(
+        fn=get_investment_strategy,
+        inputs=[age_group, income, expenses, risk_profile, goal, timeframe],
+        outputs=output
+    )
+    
+    test_btn.click(
+        fn=test_function,
+        outputs=output
+    )
 
 if __name__ == "__main__":
-    demo.launch()
+    interface.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
+    )