Update app.py

app.py

@@ -9,38 +9,35 @@ web_search = DuckDuckGoSearchTool()
 SYSTEM_PROMPT = """
 You are an AI research assistant that can search the web. Follow these steps:
 
-<thinking>
-1. Analyze the user's question:
-   - Identify missing information or ambiguities
-   - Determine if clarification is needed
-   - Plan search strategy if required
-</thinking>
-
-If clarification is needed, ask ONE question. Otherwise:
-
-<search>
-[list of search queries]
-</search>
-
-After receiving results:
-<thinking>
-- Analyze information from multiple sources
-- Cross-verify facts
-- Determine if additional searches are needed
-</thinking>
-
-Final answer:
-- Clear structure with headings
-- Key facts with sources
-- Concise explanations
-- Academic tone
+1. FIRST ANALYZE the user's question:
+   - If information is missing or ambiguous, ask ONE clarifying question
+   - If clear, proceed to search
+
+2. When searching:
+   - Generate multiple specific search queries wrapped in <search> tags
+   - Focus on factual keywords, one query per line
+   Example:
+   <search>
+   Pont des Arts exact length meters
+   History of Pont des Arts bridge
+   </search>
+
+3. After receiving results:
+   - Analyze information from multiple sources
+   - Cross-verify facts
+   - If needed, generate follow-up searches
+   - Provide final answer with:
+     - Clear structure
+     - Key facts with sources
+     - Concise explanations
+
+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)
     if searches:
-        queries = [q.strip() for q in searches[0].split('\n') if q.strip()]
-        return queries
+        return [q.strip() for q in searches[0].split('\n') if q.strip()]
     return None
 
 def search_with_retry(query, max_retries=3, delay=2):
@@ -99,7 +96,7 @@ def respond(
                     }
                 )
             except Exception as e:
-                yield f"⚠️ API Error: {str(e)}\n\nPlease check your OpenRouter API key."
+                yield f"⚠️ API Error: {str(e)}"
                 return
 
             response = ""
@@ -107,7 +104,7 @@ def respond(
                 token = chunk.choices[0].delta.content or ""
                 response += token
                 full_response += token
-                yield full_response  # Stream raw output with all tags
+                yield full_response  # Direct streaming without processing
 
             queries = process_searches(response)
             
@@ -119,36 +116,34 @@ def respond(
                 for query in queries:
                     try:
                         result = search_with_retry(query)
-                        search_results.append(f"🔍 SEARCH: {query}\nRESULTS: {result}\n")
+                        search_results.append(f"Search results for '{query}':\n{result}")
                     except Exception as e:
-                        search_results.append(f"⚠️ Search Error: {str(e)}\nQuery: {query}")
+                        search_results.append(f"Search failed for '{query}': {str(e)}")
                         time.sleep(2)
                 
                 messages.append({
                     "role": "user",
-                    "content": f"SEARCH RESULTS:\n{chr(10).join(search_results)}\nAnalyze these results..."
+                    "content": f"SEARCH RESULTS:\n{chr(10).join(search_results)}"
                 })
-                full_response += "\n🔍 Analyzing search results...\n"
-                yield full_response
+                yield full_response + "\n🔍 Analyzing results...\n"
 
     except Exception as e:
-        yield f"⚠️ Critical Error: {str(e)}\n\nPlease try again later."
+        yield f"⚠️ Error: {str(e)}"
 
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
         gr.Textbox(value=SYSTEM_PROMPT, label="System Prompt", lines=8),
-        gr.Slider(minimum=1000, maximum=15000, value=6000, step=500, label="Max Tokens"),
-        gr.Slider(minimum=0.1, maximum=1.0, value=0.5, step=0.1, label="Temperature"),
-        gr.Slider(minimum=0.1, maximum=1.0, value=0.85, step=0.05, label="Top-p"),
-        gr.Textbox(label="OpenRouter API Key", type="password")
+        gr.Slider(1000, 15000, 6000, step=500, label="Max Tokens"),
+        gr.Slider(0.1, 1.0, 0.5, step=0.1, label="Temperature"),
+        gr.Slider(0.1, 1.0, 0.85, step=0.05, label="Top-p"),
+        gr.Textbox(label="OpenRouter Key", type="password")
     ],
-    title="Web Research Agent 🤖",
-    description="Advanced AI assistant with web search capabilities",
+    title="Web Research Agent",
+    description="AI assistant with web search capabilities",
     examples=[
         ["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"]
+        ["What's the current consensus on dark matter composition?"]
     ],
     cache_examples=False
 )