push

app.py

@@ -91,7 +91,8 @@ def search(query):
     return all_results
 
 def process_query(user_input, history):
-    yield 'Preparing your request 🛠️'
+   
+    yield {"role": "assistant", "content": 'Preparing your request 🛠️'}
     
     # Step 1: Generate a search term based on the user query
     stream_search = client.chat_completion(
@@ -108,7 +109,8 @@ def process_query(user_input, history):
         search_query += content
 
     # Step 2: Perform the web search with the generated term
-    yield 'Searching the web for relevant information 🌐'
+    
+    yield {"role": "assistant", "content": 'Searching the web for relevant information 🌐'}
     search_results = search(search_query)
     
     # Format results as a JSON string for model input
@@ -122,14 +124,14 @@ def process_query(user_input, history):
         stream=True
     )
 
-    yield "Analyzing the search results and crafting a response 📊"
     
+    yield {"role": "assistant", "content": 'Analyzing the search results and crafting a response 📊'}
     # Stream final response
     final_response = ""
     for chunk in response:
         content = chunk.choices[0].delta.content or ''
         final_response += content
-        yield final_response
+        yield {"role": "assistant", "content": final_response}
 
 theme = gr.themes.Citrus(
     primary_hue="blue",