Update app.py

app.py

@@ -3,28 +3,34 @@ import requests
 
 MCP_URL = "https://free-wombats-make.loca.lt"
 
-API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
+API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-base"
 
 def query_llm(prompt):
     response = requests.post(API_URL, json={"inputs": prompt})
-    print(response.status_code, response.text)  # Debugging line
-    return response.json()[0]['generated_text']
-
+    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)
+    else:
+        raise ValueError(f"API Error: {response.status_code}")
 
 def extract_query_params(query):
     prompt = f"""
-    Extract date and quantity from: '{query}'.
-    Respond with JSON format only:
-    {{"date_from":"YYYY-MM-DD","min_quantity":number}}
+    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}}
     """
+
     llm_response = query_llm(prompt)
 
-    # Clean up and safely parse response
     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):