Fix TypeError in submit_query and set Chatbot type to messages

app.py

@@ -152,7 +152,7 @@ async def ask_gemini(query: str, history: List[Dict[str, str]]) -> str:
         logger.error(f"Gemini Error: {str(e)}")
         return f"Error: Gemini Error: {str(e)}"
 
-async def query_model(query: str, provider: str, history: List[Dict[str, str]]) -> Tuple[str, List[Dict[str, str]]]:
+async def query_model(query: str, provider: str, history: List[Dict[str, str]]) -> AsyncGenerator[Tuple[str, List[Dict[str, str]]], None]:
     provider = provider.lower()
     response = ""
     
@@ -174,6 +174,16 @@ async def query_model(query: str, provider: str, history: List[Dict[str, str]])
     updated_history = history + [{"user": query, "bot": response}]
     yield response, updated_history  # Final yield with updated history
 
+async def submit_query(query: str, provider: str, history: List[Dict[str, str]]) -> Tuple[str, List[Dict[str, str]], List[Dict[str, str]]]:
+    if not query.strip():
+        return "", history, history
+    
+    response = ""
+    async for response_chunk, updated_history in query_model(query, provider, history):
+        response += response_chunk
+        yield "", updated_history, updated_history  # Yield intermediate updates for streaming
+    return "", updated_history, updated_history  # Final return with cleared query
+
 # Gradio interface
 def clear_history():
     return [], []
@@ -184,18 +194,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     
     provider = gr.Dropdown(choices=["OpenAI", "Anthropic", "Gemini"], label="Select Provider", value="OpenAI")
     history_state = gr.State(value=[])
-    chatbot = gr.Chatbot(label="Conversation")
+    chatbot = gr.Chatbot(label="Conversation", type="messages")
     query = gr.Textbox(label="Enter your query", placeholder="e.g., What is the capital of the United States?")
     submit_button = gr.Button("Submit")
     clear_button = gr.Button("Clear History")
 
-    def submit_query(query, provider, history):
-        if not query.strip():
-            return "", history, history
-        for response_chunk, updated_history in query_model(query, provider, history):
-            yield response_chunk, updated_history, updated_history
-        return "", updated_history, updated_history
-
     submit_button.click(
         fn=submit_query,
         inputs=[query, provider, history_state],