Update app.py

app.py

@@ -66,6 +66,18 @@ def is_apod_query(user_text):
     keywords = ["apod", "image", "picture", "photo", "astronomy picture"]
     return any(keyword in user_text.lower() for keyword in keywords)
 
+def predict_action(user_text):
+    """
+    Determines the type of response for the user's query.
+    If the query contains "NASA" or "space" and also indicates an APOD query,
+    it returns "nasa_info". Otherwise, it returns "general_query", even if "NASA" is present.
+    """
+    user_text_lower = user_text.lower()
+    if "nasa" in user_text_lower or "space" in user_text_lower:
+        if is_apod_query(user_text):
+            return "nasa_info"
+    return "general_query"
+
 def generate_follow_up(user_text):
     """
     Generates two variant follow-up questions and randomly selects one.