Update app.py

app.py

@@ -37,23 +37,6 @@ def inject_custom_css():
             color: black;
         }
         
-        .answer-btn {
-            border-radius: 12px !important;
-            padding: 0.5rem 1.5rem !important;
-            font-weight: 600 !important;
-            margin: 0.5rem !important;
-        }
-        
-        .yes-btn {
-            background: #6C63FF !important;
-            color: white !important;
-        }
-        
-        .no-btn {
-            background: #FF6B6B !important;
-            color: white !important;
-        }
-        
         .final-reveal {
             animation: fadeIn 2s;
             font-size: 2.5rem;
@@ -76,10 +59,23 @@ def inject_custom_css():
             pointer-events: none;
             z-index: 1000;
         }
+        
+        .loading {
+            display: inline-block;
+            width: 20px;
+            height: 20px;
+            border: 3px solid rgba(108,99,255,.3);
+            border-radius: 50%;
+            border-top-color: #6C63FF;
+            animation: spin 1s ease-in-out infinite;
+        }
+        
+        @keyframes spin {
+            to { transform: rotate(360deg); }
+        }
     </style>
     """, unsafe_allow_html=True)
 
-# Confetti animation (for final reveal)
 def show_confetti():
     html("""
     <canvas id="confetti-canvas" class="confetti"></canvas>
@@ -87,177 +83,170 @@ 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):
+def ask_llama(messages, category, is_final=False):
     api_url = "https://api.groq.com/openai/v1/chat/completions"
     headers = {
         "Authorization": "Bearer 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",
         "Content-Type": "application/json"
     }
     
-    messages = [
-        {
-            "role": "system",
-            "content": f"You're playing 20 questions to guess a {category}. Ask strategic yes/no questions one at a time based on previous answers."
-        }
-    ] + conversation_history
+    prompt = ("You're playing 20 questions to guess a " + category + 
+             ". Ask strategic yes/no questions one at a time." +
+             (" Based on the answers, what is your final guess? State only the guess." if is_final else ""))
     
     data = {
         "model": "llama-3.3-70b-versatile",
-        "messages": messages,
+        "messages": [{"role": "system", "content": prompt}] + messages,
         "temperature": 0.7,
         "max_tokens": 100
     }
     
     try:
-        response = requests.post(api_url, headers=headers, json=data)
-        response.raise_for_status()
-        return response.json()["choices"][0]["message"]["content"]
+        with st.spinner("Thinking..."):
+            response = requests.post(api_url, headers=headers, json=data, timeout=10)
+            response.raise_for_status()
+            return response.json()["choices"][0]["message"]["content"]
     except Exception as e:
-        st.error(f"Error calling Llama API: {str(e)}")
-        return "Could not generate question"
+        st.error(f"API Error: {str(e)}")
+        return None
 
-# 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)
-    
-    # Initialize session state
-    if 'game_state' not in st.session_state:
-        st.session_state.game_state = "start"
-        st.session_state.questions = []
-        st.session_state.current_q = 0
-        st.session_state.answers = []
-        st.session_state.conversation_history = []
-        st.session_state.category = None
-        st.session_state.waiting_for_answer = True
-        st.session_state.question_generated = False
 
-    # Start screen
-    if st.session_state.game_state == "start":
+    # Initialize game state
+    if 'game' not in st.session_state:
+        st.session_state.game = {
+            'state': 'start',
+            'category': None,
+            'questions': [],
+            'answers': [],
+            'conversation': []
+        }
+
+    # 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 with yes/no questions!</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)</li>
+                <li><strong>place</strong> (city, country, location)</li>
+                <li><strong>object</strong> (something you can touch)</li>
             </ul>
-            <p><strong>Type one of these categories below</strong> to begin: <code>person</code>, <code>place</code>, or <code>object</code>.</p>
         </div>
         """, unsafe_allow_html=True)
 
-        with st.form("start_form"):
-            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!")
-                elif category_input not in ["person", "place", "object"]:
-                    st.error("Please enter either 'person', 'place', or 'object'!")
-                else:
-                    st.session_state.category = category_input
-                    st.session_state.game_state = "gameplay"
-                    st.rerun()
+        col1, col2, col3 = st.columns(3)
+        with col1:
+            if st.button("Person", use_container_width=True):
+                st.session_state.game['category'] = 'person'
+                st.session_state.game['state'] = 'playing'
+                st.rerun()
+        with col2:
+            if st.button("Place", use_container_width=True):
+                st.session_state.game['category'] = 'place'
+                st.session_state.game['state'] = 'playing'
+                st.rerun()
+        with col3:
+            if st.button("Object", use_container_width=True):
+                st.session_state.game['category'] = 'object'
+                st.session_state.game['state'] = 'playing'
+                st.rerun()
 
-    # Gameplay screen - generate first question if needed
-    elif st.session_state.game_state == "gameplay":
-        if not st.session_state.questions:
-            # Generate first question
-            first_question = ask_llama([{
-                "role": "user", 
-                "content": "Ask your first yes/no question."
-            }], st.session_state.category)
-            st.session_state.questions = [first_question]
-            st.session_state.conversation_history = [
-                {"role": "assistant", "content": first_question}
-            ]
-            st.session_state.waiting_for_answer = True
-            st.rerun()
+    # Game Play Screen
+    elif st.session_state.game['state'] == 'playing':
+        # Generate first question if none exists
+        if not st.session_state.game['questions']:
+            question = ask_llama(
+                [{"role": "user", "content": "Ask your first yes/no question."}],
+                st.session_state.game['category']
+            )
+            if question:
+                st.session_state.game['questions'].append(question)
+                st.session_state.game['conversation'].append({"role": "assistant", "content": question})
+                st.rerun()
+            else:
+                st.error("Failed to generate question. Please try again.")
+                st.session_state.game['state'] = 'start'
+                st.rerun()
 
         # Show current question
-        current_question = st.session_state.questions[st.session_state.current_q]
-        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)
-        
+        current_q = len(st.session_state.game['questions']) - 1
+        st.markdown(f"""
+        <div class="question-box">
+            Question {current_q + 1}:<br><br>
+            <strong>{st.session_state.game['questions'][current_q]}</strong>
+        </div>
+        """, unsafe_allow_html=True)
+
         # Answer form
         with st.form("answer_form"):
-            answer_input = st.text_input("Your answer (yes/no):").strip().lower()
-
+            answer = st.radio("Your answer:", ["Yes", "No"], horizontal=True)
             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}
-                    )
-                    st.session_state.waiting_for_answer = False
-                    
-                    # Check if we've reached max questions
-                    if st.session_state.current_q >= 19:
-                        st.session_state.game_state = "result"
-                    st.rerun()
+                # Record answer
+                answer_lower = answer.lower()
+                st.session_state.game['answers'].append(answer_lower)
+                st.session_state.game['conversation'].append({"role": "user", "content": answer_lower})
 
-        # Generate next question if answer was submitted
-        if not st.session_state.waiting_for_answer and st.session_state.current_q < 19:
-            # Generate next question based on all previous Q&A
-            next_question = ask_llama(
-                st.session_state.conversation_history,
-                st.session_state.category
-            )
-            st.session_state.questions.append(next_question)
-            st.session_state.conversation_history.append(
-                {"role": "assistant", "content": next_question}
-            )
-            st.session_state.current_q += 1
-            st.session_state.waiting_for_answer = True
-            st.rerun()
+                # Check if AI can guess now
+                if current_q >= 4:  # Minimum 5 questions before checking
+                    ready = ask_llama(
+                        st.session_state.game['conversation'] + [
+                            {"role": "user", "content": "Can you guess now? Answer only 'yes' or 'no'."}
+                        ],
+                        st.session_state.game['category']
+                    )
+                    if ready and ready.strip().lower() == 'yes':
+                        st.session_state.game['state'] = 'result'
+                        st.rerun()
 
-        # Early guess button
-        if st.button("I think I can guess now!"):
-            st.session_state.game_state = "result"
-            st.rerun()
+                # Generate next question if under 20
+                if current_q < 19:
+                    next_q = ask_llama(
+                        st.session_state.game['conversation'],
+                        st.session_state.game['category']
+                    )
+                    if next_q:
+                        st.session_state.game['questions'].append(next_q)
+                        st.session_state.game['conversation'].append({"role": "assistant", "content": next_q})
+                        st.rerun()
+                    else:
+                        st.error("Failed to generate next question")
+                else:
+                    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
+    # Result Screen
+    elif st.session_state.game['state'] == 'result':
+        guess = ask_llama(
+            st.session_state.game['conversation'],
+            st.session_state.game['category'],
+            is_final=True
         )
         
-        show_confetti()
-        st.markdown('<div class="final-reveal">🎉 I think it\'s...</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>', 
-                    unsafe_allow_html=True)
-        
-        st.markdown("<br><br>", unsafe_allow_html=True)
+        if guess:
+            show_confetti()
+            st.markdown('<div class="final-reveal">🎉 I think it\'s...</div>', unsafe_allow_html=True)
+            time.sleep(1)
+            st.markdown(f'<div class="final-reveal" style="font-size:3.5rem;color:#6C63FF;">{guess}</div>', 
+                        unsafe_allow_html=True)
+        else:
+            st.error("Sorry, I couldn't make a guess. Let's try again!")
         
-        # Play again button
-        if st.button("Play Again", key="play_again"):
+        if st.button("Play Again", type="primary"):
             st.session_state.clear()
             st.rerun()