Update NLP_dispatcher.py

NLP_dispatcher.py

@@ -1,48 +1,58 @@
-import gradio as gr
-
-# Define routing keywords for Extension Agents and Ag Lenders
-extension_keywords = {
-    "Jerry": ["depreciation", "tractor", "equipment", "machinery"],
-    "Amanda": ["enterprise", "budget", "revenue", "expenses"],
-    "James": ["partial budget", "compare", "replace", "cost saving"],
-    "Edward": ["balance sheet", "assets", "liabilities", "net worth"],
-    "Alessandra": ["income statement", "profit", "loss", "cash flow"]
-}
-
-lender_keywords = {
-    "Carlos": ["loan", "financing", "refinance", "repayment"],
-    "Warren": ["approval", "final plan", "executive", "capstone"]
-}
-
-# Match farm plan to best agent or lender
-def route_to_agent(farm_plan):
-    farm_plan_lower = farm_plan.lower()
-
-    for agent, keywords in extension_keywords.items():
-        for word in keywords:
-            if word in farm_plan_lower:
-                return f"📞 Connecting you to Extension Agent {agent}. They can help you with that topic and guide you through Extension resources."
-
-    for lender, keywords in lender_keywords.items():
-        for word in keywords:
-            if word in farm_plan_lower:
-                return f"🏦 Forwarding you to Loan Officer {lender} for project review and financial analysis."
-
-    return "🤖 Sorry, I couldn't match your plan to a specific agent or lender. Try using keywords like 'depreciation', 'income', or 'loan'."
-
-
-with gr.Blocks() as app:
-    gr.Markdown("""
-    # 🧭 AgLendGPT Directory: Extension Agents & Lenders
-    Submit your farm plan below. We’ll connect you to the right Extension agent for support, or to an Ag Credit officer for projects.
-    """)
-
-    farm_plan_input = gr.Textbox(label="Describe your farm plan", placeholder="e.g., I want to expand with new equipment and need help calculating depreciation.", lines=4)
-    submit_btn = gr.Button("📡 Connect Me")
-    response = gr.Textbox(label="Routing Result")
-
-    submit_btn.click(fn=route_to_agent, inputs=farm_plan_input, outputs=response)
-
-
-if __name__ == "__main__":
-    app.launch()
+import gradio as gr
+
+# Define routing keywords for Extension Agents and Ag Lenders
+extension_keywords = {
+    "Jerry": ["depreciation", "tractor", "equipment", "machinery"],
+    "Amanda": ["enterprise", "budget", "revenue", "expenses"],
+    "James": ["partial budget", "compare", "replace", "cost saving"],
+    "Edward": ["balance sheet", "assets", "liabilities", "net worth"],
+    "Alessandra": ["income statement", "profit", "loss", "cash flow"]
+}
+
+lender_keywords = {
+    "Carlos": ["loan", "financing", "refinance", "repayment"],
+    "Warren": ["approval", "final plan", "executive", "capstone"]
+}
+
+JERRY_URL = os.environ.get("JERRY_URL", "https://huggingface.co/spaces/jeffrey1963/Jerry_Depreciation")
+
+
+# Match farm plan to best agent or lender
+def route_to_agent(farm_plan):
+    farm_plan_lower = farm_plan.lower()
+
+    for agent, keywords in extension_keywords.items():
+        for word in keywords:
+            if word in farm_plan_lower:
+                return f"📞 Connecting you to Extension Agent {agent}. They can help you with that topic and guide you through Extension resources."
+
+    for lender, keywords in lender_keywords.items():
+        for word in keywords:
+            if word in farm_plan_lower:
+                return f"🏦 Forwarding you to Loan Officer {lender} for project review and financial analysis."
+
+    return "🤖 Sorry, I couldn't match your plan to a specific agent or lender. Try using keywords like 'depreciation', 'income', or 'loan'."
+
+def render_result(route, student_id):
+    if route == "extension_jerry":
+        link = f"{JERRY_URL}?student_id={student_id or ''}"
+        return (f"☎️ Connecting you to **Extension Agent Jerry** for depreciation help.\n\n"
+                f"➡️ **Open Jerry**: {link}")
+    # ... other routes ...
+
+
+with gr.Blocks() as app:
+    gr.Markdown("""
+    # 🧭 AgLendGPT Directory: Extension Agents & Lenders
+    Submit your farm plan below. We’ll connect you to the right Extension agent for support, or to an Ag Credit officer for projects.
+    """)
+
+    farm_plan_input = gr.Textbox(label="Describe your farm plan", placeholder="e.g., I want to expand with new equipment and need help calculating depreciation.", lines=4)
+    submit_btn = gr.Button("📡 Connect Me")
+    response = gr.Textbox(label="Routing Result")
+
+    submit_btn.click(fn=route_to_agent, inputs=farm_plan_input, outputs=response)
+
+
+if __name__ == "__main__":
+    app.launch()