Create app.py

app.py

@@ -0,0 +1,90 @@
+import streamlit as st
+from transformers import pipeline
+
+# 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")
+
+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:"
+    
+    # Generate the next question
+    response = model(prompt, max_length=100, num_return_sequences=1)
+    question = response[0]['generated_text'].split("Q:")[-1].strip()
+    
+    # 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
+
+def main():
+    st.title("KASOTI - The Guessing Game")
+    
+    # Initialize the model and game state
+    model = load_model()
+    init_game()
+    
+    # Display game header
+    st.header("Think of a famous person, place, or object")
+    st.write(f"Questions asked: {st.session_state.questions_asked}/20")
+    
+    # Display the current question
+    if not st.session_state.game_over:
+        st.subheader(st.session_state.current_question)
+        
+        # Answer buttons
+        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()
+        
+        # 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()
+            st.rerun()
+        
+        if st.button("Play Again"):
+            st.session_state.clear()
+            st.rerun()
+
+if __name__ == "__main__":
+    main()