Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 import time
 import requests
-import json
 from streamlit.components.v1 import html
 
 # Custom CSS for professional look
@@ -77,10 +76,17 @@ def inject_custom_css():
             pointer-events: none;
             z-index: 1000;
         }
+        
+        .confidence-meter {
+            height: 10px;
+            background: linear-gradient(90deg, #FF6B6B 0%, #6C63FF 100%);
+            border-radius: 5px;
+            margin: 10px 0;
+        }
     </style>
     """, unsafe_allow_html=True)
 
-# Confetti animation (for final reveal)
+# Confetti animation
 def show_confetti():
     html("""
     <canvas id="confetti-canvas" class="confetti"></canvas>
@@ -88,38 +94,48 @@ def show_confetti():
     <script>
     const canvas = document.getElementById('confetti-canvas');
     const confetti = confetti.create(canvas, { resize: true });
-
     confetti({
         particleCount: 150,
         spread: 70,
         origin: { y: 0.6 }
     });
-
-    setTimeout(() => {
-        canvas.remove();
-    }, 5000);
+    setTimeout(() => { canvas.remove(); }, 5000);
     </script>
     """)
 
-# Groq Llama AI API call
-def ask_llama(conversation_history, category):
-    api_url = "https://api.groq.com/openai/v1/chat/completions"  # Updated API URL
+# Enhanced AI question generation
+def ask_llama(conversation_history, category, is_final_guess=False):
+    api_url = "https://api.groq.com/openai/v1/chat/completions"
     headers = {
-        "Authorization": "Bearer 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",  # Your new API key
+        "Authorization": "Bearer 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",
         "Content-Type": "application/json"
     }
 
+    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 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, or population
+5. For people: ask about 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:
+{conversation_history}"""
+    else:
+        prompt = "Ask your next strategic yes/no question that will best narrow down the possibilities."
+
     messages = [
-        {
-            "role": "system",
-            "content": f"You're playing 20 questions to guess a {category}. Ask strategic yes/no questions."
-        }
-    ] + conversation_history
+        {"role": "system", "content": system_prompt},
+        *conversation_history,
+        {"role": "user", "content": prompt}
+    ]
 
     data = {
-        "model": "llama-3.3-70b-versatile",  # The model you are using
+        "model": "llama-3.3-70b-versatile",
         "messages": messages,
-        "temperature": 0.7,
+        "temperature": 0.7 if is_final_guess else 0.8,
         "max_tokens": 100
     }
 
@@ -131,12 +147,12 @@ def ask_llama(conversation_history, category):
         st.error(f"Error calling Llama API: {str(e)}")
         return "Could not generate question"
 
-# Game logic
+# Main game logic
 def main():
     inject_custom_css()
 
     st.markdown('<div class="title">KASOTI</div>', unsafe_allow_html=True)
-    st.markdown('<div class="subtitle">The Ultimate Guessing Game</div>', unsafe_allow_html=True)
+    st.markdown('<div class="subtitle">The Smart Guessing Game</div>', unsafe_allow_html=True)
 
     if 'game_state' not in st.session_state:
         st.session_state.game_state = "start"
@@ -145,26 +161,27 @@ def main():
         st.session_state.answers = []
         st.session_state.conversation_history = []
         st.session_state.category = None
+        st.session_state.final_guess = None
 
     # Start screen
     if st.session_state.game_state == "start":
         st.markdown("""
         <div class="question-box">
             <h3>Welcome to <span style='color:#6C63FF;'>KASOTI 🎯</span></h3>
-            <p>This is a fun guessing game! You'll think of something, and I'll ask maximum 20 Yes/No questions to guess what it is.</p>
-            <p>You can choose from three categories:</p>
+            <p>Think of something and I'll try to guess it in 20 questions or less!</p>
+            <p>Choose a category:</p>
             <ul>
-                <li><strong>person</strong> – like a celebrity, fictional character, etc.</li>
-                <li><strong>place</strong> – like a city, country, or location</li>
-                <li><strong>object</strong> – like something you can touch or use</li>
+                <li><strong>Person</strong> - celebrity, fictional character, historical figure</li>
+                <li><strong>Place</strong> - city, country, landmark, geographical location</li>
+                <li><strong>Object</strong> - everyday item, tool, vehicle, etc.</li>
             </ul>
-            <p><strong>Type one of these categories below</strong> to begin: <code>person</code>, <code>place</code>, or <code>object</code>.</p>
+            <p>Type your category below to begin:</p>
         </div>
         """, unsafe_allow_html=True)
 
         with st.form("start_form"):
-            category_input = st.text_input("Enter category (person / place / object):").strip().lower()
-
+            category_input = st.text_input("Enter category (person/place/object):").strip().lower()
+            
             if st.form_submit_button("Start Game"):
                 if not category_input:
                     st.error("Please enter a category!")
@@ -172,11 +189,9 @@ def main():
                     st.error("Please enter either 'person', 'place', or 'object'!")
                 else:
                     st.session_state.category = category_input
-                    # Generate first question
-                    first_question = ask_llama([{
-                        "role": "user",
-                        "content": "Ask your first yes/no question."
-                    }], category_input)
+                    first_question = ask_llama([
+                        {"role": "user", "content": "Ask your first strategic yes/no question."}
+                    ], category_input)
                     st.session_state.questions = [first_question]
                     st.session_state.conversation_history = [
                         {"role": "assistant", "content": first_question}
@@ -187,59 +202,80 @@ def main():
     # Gameplay screen
     elif st.session_state.game_state == "gameplay":
         current_question = st.session_state.questions[st.session_state.current_q]
+        
+        # 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 = "result"
+            st.rerun()
+        
         st.markdown(f'<div class="question-box">Question {st.session_state.current_q + 1}/20:<br><br>'
                     f'<strong>{current_question}</strong></div>',
                     unsafe_allow_html=True)
-
+        
         with st.form("answer_form"):
-            answer_input = st.text_input("Your answer (yes/no):", key=f"answer_{st.session_state.current_q}").strip().lower()
-
+            answer_input = st.text_input("Your answer (yes/no):", 
+                                       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'!")
                 else:
-                    # Record answer
                     st.session_state.answers.append(answer_input)
                     st.session_state.conversation_history.append(
                         {"role": "user", "content": answer_input}
                     )
 
-                    # Check if we've reached max questions
-                    if st.session_state.current_q >= 19:  # 0-based index
+                    # 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 = "result"
                     else:
-                        # Generate next question
-                        next_question = ask_llama(
-                            st.session_state.conversation_history,
-                            st.session_state.category
-                        )
-                        st.session_state.questions.append(next_question)
+                        st.session_state.questions.append(next_response)
                         st.session_state.conversation_history.append(
-                            {"role": "assistant", "content": next_question}
+                            {"role": "assistant", "content": next_response}
                         )
                         st.session_state.current_q += 1
 
+                        # Stop after 8 questions if no guess yet
+                        if st.session_state.current_q >= 8:
+                            st.session_state.game_state = "result"
+                    
                     st.rerun()
 
     # Result screen
     elif st.session_state.game_state == "result":
-        # Generate final guess
-        final_guess = ask_llama(
-            st.session_state.conversation_history + [
-                {"role": "user", "content": "Based on all my answers, what is your final guess? Just state the guess directly."}
-            ],
-            st.session_state.category
-        )
+        if st.session_state.final_guess is None:
+            # Generate final guess if not already made
+            qa_history = "\n".join(
+                [f"Q{i+1}: {q}\nA: {a}" 
+                 for i, (q, a) in enumerate(zip(st.session_state.questions, st.session_state.answers))]
+            )
+            
+            st.session_state.final_guess = ask_llama(
+                [{"role": "user", "content": qa_history}],
+                st.session_state.category,
+                is_final_guess=True
+            )
 
         show_confetti()
-        st.markdown('<div class="final-reveal">🎉 I think it\'s...</div>', unsafe_allow_html=True)
+        st.markdown('<div class="final-reveal">🎉 My guess is...</div>', unsafe_allow_html=True)
         time.sleep(1)
-        st.markdown(f'<div class="final-reveal" style="font-size:3.5rem;color:#6C63FF;">{final_guess}</div>',
+        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 {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()
 
 if __name__ == "__main__":
-    main()
+    main()