Update app.py

app.py

@@ -1,199 +1,180 @@
 import streamlit as st
-from transformers import pipeline
-import streamlit as st
-from transformers import pipeline
-import os
-
-# Disable parallelism to avoid warnings
-os.environ["TOKENIZERS_PARALLELISM"] = "false"
-
-# Initialize the game
-def init_game():
-    if 'questions_asked' not in st.session_state:
-        st.session_state.questions_asked = 0
-        st.session_state.answers = []
-        st.session_state.game_over = False
-        st.session_state.current_question = "Is it a living thing?"
-
-# Load a smaller model with caching
-@st.cache_resource(show_spinner="Loading game engine...")
-def load_model():
-    try:
-        return pipeline(
-            "text-generation",
-            model="distilgpt2",  # Smaller than GPT-2
-            device=-1,  # Force CPU
-            framework="pt",  # Explicitly use PyTorch
-            torch_dtype="auto"
-        )
-    except Exception as e:
-        st.error(f"Failed to load model: {str(e)}")
-        return None
+import time
+import random
+from streamlit.components.v1 import html
 
-def generate_question(model, previous_answers):
-    if not previous_answers:
-        return "Is it a living thing?"
-    
-    prompt = """We're playing a guessing game. Here are the previous Q&A:
-"""
-    for i, (q, a) in enumerate(previous_answers, 1):
-        prompt += f"{i}. Q: {q} A: {'Yes' if a else 'No'}\n"
-    prompt += "\nWhat should be the next yes/no question to narrow down the possible options?\nQ:"
-    
-    try:
-        response = model(
-            prompt,
-            max_new_tokens=30,  # Shorter response
-            do_sample=True,
-            temperature=0.7,
-            top_p=0.9
-        )
-        question = response[0]['generated_text'].split("Q:")[-1].strip()
-        question = question.split("\n")[0].split("?")[0] + "?" if "?" not in question else question.split("?")[0] + "?"
-        return question
-    except Exception as e:
-        st.error(f"Error generating question: {str(e)}")
-        return "Is it something you can hold in your hand?"
-
-def main():
-    st.title("KASOTI - The Guessing Game")
-    
-    # Initialize with loading state
-    with st.spinner("Setting up the game..."):
-        model = load_model()
-    
-    if model is None:
-        st.error("Failed to initialize the game. Please refresh the page.")
-        return
-    
-    init_game()
-    
-    st.header("Think of a famous person, place, or object")
-    st.write(f"Questions asked: {st.session_state.questions_asked}/20")
-    
-    if not st.session_state.game_over:
-        st.subheader(st.session_state.current_question)
+# 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');
         
-        col1, col2, col3 = st.columns(3)
-        with col1:
-            if st.button("Yes"):
-                st.session_state.answers.append((st.session_state.current_question, True))
-                st.session_state.questions_asked += 1
-                st.session_state.current_question = generate_question(model, st.session_state.answers)
-                st.rerun()
-        with col2:
-            if st.button("No"):
-                st.session_state.answers.append((st.session_state.current_question, False))
-                st.session_state.questions_asked += 1
-                st.session_state.current_question = generate_question(model, st.session_state.answers)
-                st.rerun()
-        with col3:
-            if st.button("I don't know"):
-                st.session_state.answers.append((st.session_state.current_question, None))
-                st.session_state.questions_asked += 1
-                st.session_state.current_question = generate_question(model, st.session_state.answers)
-                st.rerun()
+        * {
+            font-family: 'Poppins', sans-serif;
+        }
         
-        if st.session_state.questions_asked >= 20:
-            st.session_state.game_over = True
-    
-    if st.session_state.game_over:
-        st.subheader("Game Over!")
-        st.write("I've run out of questions. What were you thinking of?")
-        user_input = st.text_input("Enter what you were thinking of:")
-        if user_input:
-            st.write(f"Ah! I was thinking of {user_input}. Let's play again!")
-            st.session_state.clear()
-            st.rerun()
+        .title {
+            font-size: 3rem !important;
+            font-weight: 700 !important;
+            color: #6C63FF !important;
+            text-align: center;
+            margin-bottom: 0.5rem;
+        }
         
-        if st.button("Play Again"):
-            st.session_state.clear()
-            st.rerun()
-
-if __name__ == "__main__":
-    main()
-# Initialize the game
-def init_game():
-    if 'questions_asked' not in st.session_state:
-        st.session_state.questions_asked = 0
-        st.session_state.answers = []
-        st.session_state.game_over = False
-        st.session_state.current_question = "Is it a living thing?"
-
-# Load the LLM model
-@st.cache_resource
-def load_model():
-    return pipeline("text-generation", model="gpt2")
+        .subtitle {
+            font-size: 1.2rem !important;
+            text-align: center;
+            color: #666 !important;
+            margin-bottom: 2rem;
+        }
+        
+        .question-box {
+            background: #F8F9FA;
+            border-radius: 15px;
+            padding: 2rem;
+            margin: 1.5rem 0;
+            box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+        }
+        
+        .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;
+            color: #6C63FF;
+            text-align: center;
+            margin: 2rem 0;
+        }
+        
+        @keyframes fadeIn {
+            from { opacity: 0; }
+            to { opacity: 1; }
+        }
+        
+        .confetti {
+            position: fixed;
+            top: 0;
+            left: 0;
+            width: 100%;
+            height: 100%;
+            pointer-events: none;
+            z-index: 1000;
+        }
+    </style>
+    """, unsafe_allow_html=True)
 
-def generate_question(model, previous_answers):
-    if not previous_answers:
-        return "Is it a living thing?"
-    
-    # Create a prompt for the LLM based on previous answers
-    prompt = "We're playing a guessing game. Here are the previous Q&A:\n"
-    for i, (q, a) in enumerate(previous_answers, 1):
-        prompt += f"{i}. Q: {q} A: {'Yes' if a else 'No'}\n"
-    prompt += "What should be the next yes/no question to narrow down the possible options?\nQ:"
+# Confetti animation (for final reveal)
+def show_confetti():
+    html("""
+    <canvas id="confetti-canvas" class="confetti"></canvas>
+    <script src="https://cdn.jsdelivr.net/npm/[email protected]/dist/confetti.browser.min.js"></script>
+    <script>
+    const canvas = document.getElementById('confetti-canvas');
+    const confetti = confetti.create(canvas, { resize: true });
     
-    # Generate the next question
-    response = model(prompt, max_length=100, num_return_sequences=1)
-    question = response[0]['generated_text'].split("Q:")[-1].strip()
+    confetti({
+        particleCount: 150,
+        spread: 70,
+        origin: { y: 0.6 }
+    });
     
-    # Clean up the question (remove anything after newlines or multiple questions)
-    question = question.split("\n")[0].split("?")[0] + "?" if "?" not in question else question.split("?")[0] + "?"
-    return question
+    setTimeout(() => {
+        canvas.remove();
+    }, 5000);
+    </script>
+    """)
 
+# Game logic with fake data
 def main():
-    st.title("KASOTI - The Guessing Game")
+    inject_custom_css()
     
-    # Initialize the model and game state
-    model = load_model()
-    init_game()
+    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)
     
-    # Display game header
-    st.header("Think of a famous person, place, or object")
-    st.write(f"Questions asked: {st.session_state.questions_asked}/20")
+    if 'game_state' not in st.session_state:
+        st.session_state.game_state = "start"
+        st.session_state.questions = [
+            "Is it something you can hold in your hand?",
+            "Is it commonly found in households?",
+            "Can it be used as a tool?",
+            "Is it primarily made of metal?",
+            "Does it have moving parts?",
+            "Is it electronic?",
+            "Is it something you would gift to someone?",
+            "Can it be found in an office?",
+            "Does it require electricity to function?",
+            "Is it typically under $100?"
+        ]
+        st.session_state.current_q = 0
+        st.session_state.answers = []
+        st.session_state.target = "smartphone"  # Fake target for testing
     
-    # Display the current question
-    if not st.session_state.game_over:
-        st.subheader(st.session_state.current_question)
+    # Start screen
+    if st.session_state.game_state == "start":
+        st.image("https://via.placeholder.com/600x300?text=Think+of+Something", use_column_width=True)
         
-        # Answer buttons
-        col1, col2, col3 = st.columns(3)
+        with st.form("start_form"):
+            category = st.radio("What are you thinking of?", 
+                              ["A famous person", "An object", "A place"],
+                              index=1)
+            user_input = st.text_input("Type it here (we won't peek!):").strip().lower()
+            
+            if st.form_submit_button("Start Game"):
+                if not user_input:
+                    st.error("Please type something!")
+                elif len(user_input) < 3:
+                    st.error("Too short! Type properly.")
+                else:
+                    st.session_state.game_state = "gameplay"
+                    st.session_state.user_input = user_input
+                    st.rerun()
+    
+    # Gameplay screen
+    elif st.session_state.game_state == "gameplay":
+        col1, col2 = st.columns([1, 3])
         with col1:
-            if st.button("Yes"):
-                st.session_state.answers.append((st.session_state.current_question, True))
-                st.session_state.questions_asked += 1
-                st.session_state.current_question = generate_question(model, st.session_state.answers)
-                st.rerun()
+            st.image("https://via.placeholder.com/200?text=Question", width=150)
         with col2:
-            if st.button("No"):
-                st.session_state.answers.append((st.session_state.current_question, False))
-                st.session_state.questions_asked += 1
-                st.session_state.current_question = generate_question(model, st.session_state.answers)
-                st.rerun()
-        with col3:
-            if st.button("I don't know"):
-                st.session_state.answers.append((st.session_state.current_question, None))
-                st.session_state.questions_asked += 1
-                st.session_state.current_question = generate_question(model, st.session_state.answers)
-                st.rerun()
+            st.markdown(f'<div class="question-box">Question {st.session_state.current_q + 1}/10:<br><br>'
+                       f'<strong>{st.session_state.questions[st.session_state.current_q]}</strong></div>', 
+                       unsafe_allow_html=True)
         
-        # Check if game is over
-        if st.session_state.questions_asked >= 20:
-            st.session_state.game_over = True
-    
-    # Game over state
-    if st.session_state.game_over:
-        st.subheader("Game Over!")
-        st.write("I've run out of questions. What were you thinking of?")
-        user_input = st.text_input("Enter what you were thinking of:")
-        if user_input:
-            st.write(f"Ah! I was thinking of {user_input}. Let's play again!")
-            st.session_state.clear()
+        answer = st.radio("Your answer:", ["Yes", "No"], horizontal=True, label_visibility="collapsed")
+        
+        if st.button("Submit Answer", type="primary"):
+            st.session_state.answers.append(answer == "Yes")
+            st.session_state.current_q += 1
+            
+            if st.session_state.current_q >= len(st.session_state.questions):
+                st.session_state.game_state = "result"
             st.rerun()
+    
+    # Result screen
+    elif st.session_state.game_state == "result":
+        show_confetti()
+        st.markdown('<div class="final-reveal">🎉 I think it\'s a...</div>', unsafe_allow_html=True)
+        time.sleep(1)
+        st.markdown(f'<div class="final-reveal" style="font-size:3.5rem;color:#6C63FF;">{st.session_state.target.upper()}</div>', 
+                   unsafe_allow_html=True)
+        
+        st.image("https://via.placeholder.com/400x200?text=Congratulations", use_column_width=True)
         
-        if st.button("Play Again"):
+        if st.button("Play Again", key="play_again"):
             st.session_state.clear()
             st.rerun()