Update app.py

app.py

@@ -511,11 +511,26 @@ def run_discovery(state: EnhancedAgentState, progress_callback=None) -> Enhanced
         topic = state.get("search_topic", "general science")
         
         system_msg = SystemMessage(content=f"""You are the Discovery Agent for Tuesday Trivia.
-        Mission: Find 10-15 recent scientific breakthroughs (2-4 weeks).
-        Tools: search_science_breakthroughs, get_all_previous_posts, check_topic_similarity.
-        Output: List 10-15 unique breakthroughs with Title, Desc, URL, Why interesting.""")
+
+CRITICAL INSTRUCTIONS:
+1. You MUST use the search_science_breakthroughs tool to find recent articles
+2. You MUST use get_all_previous_posts to check for duplicates
+3. You MUST use check_topic_similarity to avoid similar topics
+4. Find 10-15 RECENT breakthroughs from the last 2-4 weeks
+5. Each breakthrough needs: Title, Brief Description, Source URL, Why it's interesting
+
+DO NOT provide generic information. ALWAYS use the tools to search for current data.
+
+Output Format:
+For each breakthrough provide:
+**Title:** [Exact title from source]
+**Description:** [2-3 sentences]
+**URL:** [Source link]
+**Why Interesting:** [1 sentence hook]
+---
+""")
         
-        user_msg = HumanMessage(content=f"Search for recent breakthroughs in {topic}.")
+        user_msg = HumanMessage(content=f"Search for recent breakthroughs in {topic}. Use the search tool to find current articles from the past 2-4 weeks.")
         
         discovery_llm = llm_small.bind_tools([search_science_breakthroughs, get_all_previous_posts, check_topic_similarity])
         
@@ -525,8 +540,10 @@ def run_discovery(state: EnhancedAgentState, progress_callback=None) -> Enhanced
         response = discovery_llm.invoke([system_msg, user_msg])
         conversation = [system_msg, user_msg, response]
         
-        max_steps = 5
+        max_steps = 8  # Increased to ensure tool usage
         steps = 0
+        tool_was_called = False
+        
         while hasattr(response, 'tool_calls') and response.tool_calls and steps < max_steps:
             if progress_callback:
                 progress_callback(f"🔧 Executing tools (step {steps+1}/{max_steps})...")
@@ -535,6 +552,10 @@ def run_discovery(state: EnhancedAgentState, progress_callback=None) -> Enhanced
             for tool_call in response.tool_calls:
                 name = tool_call['name']
                 args = tool_call['args']
+                tool_was_called = True
+                
+                if progress_callback:
+                    progress_callback(f"🔧 Using {name}...")
                 
                 if name == 'search_science_breakthroughs': 
                     res = search_science_breakthroughs.invoke(args)
@@ -552,6 +573,20 @@ def run_discovery(state: EnhancedAgentState, progress_callback=None) -> Enhanced
             conversation.append(response)
             steps += 1
         
+        # If no tools were called, force a search
+        if not tool_was_called:
+            if progress_callback:
+                progress_callback("⚠️ Agent didn't search, forcing search...")
+            
+            # Manually call the search tool
+            search_query = f"recent {topic} breakthroughs scientific discoveries"
+            search_result = search_science_breakthroughs.invoke({"query": search_query})
+            
+            # Create a follow-up prompt
+            force_msg = HumanMessage(content=f"Here are the search results. Now format them into 10-15 breakthroughs:\n\n{search_result}")
+            conversation.append(force_msg)
+            response = discovery_llm.invoke(conversation)
+        
         # Parse candidates from response
         candidates_text = response.content
         
@@ -1032,7 +1067,24 @@ css = """
     overflow-y: auto;
     padding: 20px;
     border-radius: 8px;
-    background: #f8f9fa;
+    background: #ffffff;
+    border: 1px solid #e0e0e0;
+    color: #000000 !important;
+}
+.output-box * {
+    color: #000000 !important;
+}
+.output-box h1, .output-box h2, .output-box h3, .output-box h4 {
+    color: #1a1a1a !important;
+    font-weight: bold;
+}
+.output-box strong {
+    color: #000000 !important;
+    font-weight: 600;
+}
+.output-box a {
+    color: #0066cc !important;
+    text-decoration: underline;
 }
 .button-row {
     display: flex;
@@ -1046,6 +1098,9 @@ css = """
     font-weight: bold;
     margin-bottom: 10px;
 }
+.gradio-container {
+    font-family: 'Inter', system-ui, sans-serif !important;
+}
 """
 
 with gr.Blocks(css=css, title="Tuesday Trivia Agent", theme=gr.themes.Soft()) as demo: