Update app.py

app.py

@@ -7,7 +7,7 @@ from streamlit.components.v1 import html
 @st.cache_resource
 def get_help_agent():
     from transformers import pipeline
-    # Using BlenderBot 400M Distill as the public conversational model
+    # Using BlenderBot 400M Distill as the public conversational model (used elsewhere)
     return pipeline("conversational", model="facebook/blenderbot-400M-distill")
 
 # Custom CSS for professional look (fixed text color)
@@ -119,16 +119,13 @@ def ask_llama(conversation_history, category, is_final_guess=False):
     }
 
     system_prompt = f"""You're playing 20 questions to guess a {category}. Follow these rules:
-1. Ask strategic, non-repeating yes/no questions that narrow down possibilities.
-2. Consider all previous answers carefully before asking the next question.
-3. If you're very confident (80%+ sure), respond with "Final Guess: [your guess]".
-4. For places: ask about continent, climate, famous landmarks, country, city or population.
-5. For people: ask about fictional or real, profession, gender, alive/dead, nationality, or fame.
-6. For objects: ask about size, color, usage, material, or where it's found.
-7. Never repeat questions and always make progress toward guessing.
-8. If the guess is generic (e.g. "bed") and the player confirms it with "yes", end the game.
-9. If the guess is generic but the answer is "no", ask additional questions to refine the guess.
-10. After a rejected final guess, continue asking questions instead of making new guesses immediately."""
+1. Ask strategic, non-repeating yes/no questions that narrow down possibilities
+2. Consider all previous answers carefully before asking next question
+3. If you're very confident (80%+ sure), respond with "Final Guess: [your guess]"
+4. For places: ask about continent, climate, famous landmarks, country, city or population
+5. For people: ask about fictional or real, profession, gender, alive/dead, nationality, or fame
+6. For objects: ask about size, color, usage, material, or where it's found
+7. Never repeat questions and always make progress toward guessing"""
 
     if is_final_guess:
         prompt = f"""Based on these answers about a {category}, provide ONLY your final guess with no extra text:
@@ -157,15 +154,20 @@ 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 a Hugging Face chatbot model
+# New function for the help AI assistant using a free Hugging Face conversational model
+# (replacing the original BlenderBot approach with DialoGPT for simplicity)
 def ask_help_agent(query):
-    from transformers import Conversation
-    # Get the cached help agent (BlenderBot)
-    help_agent = get_help_agent()
-    conversation = Conversation(query)
-    result = help_agent(conversation)
-    # The generated response is stored in generated_responses list
-    return result.generated_responses[-1]
+    try:
+        from transformers import pipeline
+        # Using DialoGPT-medium as a free conversational text-generation model
+        help_chat = pipeline("text-generation", model="microsoft/DialoGPT-medium")
+        # The conversation string can incorporate previous messages if you want a dialogue
+        # Here we simply generate a response to the current query.
+        response = help_chat(query, max_length=100, do_sample=True, top_p=0.95)
+        # Trim the response to show only the generated text
+        return response[0]['generated_text']
+    except Exception as e:
+        return f"Error in help agent: {str(e)}"
 
 # Main game logic
 def main():
@@ -217,18 +219,17 @@ def main():
                         {"role": "assistant", "content": first_question}
                     ]
                     st.session_state.game_state = "gameplay"
-                    st.session_state.current_q = 0  # start at question 0
-                    st.rerun()
+                    st.experimental_rerun()
 
     # Gameplay screen
     elif st.session_state.game_state == "gameplay":
         current_question = st.session_state.questions[st.session_state.current_q]
         
-        # If the AI made a final guess, switch to confirmation
+        # Check if AI made a guess
         if "Final Guess:" in current_question:
             st.session_state.final_guess = current_question.split("Final Guess:")[1].strip()
             st.session_state.game_state = "confirm_guess"
-            st.rerun()
+            st.experimental_rerun()
         
         st.markdown(f'<div class="question-box">Question {st.session_state.current_q + 1}/20:<br><br>'
                     f'<strong>{current_question}</strong></div>',
@@ -246,12 +247,13 @@ def main():
                         {"role": "user", "content": answer_input}
                     )
 
-                    # Generate next response from Llama
+                    # Generate next response
                     next_response = ask_llama(
                         st.session_state.conversation_history,
                         st.session_state.category
                     )
                     
+                    # Check if AI made a guess
                     if "Final Guess:" in next_response:
                         st.session_state.final_guess = next_response.split("Final Guess:")[1].strip()
                         st.session_state.game_state = "confirm_guess"
@@ -263,12 +265,12 @@ def main():
                         st.session_state.current_q += 1
 
                         # Stop after 20 questions max
-                        if st.session_state.current_q >= 19:
+                        if st.session_state.current_q >= 20:
                             st.session_state.game_state = "result"
                     
-                    st.rerun()
+                    st.experimental_rerun()
 
-        # Side Help Option: independent chat with an AI help assistant (Hugging Face model)
+        # Side Help Option: independent chat with an AI help assistant (free Hugging Face model)
         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"):
@@ -282,7 +284,7 @@ def main():
                     st.markdown(f"**You:** {msg['query']}")
                     st.markdown(f"**Help Assistant:** {msg['response']}")
 
-    # Confirm guess screen
+    # Guess confirmation screen using text input response
     elif st.session_state.game_state == "confirm_guess":
         st.markdown(f'<div class="question-box">🤖 My Final Guess:<br><br>'
                     f'<strong>Is it {st.session_state.final_guess}?</strong></div>',
@@ -296,9 +298,10 @@ def main():
                 else:
                     if confirm_input == "yes":
                         st.session_state.game_state = "result"
-                        st.rerun()
+                        st.experimental_rerun()
+                        st.stop()  # Immediately halt further execution
                     else:
-                        # Add the negative response to history and continue gameplay
+                        # Add negative response to history and continue gameplay
                         st.session_state.conversation_history.append(
                             {"role": "user", "content": "no"}
                         )
@@ -312,9 +315,7 @@ def main():
                             {"role": "assistant", "content": next_response}
                         )
                         st.session_state.current_q += 1
-                        if st.session_state.current_q >= 19:
-                            st.session_state.game_state = "result"
-                        st.rerun()
+                        st.experimental_rerun()
 
     # Result screen
     elif st.session_state.game_state == "result":
@@ -337,12 +338,12 @@ def main():
         time.sleep(1)
         st.markdown(f'<div class="final-reveal" style="font-size:3.5rem;color:#6C63FF;">{st.session_state.final_guess}</div>',
                     unsafe_allow_html=True)
-        st.markdown(f"<p style='text-align:center'>Guessed in {st.session_state.current_q + 1} questions</p>", 
+        st.markdown(f"<p style='text-align:center'>Guessed in {len(st.session_state.questions)} questions</p>", 
                     unsafe_allow_html=True)
         
         if st.button("Play Again", key="play_again"):
             st.session_state.clear()
-            st.rerun()
+            st.experimental_rerun()
 
 if __name__ == "__main__":
-    main()
+    main()