Update app.py

app.py

@@ -158,6 +158,20 @@ def function(Textbox, Textbox2, Textbox3, Dropdown):
             
         if Textbox3 == target:
             if Dropdown == "Option1": 
+                def get_current_time(location):
+                    url = "https://api.api-ninjas.com/v1/worldtime?city="
+                    url += location
+                    headers = {
+                        'X-Api-Key': 'Fl79Q6SqQz8BWQcaYGeQ5A==o5DKruyij5IfnMLM'
+                    }
+                    response = requests.request("GET", url, headers=headers)
+                    data = response.json()
+                    if response.status_code == 200:
+                        time = data
+                        return f"The current time in {location} is {time}."
+                    else:
+                        return f"Failed to retrieve the current time for {location}."
+
                 Textbox2 = Textbox2.replace("[", "").replace("]", "").replace("'", "")
                 Textbox2 = Textbox2.split(",")
                 Textbox2_edited = [x.strip() for x in Textbox2]
@@ -174,17 +188,55 @@ def function(Textbox, Textbox2, Textbox3, Dropdown):
                     )
                 try:
                     sleep(0.8)
-                    chat = openai.ChatCompletion.create(
-                        model=model, messages=data["messages"]
-                    )
-                    reply = chat.choices[0].message.content
+                    def chat_with_assistant(messages, functions=None):
+                        chat = openai.ChatCompletion.create(
+                            model=model, messages=data["messages"]
+                        )
+                        reply = chat.choices[0].message.content
+                        return reply
+
+                    functions = [
+                        {
+                            "name": "get_current_time",
+                            "description": "Get the current time in a given location",
+                            "parameters": {
+                                "type": "object",
+                                "properties": {
+                                    "location": {
+                                        "type": "string",
+                                        "description": "The city and state, e.g. San Francisco, CA",
+                                    },
+                                    "unit": {"type": "string", "enum": ["time", "date", "both"]},
+                                },
+                                "required": ["location"],
+                            },
+                        }
+                    ]
                     # response = requests.post(target2, json=data)
                     # print("target2",target2)
                     # reply = response.content.decode("utf-8")
                     # reply = reply.replace(" ", "%20")
                     # image_array = download_image(f"https://api.placid.app/u/pydav4ibo?quote[text]={reply}")
                     data["messages"].append({"role": "assistant", "content": reply})
-                    return reply
+                    assistant_response = chat_with_assistant(messages, functions)
+                    if assistant_response.get("function_call"):
+                        function_name = assistant_response["function_call"]["name"]
+                        function_args = json.loads(assistant_response["function_call"]["arguments"])
+                        if function_name == "get_current_time":
+                            time = get_current_time(function_args["location"])
+                            result = time
+                        else:
+                            return {assistant_response['content']}
+
+                         data["messages"].append({
+                            "role": "function",
+                            "name": function_name,
+                            "content": str(result)
+                        })
+                        
+                        assistant_response = chat_with_assistant(messages, functions)
+                    return {assistant_response['content']}
+                    data["messages"].append(assistant_response)
                 except Exception as e:
                     print(traceback.format_exc())
                     return "Please Wait!"