Update app.py

app.py

@@ -139,13 +139,21 @@ async def respond(message, system_prompt, history, model, temperature, num_calls
     logging.info(f"Number of API Calls: {num_calls}")
     logging.info(f"Use Embeddings: {use_embeddings}")
     logging.info(f"System Prompt: {system_prompt}")
+    logging.info(f"History: {history}")  # Log the history for debugging
 
     # Convert gradio history to the format expected by get_response_with_search
     chat_history = []
-    for human, assistant in history:
-        chat_history.append({"role": "user", "content": human})
-        if assistant:
-            chat_history.append({"role": "assistant", "content": assistant})
+    if history:
+        for entry in history:
+            if isinstance(entry, (list, tuple)) and len(entry) == 2:
+                human, assistant = entry
+                chat_history.append({"role": "user", "content": human})
+                if assistant:
+                    chat_history.append({"role": "assistant", "content": assistant})
+            elif isinstance(entry, str):
+                # If it's a string, assume it's a user message
+                chat_history.append({"role": "user", "content": entry})
+            # Ignore any other formats
 
     try:
         full_response = ""
@@ -190,8 +198,16 @@ css = """
 def create_gradio_interface():
     custom_placeholder = "Enter your question here for web search."
 
+    async def wrapped_respond(*args):
+        try:
+            async for response in respond(*args):
+                yield response
+        except Exception as e:
+            logging.error(f"Error in wrapped_respond: {str(e)}")
+            yield f"An error occurred: {str(e)}"
+
     demo = gr.ChatInterface(
-        fn=respond,
+        fn=wrapped_respond,  # Use the wrapped version
         additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=True, render=False),
         additional_inputs=[
             gr.Textbox(value=DEFAULT_SYSTEM_PROMPT, lines=6, label="System Prompt", placeholder="Enter your system prompt here"),