Update app.py

app.py

@@ -5,7 +5,8 @@ import re
 
 web_search = DuckDuckGoSearchTool()
 
-SYSTEM_PROMPT = """You are an AI research assistant that can search the web. Follow these steps:
+SYSTEM_PROMPT = """
+You are an AI research assistant that can search the web. Follow these steps:
 
 1. FIRST ANALYZE the user's question:
    - If information is missing or ambiguous, ask ONE clarifying question
@@ -29,16 +30,19 @@ SYSTEM_PROMPT = """You are an AI research assistant that can search the web. Fol
      - Key facts with sources
      - Concise explanations
 
-Never invent information. Cite sources for all facts. Use neutral, academic tone."""
+Never invent information. Cite sources for all facts. Use neutral, academic tone.
+"""
 
 def process_searches(response):
-    searches = re.findall(r'<search>(.*?)</search>', response, re.DOTALL)
+    # Preserve thinking tags while processing searches
+    formatted_response = response.replace("<thinking>", "\n💭 THINKING PROCESS:\n").replace("</thinking>", "\n")
+    searches = re.findall(r'<search>(.*?)</search>', formatted_response, re.DOTALL)
     if searches:
         queries = [q.strip() for q in searches[0].split('\n') if q.strip()]
         results = []
         for query in queries:
             search_result = web_search(query)
-            results.append(f"🔍 Search results for '{query}':\n{search_result}\n")
+            results.append(f"🔍 SEARCH: {query}\nRESULTS: {search_result}\n")
         return '\n'.join(results)
     return None
 
@@ -86,9 +90,9 @@ def respond(
             token = chunk.choices[0].delta.content or ""
             response += token
             full_response += token
-            yield full_response
-
-        search_results = process_searches(response)
+            # Display thinking tags immediately
+            if "<thinking>" in token.lower() or "</thinking>" in token.lower():
+                yield full_response
         
     if search_results:
         search_cycle = True
@@ -117,7 +121,8 @@ demo = gr.ChatInterface(
         ["Compare COVID-19 mortality rates between US and Sweden with sources"],
         ["What's the current consensus on dark matter composition?"],
         ["Latest advancements in fusion energy 2023-2024"]
-    ]
+    ],
+    cache_examples=False
 )
 
 if __name__ == "__main__":