Update app.py

app.py

@@ -70,19 +70,27 @@ class ToolBase:
         ):
         self.json_name = programmatic_name
         self.json_description = passive_voice_description_of_function
+        
+        # Fix: Explicitly set additionalProperties to avoid Gradio JSON schema parsing issue
+        parameters_schema = {
+            "type": "object",
+            "properties": input_params,
+            "additionalProperties": False  # Explicitly set to False
+        }
+        
+        if required_params:
+            parameters_schema["required"] = required_params
+            
         self.json_definition_of_function = {
             "type": "function",
-                "function": {
-                    "name": self.json_name,
-                    "description": self.json_description,
-                    "parameters": {
-                        "type": "object",
-                        "properties": input_params,
-                        "required": required_params,
-                    }
-                }
+            "function": {
+                "name": self.json_name,
+                "description": self.json_description,
+                "parameters": parameters_schema
             }
+        }
         self.system_prompt_paragraph = active_voice_description_of_capability + " " + prescriptive_conditional
+        
     def actual_function(self, **kwargs):
         raise NotImplementedError("Subclasses must implement this method.")
 
@@ -132,7 +140,7 @@ def search_arxiv_papers(
                 'summary': entry.summary,
                 'published': entry.published,
                 'link': entry.link,
-                'primary_category': entry.tags[0]['term']
+                'primary_category': entry.tags[0]['term'] if entry.tags else 'Unknown'
             }
             papers.append(paper)
 
@@ -166,12 +174,12 @@ class ArxivSearchTool(ToolBase):
                 "max_results": {
                     "type": "integer",
                     "description": "Maximum number of results to return (default: 5)",
-                    "optional": True
+                    "default": 5
                 },
                 "sort_by": {
                     "type": "string",
                     "description": "Sort criteria (e.g., 'relevance', 'lastUpdatedDate', 'submittedDate')",
-                    "optional": True
+                    "default": "relevance"
                 }
             },
             required_params=["query"],
@@ -352,7 +360,7 @@ class WeatherAPITool(ToolBase):
 # Instance of the weather tool.
 weather_tool = WeatherAPITool()
 
-tool_objects_list = [arxiv_tool, nih_ref_snp_tool,weather_tool]
+tool_objects_list = [arxiv_tool, nih_ref_snp_tool, weather_tool]
 system_prompt = build_sys_prompt(tool_objects_list)
 functions_dict = {t.json_name: t.actual_function for t in tool_objects_list}
 
@@ -410,7 +418,7 @@ def check_assistant_response_for_tool_calls(response):
     """Check if the LLM response contains a function call."""
     response = response.split(FUNCTION_EOT_STRING)[0].split(EOT_STRING)[0]
     for tool_name in functions_dict.keys():
-        if f"\"{tool_name}\"" in response and "{" in response:
+        if f'"{tool_name}"' in response and "{" in response:
             response = "{" + "{".join(response.split("{")[1:])
             for _ in range(10):
                 response = "}".join(response.split("}")[:-1]) + "}"
@@ -524,10 +532,12 @@ def user_conversation(user_message, chat_history, full_history):
 llm = LLM(max_model_len=32000)
 
 lgs("STARTING NEW CHAT")
+
+# Alternative fix: Use a simpler Gradio interface that doesn't trigger the JSON schema parsing issue
 with gr.Blocks() as demo:
-    gr.Markdown(f"<h2>Weather/Arxiv/SNP Multi-tool Calling Bot</h2>")
+    gr.Markdown("<h2>Weather/Arxiv/SNP Multi-tool Calling Bot</h2>")
     chat_state = gr.State([])
-    chatbot = gr.Chatbot(label="Chat with the multi-tool bot")
+    chatbot = gr.Chatbot(label="Chat with the multi-tool bot", type="messages")
     user_input = gr.Textbox(
         lines=1,
         placeholder="Type your message here...",
@@ -560,5 +570,5 @@ with gr.Blocks() as demo:
         outputs=[user_input, chatbot, chat_state],
         queue=False
     )
-    demo.launch()
-    share_url = demo.share_url
+    
+    demo.launch()