Update app.py

app.py

@@ -59,20 +59,6 @@ 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)
 
@@ -99,9 +85,9 @@ def ask_llama(messages, category, is_final=False):
         "Content-Type": "application/json"
     }
     
-    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 ""))
+    prompt = (f"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",
@@ -129,9 +115,10 @@ def main():
         st.session_state.game = {
             'state': 'start',
             'category': None,
-            'questions': [],
-            'answers': [],
-            'conversation': []
+            'current_question': None,
+            'conversation': [],
+            'question_count': 0,
+            'waiting_for_answer': False
         }
 
     # Start Screen
@@ -141,11 +128,6 @@ def main():
             <h3>Welcome to <span style='color:#6C63FF;'>KASOTI 🎯</span></h3>
             <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> (celebrity, fictional character)</li>
-                <li><strong>place</strong> (city, country, location)</li>
-                <li><strong>object</strong> (something you can touch)</li>
-            </ul>
         </div>
         """, unsafe_allow_html=True)
 
@@ -153,56 +135,60 @@ def main():
         with col1:
             if st.button("Person", use_container_width=True):
                 st.session_state.game['category'] = 'person'
-                st.session_state.game['state'] = 'playing'
+                st.session_state.game['state'] = 'generate_question'
                 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.session_state.game['state'] = 'generate_question'
                 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.session_state.game['state'] = 'generate_question'
                 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()
+    # Generate Question State
+    elif st.session_state.game['state'] == 'generate_question':
+        # Generate new question
+        question = ask_llama(
+            st.session_state.game['conversation'],
+            st.session_state.game['category']
+        )
+        
+        if question:
+            st.session_state.game['current_question'] = question
+            st.session_state.game['conversation'].append({"role": "assistant", "content": question})
+            st.session_state.game['question_count'] += 1
+            st.session_state.game['state'] = 'wait_for_answer'
+            st.session_state.game['waiting_for_answer'] = True
+            st.rerun()
+        else:
+            st.error("Failed to generate question. Please try again.")
+            st.session_state.game['state'] = 'start'
+            st.rerun()
 
-        # Show current question
-        current_q = len(st.session_state.game['questions']) - 1
+    # Wait for Answer State
+    elif st.session_state.game['state'] == 'wait_for_answer':
         st.markdown(f"""
         <div class="question-box">
-            Question {current_q + 1}:<br><br>
-            <strong>{st.session_state.game['questions'][current_q]}</strong>
+            Question {st.session_state.game['question_count']}:<br><br>
+            <strong>{st.session_state.game['current_question']}</strong>
         </div>
         """, unsafe_allow_html=True)
 
-        # Answer form
         with st.form("answer_form"):
-            answer = st.radio("Your answer:", ["Yes", "No"], horizontal=True)
-            if st.form_submit_button("Submit"):
+            answer = st.radio("Your answer:", ["Yes", "No"], index=None, horizontal=True)
+            submitted = st.form_submit_button("Submit")
+            
+            if submitted and answer:
                 # 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})
-
-                # Check if AI can guess now
-                if current_q >= 4:  # Minimum 5 questions before checking
+                st.session_state.game['waiting_for_answer'] = False
+                
+                # Check if we should guess or ask another question
+                if st.session_state.game['question_count'] >= 5:  # 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'."}
@@ -212,19 +198,11 @@ def main():
                     if ready and ready.strip().lower() == 'yes':
                         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")
+                
+                # Continue with next question if under 20
+                if st.session_state.game['question_count'] < 20:
+                    st.session_state.game['state'] = 'generate_question'
+                    st.rerun()
                 else:
                     st.session_state.game['state'] = 'result'
                     st.rerun()