Update app.py

app.py

@@ -91,25 +91,30 @@ def astronomy_image_agent(user_input: str):
     """
     Retrieves astronomy-related images from NASA's Image and Video Library or the Astronomy Picture of the Day (APOD) API.
     
-    If the user query contains "galaxy", it fetches galaxy images from NASA's Image and Video Library. 
-    If no specific keyword is found, it defaults to fetching the Astronomy Picture of the Day (APOD).
+    If the user query contains specific keywords (e.g., "galaxy", "jupiter", "mars", etc.), it fetches images
+    from NASA's Image and Video Library using that keyword. If no specific keyword is found, it defaults to fetching 
+    the Astronomy Picture of the Day (APOD).
 
     Args:
         user_input (str): The user input containing the query.
 
     Returns:
-        dict: A dictionary containing details about the requested astronomy image, 
-              including the title, description, and URL. If an error occurs, an error message is returned.
-
-    Raises:
-        HTTPError: If the request to the NASA API fails.
-        ValueError: If no relevant image or data is found.
+        dict: A dictionary containing details about the requested astronomy image, including the title, 
+              description, and URL. If an error occurs, an error message is returned.
     """
-    user_input = user_input.lower()
-    if "galaxy" in user_input:
-        # Query NASA's Image and Video Library API for galaxy images.
+    user_input_lower = user_input.lower()
+    # List of keywords that trigger a search on NASA's Image and Video Library
+    search_keywords = ["galaxy", "jupiter", "mars", "nebula", "saturn", "comet", "asteroid", "moon"]
+    found_keyword = None
+    for keyword in search_keywords:
+        if keyword in user_input_lower:
+            found_keyword = keyword
+            break
+
+    if found_keyword:
+        # Query NASA's Image and Video Library API using the found keyword.
         search_url = "https://images-api.nasa.gov/search"
-        params = {"q": "galaxy", "media_type": "image"}
+        params = {"q": found_keyword, "media_type": "image"}
         try:
             response = requests.get(search_url, params=params)
             response.raise_for_status()
@@ -126,13 +131,13 @@ def astronomy_image_agent(user_input: str):
                     f"Description: {description}\n"
                     f"Image URL: {image_url}"
                 )
-                return {"output": f"✨ Galaxy Image Agent:\n\n{result}"}
+                return {"output": f"✨ {found_keyword.capitalize()} Image Agent:\n\n{result}"}
             else:
-                return {"output": "No galaxy images found."}
+                return {"output": f"No images found for {found_keyword}."}
         except Exception as e:
-            return {"output": f"Galaxy image search failed: {str(e)}"}
+            return {"output": f"{found_keyword.capitalize()} image search failed: {str(e)}"}
     else:
-        # Default to the APOD API if no keyword is detected.
+        # Default to the APOD API if no specific keyword is detected.
         url = f"https://api.nasa.gov/planetary/apod?api_key={os.getenv('NASA_API_KEY')}"
         try:
             response = requests.get(url)
@@ -237,7 +242,6 @@ search = TavilySearchResults(tavily_api_key=os.getenv("TAVILY_API_KEY"))
 tools = [search, iss_location_agent, space_events_agent, astronomy_image_agent, educational_resources_agent]
 llm_with_tools = llm.bind_tools(tools)
 
-# System message
 sys_msg = SystemMessage(content='''This system is designed to provide real-time astronomical data, visualizations, and educational content. Below are the key functions and tools integrated into the system and their specific purposes:
 
 1. **Tavily Search (`search`) Integration**:
@@ -285,7 +289,6 @@ sys_msg = SystemMessage(content='''This system is designed to provide real-time
 
 This setup ensures that the assistant can efficiently address a wide range of user queries related to space, astronomy, and related educational content while keeping interactions intuitive and engaging.
 ''')
-
 # Node
 def assistant(state: MessagesState) -> MessagesState:
     return {"messages": [llm_with_tools.invoke([sys_msg] + state["messages"][-10:])]}
@@ -325,7 +328,7 @@ for message in st.session_state.messages:
             st.write(message["content"])
 
 # User input via chat
-if user_input := st.chat_input("Ask about space news, ISS location, or astronomy images"):
+if user_input := st.chat_input("Ask about space news, ISS location, Educational resource or astronomy images"):
     st.session_state.messages.append({"role": "user", "content": user_input})
     st.chat_message("user").write(user_input)