Update app.py

app.py

@@ -1,18 +1,19 @@
 import streamlit as st
 import time
 import requests
+import json
 from streamlit.components.v1 import html
 
-# Inject custom CSS
+# Custom CSS for professional look
 def inject_custom_css():
     st.markdown("""
     <style>
         @import url('https://fonts.googleapis.com/css2?family=Poppins:wght@400;600;700&display=swap');
-        
+
         * {
             font-family: 'Poppins', sans-serif;
         }
-        
+
         .title {
             font-size: 3rem !important;
             font-weight: 700 !important;
@@ -20,14 +21,14 @@ def inject_custom_css():
             text-align: center;
             margin-bottom: 0.5rem;
         }
-        
+
         .subtitle {
             font-size: 1.2rem !important;
             text-align: center;
             color: #666 !important;
             margin-bottom: 2rem;
         }
-        
+
         .question-box {
             background: #F8F9FA;
             border-radius: 15px;
@@ -36,7 +37,24 @@ def inject_custom_css():
             box-shadow: 0 4px 6px rgba(0,0,0,0.1);
             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;
@@ -44,12 +62,12 @@ def inject_custom_css():
             text-align: center;
             margin: 2rem 0;
         }
-        
+
         @keyframes fadeIn {
             from { opacity: 0; }
             to { opacity: 1; }
         }
-        
+
         .confetti {
             position: fixed;
             top: 0;
@@ -62,7 +80,7 @@ def inject_custom_css():
     </style>
     """, unsafe_allow_html=True)
 
-# Show confetti effect
+# Confetti animation (for final reveal)
 def show_confetti():
     html("""
     <canvas id="confetti-canvas" class="confetti"></canvas>
@@ -70,159 +88,156 @@ 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>
     """)
 
-# Call Groq API (LLama model)
-def ask_llama(messages, category, is_final=False):
-    api_url = "https://api.groq.com/openai/v1/chat/completions"
+# Groq Llama AI API call
+def ask_llama(conversation_history, category):
+    api_url = "https://api.groq.com/openai/v1/chat/completions"  # Updated API URL
     headers = {
-        "Authorization": "Bearer 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",
+        "Authorization": "Bearer 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",  # Your new API key
         "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 ""))
+
+    messages = [
+        {
+            "role": "system",
+            "content": f"You're playing 20 questions to guess a {category}. Ask strategic yes/no questions."
+        }
+    ] + conversation_history
 
     data = {
-        "model": "llama-3-70b-8192",
-        "messages": [{"role": "system", "content": prompt}] + messages,
+        "model": "llama-3.3-70b-versatile",  # The model you are using
+        "messages": messages,
         "temperature": 0.7,
         "max_tokens": 100
     }
 
     try:
-        with st.spinner("Thinking..."):
-            response = requests.post(api_url, headers=headers, json=data, timeout=15)
-            response.raise_for_status()
-            return response.json()["choices"][0]["message"]["content"]
+        response = requests.post(api_url, headers=headers, json=data)
+        response.raise_for_status()
+        return response.json()["choices"][0]["message"]["content"]
     except Exception as e:
-        st.error(f"API Error: {str(e)}")
-        return None
+        st.error(f"Error calling Llama API: {str(e)}")
+        return "Could not generate question"
 
-# Main Game Logic
+# 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)
 
-    # Session State
-    if 'game' not in st.session_state:
-        st.session_state.game = {
-            'state': 'start',
-            'category': None,
-            'questions': [],
-            'conversation': [],
-            'input_buffer': ""
-        }
-
-    game = st.session_state.game
+    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
 
     # Start screen
-    if game['state'] == 'start':
+    if st.session_state.game_state == "start":
         st.markdown("""
         <div class="question-box">
             <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>
+            <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>
             <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>
+                <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>
             </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)
 
-        col1, col2, col3 = st.columns(3)
-        with col1:
-            if st.button("Person", use_container_width=True):
-                game['category'] = 'person'
-                game['state'] = 'playing'
-                st.rerun()
-        with col2:
-            if st.button("Place", use_container_width=True):
-                game['category'] = 'place'
-                game['state'] = 'playing'
-                st.rerun()
-        with col3:
-            if st.button("Object", use_container_width=True):
-                game['category'] = 'object'
-                game['state'] = 'playing'
-                st.rerun()
-
-    # Game play screen
-    elif game['state'] == 'playing':
-        if not game['questions']:
-            first_q = ask_llama([{"role": "user", "content": "Ask your first yes/no question."}], game['category'])
-            if first_q:
-                game['questions'].append(first_q)
-                game['conversation'].append({"role": "assistant", "content": first_q})
-                st.rerun()
-            else:
-                st.error("Failed to generate question.")
-                game['state'] = 'start'
-                st.rerun()
-
-        current_q_index = len(game['questions']) - 1
-        current_q = game['questions'][current_q_index]
-
-        st.markdown(f"""
-        <div class="question-box">
-            Question {current_q_index + 1}:<br><br>
-            <strong>{current_q}</strong>
-        </div>
-        """, unsafe_allow_html=True)
+        with st.form("start_form"):
+            category_input = st.text_input("Enter category (person / place / object):").strip().lower()
 
-        with st.form("answer_form"):
-            user_input = st.text_input("Your answer (Yes, No, or more detailed response):")
-            submitted = st.form_submit_button("Submit")
+            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
+                    # Generate first question
+                    first_question = ask_llama([{
+                        "role": "user",
+                        "content": "Ask your first yes/no question."
+                    }], category_input)
+                    st.session_state.questions = [first_question]
+                    st.session_state.conversation_history = [
+                        {"role": "assistant", "content": first_question}
+                    ]
+                    st.session_state.game_state = "gameplay"
+                    st.rerun()
 
-            if submitted and user_input.strip() != "":
-                game['conversation'].append({"role": "user", "content": user_input.strip()})
+    # Gameplay screen
+    elif st.session_state.game_state == "gameplay":
+        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)
 
-                # Ask if ready to guess after 5 questions
-                if current_q_index >= 4:
-                    can_guess = ask_llama(
-                        game['conversation'] + [{"role": "user", "content": "Can you guess now? Answer only yes or no."}],
-                        game['category']
-                    )
-                    if can_guess and can_guess.strip().lower() == 'yes':
-                        game['state'] = 'result'
-                        st.rerun()
-
-                # Generate next question
-                if current_q_index < 19:
-                    next_q = ask_llama(game['conversation'], game['category'])
-                    if next_q:
-                        game['questions'].append(next_q)
-                        game['conversation'].append({"role": "assistant", "content": next_q})
-                        st.rerun()
-                    else:
-                        st.error("Couldn't get next question.")
+        with st.form("answer_form"):
+            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:
-                    game['state'] = 'result'
-                    st.rerun()
+                    # Record answer
+                    st.session_state.answers.append(answer_input)
+                    st.session_state.conversation_history.append(
+                        {"role": "user", "content": answer_input}
+                    )
 
-    # Result Screen
-    elif game['state'] == 'result':
-        final_guess = ask_llama(game['conversation'], game['category'], is_final=True)
+                    # Check if we've reached max questions
+                    if st.session_state.current_q >= 19:  # 0-based index
+                        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.conversation_history.append(
+                            {"role": "assistant", "content": next_question}
+                        )
+                        st.session_state.current_q += 1
 
-        if final_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;">{final_guess}</div>', unsafe_allow_html=True)
-        else:
-            st.error("Sorry, I couldn't make a guess.")
+                    st.rerun()
 
-        if st.button("Play Again", type="primary"):
+    # 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
+        )
+
+        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)
+
+        if st.button("Play Again", key="play_again"):
             st.session_state.clear()
             st.rerun()