| import gradio as gr | |
| from investment_recommendation import recommend_investment | |
| def recommend_api(risk, horizon, investment_amount): | |
| try: | |
| user = { | |
| "risk": risk, | |
| "horizon": horizon, | |
| "investment_amount": float(investment_amount) | |
| } | |
| result = recommend_investment(user) | |
| return {"status": "success", "data": result} | |
| except Exception as e: | |
| return {"status": "error", "message": str(e)} | |
| with gr.Blocks(title="FinTrack Investment Recommendation") as demo: | |
| gr.Markdown("## 💹 FinTrack Investment Recommendation API") | |
| risk = gr.Dropdown(choices=["Aggressive", "Conservative", "Moderate"], label="Risk appetite") | |
| horizon = gr.Dropdown(choices=["Short-term", "Medium-term", "Long-term"], label="Investment horizon") | |
| amount = gr.Number(label="Investment amount (₹)", precision=2) | |
| output = gr.JSON(label="Recommendation") | |
| btn = gr.Button("Get Recommendation") | |
| btn.click(recommend_api, inputs=[risk, horizon, amount], outputs=output) | |
| demo.queue() # enables API calls | |
| if __name__ == "__main__": | |
| demo.launch() | |