Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,3 +1,5 @@
+# app.py
+
 import streamlit as st
 import speech_recognition as sr
 import tempfile
@@ -6,13 +8,13 @@ import os
 import requests
 import numpy as np
 
-# API Keys
+# Set API Keys
 os.environ["MISTRAL_API_KEY"] = "cNjUx79Hl0A2AeiAMf6yi7o7ah4APoZy"
 os.environ["GROQ_API_KEY"] = "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"
 
-# Game State
-if "state" not in st.session_state:
-    st.session_state.state = {
+# Initialize session state
+if "game_state" not in st.session_state:
+    st.session_state.game_state = {
         "active": False,
         "questions_asked": 0,
         "answers": [],
@@ -20,22 +22,20 @@ if "state" not in st.session_state:
         "consult_mode": False
     }
 
-# Functions
-def transcribe_audio(uploaded_file, language):
-    if uploaded_file is None:
-        return ""
+def transcribe_audio(audio_bytes, language):
     try:
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
-            tmp_file.write(uploaded_file.read())
-            tmp_file_path = tmp_file.name
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp:
+            tmp.write(audio_bytes)
+            tmp_path = tmp.name
 
         recognizer = sr.Recognizer()
-        with sr.AudioFile(tmp_file_path) as source:
+        with sr.AudioFile(tmp_path) as source:
             audio = recognizer.record(source)
             lang_code = "en-US" if language == "English" else "ur-PK"
-            return recognizer.recognize_google(audio, language=lang_code).lower()
+            text = recognizer.recognize_google(audio, language=lang_code)
+            return text.lower()
     except Exception as e:
-        return f"Transcription error: {e}"
+        return f"⚠️ Transcription failed: {e}"
 
 def query_llm(api, messages, model=None):
     headers = {
@@ -46,90 +46,115 @@ def query_llm(api, messages, model=None):
         "messages": messages,
         "model": model or ("llama3-70b-8192" if api == "GROQ" else "mistral-medium")
     }
-    url = "https://api.groq.com/openai/v1/chat/completions" if api == "GROQ" else "https://api.mistral.ai/v1/chat/completions"
-    response = requests.post(url, headers=headers, json=payload)
-    if response.ok:
-        return response.json()["choices"][0]["message"]["content"]
-    return "API Error"
+    endpoint = {
+        "MISTRAL": "https://api.mistral.ai/v1/chat/completions",
+        "GROQ": "https://api.groq.com/openai/v1/chat/completions"
+    }[api]
 
-def normalize_answer(ans):
-    ans = ans.strip().lower()
-    return "yes" if ans in ["yes", "y", "ہاں", "haan"] else "no" if ans in ["no", "n", "نہیں", "nahi"] else None
+    response = requests.post(endpoint, headers=headers, json=payload)
+    return response.json()["choices"][0]["message"]["content"] if response.status_code == 200 else None
 
 def generate_question(answers):
-    prompt = "You are playing Kasoti. Ask the next best yes/no question.\n\n"
+    prompt = "You are playing Kasoti. Based on these yes/no answers, ask the next best question:\n\n"
     for i, (q, a) in enumerate(answers, 1):
         prompt += f"{i}. Q: {q}\n   A: {a}\n"
-    return query_llm("GROQ", [{"role": "user", "content": prompt}])
+    prompt += "\nAsk ONLY the next best yes/no question."
+    return query_llm("GROQ", [{"role": "user", "content": prompt}]) or "Is it something you can hold?"
 
 def make_guess(answers):
-    prompt = "Based on this history, make a best guess:\n\n"
+    prompt = "Based on this history, make a guess:\n\n"
     for i, (q, a) in enumerate(answers, 1):
         prompt += f"{i}. Q: {q}\n   A: {a}\n"
-    return query_llm("GROQ", [{"role": "user", "content": prompt}])
+    return query_llm("GROQ", [{"role": "user", "content": prompt}]) or "I need more information."
 
 def get_hint(question, answers):
-    prompt = f"The player is unsure about: {question}\n\nHistory:\n"
+    prompt = f"The player is unsure about answering: '{question}'. Previous history:\n"
     for q, a in answers:
         prompt += f"- Q: {q}\n  A: {a}\n"
-    return query_llm("MISTRAL", [{"role": "user", "content": prompt}])
+    prompt += "\nSuggest a helpful hint."
+    return query_llm("MISTRAL", [{"role": "user", "content": prompt}]) or "Try to think about what it really means."
 
-# Streamlit UI
-st.set_page_config(page_title="🎮 Kasoti 20Q", layout="centered")
-
-st.title("🎮 Kasoti - 20 Questions Game")
-st.info("Think of a famous person, place, or object. Answer with **yes/no** or **ہاں/نہیں** only.")
+def normalize_answer(ans):
+    ans = ans.strip().lower()
+    return {"yes": "yes", "y": "yes", "ہاں": "yes", "haan": "yes",
+            "no": "no", "n": "no", "نہیں": "no", "nahi": "no"}.get(ans, None)
 
-if st.button("🔄 Start Game"):
-    st.session_state.state.update({
+def start_game():
+    st.session_state.game_state.update({
         "active": True,
         "questions_asked": 1,
         "answers": [],
-        "current_question": "Is it a living thing?",
-        "consult_mode": False
+        "current_question": "Is it a living thing?"
     })
-
-if st.session_state.state["active"]:
-    st.success(f"🤔 Question {st.session_state.state['questions_asked']}: {st.session_state.state['current_question']}")
-
-    with st.form(key="answer_form"):
-        audio_file = st.file_uploader("🎙️ Upload your answer (WAV only)", type=["wav"])
-        language = st.selectbox("🗣️ Language", ["English", "Urdu"])
-        manual_input = st.text_input("💬 Or type your answer:")
-        submit = st.form_submit_button("➡️ Submit")
-
-    if submit:
-        answer = manual_input or transcribe_audio(audio_file, language)
-        normalized = normalize_answer(answer)
-        if not normalized:
-            st.warning("⚠️ Please reply with 'yes' or 'no' (or 'ہاں/نہیں').")
+    st.success("🎯 Think of something. I’ll guess it in 20 questions or less!")
+    st.info(st.session_state.game_state["current_question"])
+
+def handle_answer(user_input):
+    gs = st.session_state.game_state
+    norm_ans = normalize_answer(user_input)
+    if not norm_ans:
+        st.warning("Please answer 'yes' or 'no' (or ہاں / نہیں).")
+        return
+
+    # Handle guess confirmation
+    if gs["current_question"] and "is this correct?" in gs["current_question"].lower():
+        if norm_ans == "yes":
+            st.success("🎉 YAY! I guessed it!")
+            gs["active"] = False
         else:
-            question = st.session_state.state["current_question"]
-            st.session_state.state["answers"].append((question, normalized))
-
-            if "is this correct?" in question.lower():
-                if normalized == "yes":
-                    st.balloons()
-                    st.success("🎉 I guessed it right!")
-                    st.session_state.state["active"] = False
-                else:
-                    next_q = generate_question(st.session_state.state["answers"])
-                    st.session_state.state["current_question"] = next_q
-                    st.session_state.state["questions_asked"] += 1
-            elif st.session_state.state["questions_asked"] >= 20:
-                st.session_state.state["active"] = False
-                guess = make_guess(st.session_state.state["answers"])
-                st.error(f"❌ Game over! Final guess: {guess}")
-            elif st.session_state.state["questions_asked"] % 5 == 0:
-                guess = make_guess(st.session_state.state["answers"])
-                st.session_state.state["current_question"] = guess + " Is this correct?"
-                st.session_state.state["questions_asked"] += 1
-            else:
-                next_q = generate_question(st.session_state.state["answers"])
-                st.session_state.state["current_question"] = next_q
-                st.session_state.state["questions_asked"] += 1
-
-if st.session_state.state["active"]:
-    if st.toggle("💡 Enable Consult Mode"):
-        hint = get_hint(st.session_state.state["current_question"], st.session_state.state["answers"])
-        st.info(f"💭 Hint: {hint}")
+            q = generate_question(gs["answers"])
+            gs["current_question"] = q
+            gs["questions_asked"] += 1
+            st.info(q)
+        return
+
+    gs["answers"].append((gs["current_question"], norm_ans))
+    if gs["questions_asked"] >= 20:
+        final = make_guess(gs["answers"])
+        st.error(f"🎮 Game Over! My final guess: {final}")
+        gs["active"] = False
+        return
+
+    if gs["questions_asked"] % 5 == 0:
+        guess = make_guess(gs["answers"])
+        if guess.startswith("I think it's"):
+            gs["current_question"] = guess + " Is this correct?"
+            st.warning(gs["current_question"])
+            return
+
+    next_q = generate_question(gs["answers"])
+    gs["current_question"] = next_q
+    gs["questions_asked"] += 1
+    st.info(next_q)
+
+# UI
+st.set_page_config(page_title="Kasoti Game 🎮", layout="centered")
+st.title("🎮 Kasoti - 20 Questions Game")
+st.markdown("Think of a **person, place, or object**. Answer with **yes/no** or **ہاں/نہیں**. I will guess!")
+
+if st.button("Start Game"):
+    start_game()
+
+language = st.selectbox("Select Language for Audio Input", ["English", "Urdu"])
+audio_input = st.file_uploader("🎤 Upload your Answer (WAV format)", type=["wav"])
+if audio_input:
+    transcribed = transcribe_audio(audio_input.read(), language)
+    st.text_area("Transcribed Answer", value=transcribed, key="text_input")
+
+manual_input = st.text_input("Or type your answer here (yes/no/ہاں/نہیں):")
+
+if st.button("Submit Answer"):
+    answer_text = manual_input or st.session_state.get("text_input", "")
+    if not answer_text:
+        st.warning("Please provide an answer.")
+    else:
+        handle_answer(answer_text)
+
+if st.checkbox("Enable Consult Mode"):
+    st.session_state.game_state["consult_mode"] = True
+    if st.button("🔍 Get Hint"):
+        hint = get_hint(
+            st.session_state.game_state["current_question"],
+            st.session_state.game_state["answers"]
+        )
+        st.info(f"💡 Hint: {hint}")