Update app.py

app.py

@@ -1,36 +1,30 @@
 import gradio as gr
-import requests
+import requests, json, re
 
-MCP_URL = "https://free-wombats-make.loca.lt"
-
-API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-base"
+MCP_URL = "https://pretty-areas-make.loca.lt/mcp"
 
 def query_llm(prompt):
-    response = requests.post(API_URL, json={"inputs": prompt})
+    url = "https://huggingface-projects-falcon-7b-instruct.hf.space/run/predict"
+    response = requests.post(url, json={"data": [prompt]})
+    
     if response.status_code == 200:
         result = response.json()
-        if isinstance(result, list):
-            return result[0].get('generated_text', '')
-        else:
-            raise ValueError("Unexpected response structure:", result)
+        return result['data'][0]
     else:
         raise ValueError(f"API Error: {response.status_code}")
 
 def extract_query_params(query):
     prompt = f"""
-    Extract date and quantity from the following sentence '{query}'.
-    Return the response exactly in JSON format like:
-    {{"date_from": "YYYY-MM-DD", "min_quantity": number}}
+    Extract date and quantity from the sentence '{query}'.
+    Respond only in JSON format:
+    {{"date_from":"YYYY-MM-DD","min_quantity":number}}
     """
-
     llm_response = query_llm(prompt)
 
-    import re, json
     try:
         json_text = re.search(r"\{.*\}", llm_response, re.DOTALL).group()
         return json.loads(json_text)
     except:
-        # Default fallback if parsing fails
         return {"date_from":"2023-01-01","min_quantity":0}
 
 def agent_query(user_query):
@@ -64,7 +58,7 @@ iface = gr.Interface(
     fn=agent_query,
     inputs="text",
     outputs="text",
-    title="Dynamic MCP Agent for SAP (Free LLM)",
+    title="Dynamic MCP Agent for SAP (Free Falcon LLM)",
     description="Ask SAP queries naturally, e.g., 'Orders after 2023-04-01 with at least 10 units.'"
 )