Update app.py

app.py

@@ -42,19 +42,24 @@ def query_gemini_api(user_answers):
         # Send the request to the Gemini API
         response = requests.post(url, headers=headers, json=payload)
 
-        # Debugging: Print the raw response to check its structure
-        print("API Response:", response.json())
-
+        # Wait for the response and process it
         if response.status_code == 200:
             result = response.json()
-            # Return the result from Gemini API (mood and recommendations)
-            return result
+
+            # Check if the response contains valid mood and recommendations
+            mood = result.get("mood", None)
+            recommendations = result.get("recommendations", None)
+
+            if mood and recommendations:
+                return mood, recommendations
+            else:
+                return None, None
         else:
             print(f"Error: {response.status_code} - {response.text}")
-            return None
+            return None, None
     except requests.exceptions.RequestException as e:
         print(f"An error occurred: {e}")
-        return None
+        return None, None
 
 # Streamlit app for collecting answers
 def main():
@@ -72,27 +77,21 @@ def main():
     # If all 3 responses are collected, send them to Gemini for analysis
     if len(responses) == len(questions):
         st.write("Processing your answers...")
-        
+
         # Get mood and recommendations from Gemini API
-        result = query_gemini_api(responses)
-        
-        if result:
-            # Assuming the response contains mood and recommendations
-            mood = result.get("mood", "Unknown")
-            recommendations = result.get("recommendations", [])
+        mood, recommendations = query_gemini_api(responses)
 
+        if mood and recommendations:
             # Display the detected mood
             st.write(f"Detected Mood: {mood}")
 
             # Display the recommendations
-            if recommendations:
-                st.write("### Recommendations to Improve Your Mood:")
-                for recommendation in recommendations:
-                    st.write(f"- {recommendation}")
-            else:
-                st.write("No recommendations available.")
+            st.write("### Recommendations to Improve Your Mood:")
+            for recommendation in recommendations:
+                st.write(f"- {recommendation}")
         else:
-            st.write("Sorry, we couldn't process your answers. Please try again.")
+            # If no valid mood or recommendations are found, do nothing
+            pass
     else:
         st.write("Please answer all 3 questions to receive suggestions.")