Update app.py

app.py

@@ -248,7 +248,7 @@ def create_ui() -> gr.Blocks:
             return history
 
         def handle_end_chat(history: List[Dict[str, str]]) -> Tuple[str, str]:
-            """Ends the chat and extracts keywords from the conversation."""
+            """Ends the chat, extracts and sanitizes keywords from the conversation."""
             if not history:
                 return "", "Status: Chat is empty, nothing to analyze."
             
@@ -257,9 +257,24 @@ def create_ui() -> gr.Blocks:
             if not tuple_history:
                 return "", "Status: No completed conversations to analyze."
                 
-            keywords_str = extract_keywords_from_conversation(tuple_history)
+            # Get raw keywords string from the LLM
+            raw_keywords_str = extract_keywords_from_conversation(tuple_history)
+            
+            # Sanitize the LLM output to extract only keyword-like parts.
+            # A keyword can contain letters, numbers, underscores, spaces, and hyphens.
+            cleaned_keywords = re.findall(r'[\w\s-]+', raw_keywords_str)
+            
+            # Trim whitespace from each found keyword and filter out any empty strings
+            cleaned_keywords = [kw.strip() for kw in cleaned_keywords if kw.strip()]
+            
+            if not cleaned_keywords:
+                return "", f"Status: Could not extract valid keywords. Raw LLM output: '{raw_keywords_str}'"
+
+            # Join them into a clean, comma-separated string for the search tool
+            final_keywords_str = ", ".join(cleaned_keywords)
+            
             status = "Status: Keywords extracted. You can now use them to search."
-            return keywords_str, status
+            return final_keywords_str, status
 
         # --- Component Event Wiring ---