Update app.py

app.py

@@ -399,14 +399,7 @@ def summarize_web_results(query: str, search_results: List[Dict[str, str]], conv
         return f"An error occurred during summarization: {str(e)}"
 
 # Modify the existing respond function to handle both PDF and web search
-def respond(message, history, model, temperature, num_calls, use_web_search, selected_docs):
-    if not message:
-        return history + [(None, "No query provided. Please try again.")]
-    
-    # Remove any path-like prefixes from the message
-    message = message.split('/')[-1] if '/' in message else message
-    
-    
+def respond(message, history, model, temperature, num_calls, use_web_search, selected_docs):    
     logging.info(f"User Query: {message}")
     logging.info(f"Model Used: {model}")
     logging.info(f"Selected Documents: {selected_docs}")
@@ -478,11 +471,14 @@ def respond(message, history, model, temperature, num_calls, use_web_search, sel
             else:
                 response = f"An error occurred with the {model} model: {str(e)}. Please try again or select a different model."
 
+    # Ensure response is a string
+    response = str(response)
+
     # Update the conversation history
-    history.append((message, response))
+    new_history = history + [(message, response)]
     
     # Yield the updated history
-    yield history
+    yield new_history
 
 logging.basicConfig(level=logging.DEBUG)
 
@@ -665,14 +661,25 @@ def transcribe_and_respond(audio_file, history, model, temperature, num_calls, u
     logging.info(f"Query extracted from transcription: {query}")
     
     # Use the transcription as the query
-    response = next(respond(query, history, model, temperature, num_calls, use_web_search, document_selector))
-    
-    logging.info(f"Response: {response}")
+    response_generator = respond(query, history, model, temperature, num_calls, use_web_search, document_selector)
     
-    # Update the history with the new query and response
-    new_history = history + [(query, response)]
-    
-    return new_history
+    # Get the first (and presumably only) response from the generator
+    try:
+        new_history = next(response_generator)
+        logging.info(f"New history after response: {new_history}")
+        
+        # Ensure the last message pair in the history is correctly formatted
+        if new_history and isinstance(new_history[-1], tuple) and len(new_history[-1]) == 2:
+            query, response = new_history[-1]
+            new_history[-1] = (query, str(response))  # Ensure response is a string
+        else:
+            logging.warning(f"Unexpected format in new_history: {new_history}")
+            new_history = history + [(query, "Error: Unexpected response format")]
+        
+        return new_history
+    except StopIteration:
+        logging.warning("No response generated")
+        return history + [(query, "Error: No response generated")]
 
 def vote(data: gr.LikeData):
     if data.liked: