Update app.py

app.py

@@ -3,6 +3,9 @@ import time
 import requests
 from streamlit.components.v1 import html
 
+# Your Gemini API key for the help agent
+gemini_api_key = "AIzaSyC1IoNgTzAZ52tx11K-2_9F5ruaCptiArE"
+
 # Custom CSS for professional look
 def inject_custom_css():
     st.markdown("""
@@ -103,7 +106,7 @@ def show_confetti():
     </script>
     """)
 
-# Enhanced AI question generation
+# Enhanced AI question generation for guessing game
 def ask_llama(conversation_history, category, is_final_guess=False):
     api_url = "https://api.groq.com/openai/v1/chat/completions"
     headers = {
@@ -147,6 +150,32 @@ def ask_llama(conversation_history, category, is_final_guess=False):
         st.error(f"Error calling Llama API: {str(e)}")
         return "Could not generate question"
 
+# New function for the help AI assistant using Gemini (or any similar model)
+def ask_help_agent(query):
+    gemini_api_url = "https://api.gemini.com/v1/chat/completions"
+    headers = {
+        "Authorization": f"Bearer {gemini_api_key}",
+        "Content-Type": "application/json"
+    }
+    system_prompt = "You're an AI help assistant. Provide concise, friendly advice for clarifying questions in a guessing game. Do not refer to the guessing game details."
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": query}
+    ]
+    data = {
+        "model": "gemini-1",  # or another suitable Gemini model identifier
+        "messages": messages,
+        "temperature": 0.7,
+        "max_tokens": 150
+    }
+    try:
+        response = requests.post(gemini_api_url, headers=headers, json=data)
+        response.raise_for_status()
+        return response.json()["choices"][0]["message"]["content"]
+    except Exception as e:
+        st.error(f"Error calling Gemini API: {str(e)}")
+        return "Could not get help at the moment."
+
 # Main game logic
 def main():
     inject_custom_css()
@@ -162,6 +191,7 @@ def main():
         st.session_state.conversation_history = []
         st.session_state.category = None
         st.session_state.final_guess = None
+        st.session_state.help_conversation = []  # separate history for help agent
 
     # Start screen
     if st.session_state.game_state == "start":
@@ -214,12 +244,12 @@ def main():
                     unsafe_allow_html=True)
         
         with st.form("answer_form"):
-            answer_input = st.text_input("Your answer (yes/no):", 
+            answer_input = st.text_input("Your answer (yes/no/both):", 
                                        key=f"answer_{st.session_state.current_q}").strip().lower()
             
             if st.form_submit_button("Submit"):
-                if answer_input not in ["yes", "no"]:
-                    st.error("Please answer with 'yes' or 'no'!")
+                if answer_input not in ["yes", "no", "both"]:
+                    st.error("Please answer with 'yes', 'no', or 'both'!")
                 else:
                     st.session_state.answers.append(answer_input)
                     st.session_state.conversation_history.append(
@@ -249,6 +279,20 @@ def main():
                     
                     st.rerun()
 
+        # Side Help Option: independent chat with an AI help assistant
+        with st.expander("Need Help? Chat with AI Assistant"):
+            help_query = st.text_input("Enter your help query:", key="help_query")
+            if st.button("Send", key="send_help"):
+                if help_query:
+                    help_response = ask_help_agent(help_query)
+                    st.session_state.help_conversation.append({"query": help_query, "response": help_response})
+                else:
+                    st.error("Please enter a query!")
+            if st.session_state.help_conversation:
+                for msg in st.session_state.help_conversation:
+                    st.markdown(f"**You:** {msg['query']}")
+                    st.markdown(f"**Help Assistant:** {msg['response']}")
+
     # Result screen
     elif st.session_state.game_state == "result":
         if st.session_state.final_guess is None:
@@ -278,4 +322,4 @@ def main():
             st.rerun()
 
 if __name__ == "__main__":
-    main()
+    main()