Update app.py

app.py

@@ -1,16 +1,29 @@
 import gradio as gr
 import requests
+import json
 from huggingface_hub import InferenceClient
 
 API_TOKEN = "your_huggingface_api_token" # Replace with your actual token
 API_URL = "https://api-inference.huggingface.co/models/InterSync/Mistral-7B-Instruct-v0.2-Function-Calling"
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
-def get_weather(location, unit="celsius"):
-  """Gets the current weather in a given location."""
-  # Your code to fetch weather data using an external API (e.g., OpenWeatherMap)
-  ...
-  return weather_data
+def get_weather(location: str, unit: str = "celsius"):
+    # Replace with your actual weather API call
+    pass 
+    
+def get_weather_schema():
+    return {
+        "name": "get_weather",
+        "description": "Get the current weather in a given location",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "location": {"type": "string", "description": "The city and state, or zip code"},
+                "unit": {"type": "string", "enum": ["celsius", "fahrenheit"], "description": "Unit of temperature"}
+            },
+            "required": ["location"]
+        }
+    }
 
 def query_model(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
@@ -31,19 +44,25 @@ with gr.Blocks() as demo:
 
     def bot(history):
         user_message = history[-1][0]
-        function_call = {
-            "name": "get_weather",
-            "arguments": {"location": user_message}
-        }
         payload = {
             "inputs": user_message,
-            "parameters": {
-                "function_call": function_call
-            }
+            "parameters": {"function_call": "auto"}
         }
         output = query_model(payload)
-        bot_response = output[0]['generated_text']  # Assuming API returns text in this format
-        history[-1][1] = bot_response  # Update the last message in history with bot response
+
+        # Parse the model's response
+        if 'function_call' in output and 'name' in output['function_call']:
+            function_name = output['function_call']['name']
+            arguments = output['function_call'].get('arguments', {})
+            if function_name == "get_weather" and arguments:
+                weather_info = get_weather(**arguments)
+                response_message = f"The weather in {arguments['location']} is {weather_info['description']} with a temperature of {weather_info['temperature']} {weather_info['unit']}."
+            else:
+                response_message = "Function not found or invalid arguments."
+        else:
+            response_message = output[0]['generated_text']
+
+        history[-1][1] = response_message
         return history
 
     input_text.change(user, [input_text, output_text], [input_text, output_text], queue=False).then(