Update context handling: provide weather/space weather only when relevant to query

app.py

@@ -149,6 +149,7 @@ async def research_assistant(query, history):
     history[-1] = (query, "🔍 Gathering context...")
     yield history
     
+    # Get weather and space weather context (but don't include in prompt yet)
     weather_task = asyncio.create_task(add_weather_context())
     space_weather_task = asyncio.create_task(add_space_weather_context())
     search_task = asyncio.create_task(asyncio.to_thread(perform_search, validated_query))
@@ -173,7 +174,22 @@ async def research_assistant(query, history):
         elif result.get("type") == "source":
             search_content += f"Source: {result['content']}\n\n"
 
-    enriched_input = f"{validated_query}\n\n{answer_content}Weather: {weather_data}\nSpace Weather: {space_weather_data}\n\nSearch Results:\n{search_content}"
+    # Only include context if it seems relevant to the query
+    context_section = ""
+    lower_query = validated_query.lower()
+    
+    # Check if weather might be relevant
+    weather_keywords = ["weather", "temperature", "climate", "rain", "snow", "sun", "storm", "wind", "humidity"]
+    if any(keyword in lower_query for keyword in weather_keywords):
+        context_section += f"\nCurrent Weather Context: {weather_data}"
+    
+    # Check if space weather might be relevant
+    space_keywords = ["space", "solar", "sun", "satellite", "astronomy", "cosmic", "radiation", "flare"]
+    if any(keyword in lower_query for keyword in space_keywords):
+        context_section += f"\nSpace Weather Context: {space_weather_data}"
+
+    # Build the enriched input with context only when needed
+    enriched_input = f"{validated_query}\n\n{answer_content}Search Results:\n{search_content}{context_section}"
 
     server_status = server_monitor.check_server_status()
     if not server_status["available"]:
@@ -315,7 +331,7 @@ with gr.Blocks(
     
     gr.Markdown("# 🧠 AI Research Assistant")
     gr.Markdown("This advanced AI assistant combines web search with contextual awareness to answer complex questions. "
-                "It incorporates current weather and space weather data for richer context.")
+                "It provides weather and space weather context only when relevant to your query.")
     
     with gr.Tabs():
         with gr.TabItem("💬 Chat"):
@@ -334,8 +350,8 @@ with gr.Blocks(
                     
                     ## Features
                     - 🔍 Web search integration
-                    - 🌤️ Weather context
-                    - 🌌 Space weather context
+                    - 🌤️ Context-aware weather data (only when relevant)
+                    - 🌌 Context-aware space weather data (only when relevant)
                     - 📚 Real-time citations
                     - ⚡ Streaming output
                     """)