Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import time
-import random
 from streamlit.components.v1 import html
 
 # Custom CSS for professional look
@@ -127,30 +126,42 @@ def main():
     
     # 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 10 Yes/No questions to guess what it is.</p>
+            <p>You can choose from three categories:</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>
+            </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 = 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()
-            
+            category_input = st.text_input("Enter category (person / place / object):").strip().lower()
+            user_input = st.text_input("Now think of something in that category (we won't peek!):").strip().lower()
+
             if st.form_submit_button("Start Game"):
-                if not user_input:
-                    st.error("Please type something!")
+                if not category_input or not user_input:
+                    st.error("Please fill out both fields!")
+                elif category_input not in ["person", "place", "object"]:
+                    st.error("Oops! Type either person, place or object to start the game!")
                 elif len(user_input) < 3:
                     st.error("Too short! Type properly.")
                 else:
-                    st.session_state.game_state = "gameplay"
+                    st.session_state.category = category_input
                     st.session_state.user_input = user_input
+                    st.session_state.game_state = "gameplay"
                     st.rerun()
-    
+
     # Gameplay screen
     elif st.session_state.game_state == "gameplay":
-        
         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)
+                    f'<strong>{st.session_state.questions[st.session_state.current_q]}</strong></div>', 
+                    unsafe_allow_html=True)
         
         answer = st.radio("Your answer:", ["Yes", "No"], horizontal=True, label_visibility="collapsed")
         
@@ -161,19 +172,18 @@ def main():
             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.markdown(f'<div class="final-reveal" style="font-size:3.5rem;color:#6C63FF;">{st.session_state.user_input.upper()}</div>', 
+                    unsafe_allow_html=True)
         
         if st.button("Play Again", key="play_again"):
             st.session_state.clear()
             st.rerun()
 
 if __name__ == "__main__":
-    main()
+    main()