src/streamlit_app.py

import streamlit as st
import speech_recognition as sr
import tempfile
import scipy.io.wavfile
import os
import requests
import numpy as np

# Set API Keys
os.environ["MISTRAL_API_KEY"] = "cNjUx79Hl0A2AeiAMf6yi7o7ah4APoZy"
os.environ["GROQ_API_KEY"] = "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"

# Initialize session state
if "game_state" not in st.session_state:
    st.session_state.game_state = {
        "active": False,
        "questions_asked": 0,
        "answers": [],
        "current_question": None,
        "consult_mode": False
    }

def transcribe_audio(audio_bytes, language):
    try:
        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp:
            tmp.write(audio_bytes)
            tmp_path = tmp.name

        recognizer = sr.Recognizer()
        with sr.AudioFile(tmp_path) as source:
            audio = recognizer.record(source)
            lang_code = "en-US" if language == "English" else "ur-PK"
            text = recognizer.recognize_google(audio, language=lang_code)
            return text.lower()
    except Exception as e:
        return f"⚠️ Transcription failed: {e}"

def query_llm(api, messages, model=None):
    headers = {
        "Authorization": f"Bearer {os.environ[f'{api}_API_KEY']}",
        "Content-Type": "application/json"
    }
    payload = {
        "messages": messages,
        "model": model or ("llama3-70b-8192" if api == "GROQ" else "mistral-medium")
    }
    endpoint = {
        "MISTRAL": "https://api.mistral.ai/v1/chat/completions",
        "GROQ": "https://api.groq.com/openai/v1/chat/completions"
    }[api]

    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. 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"
    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 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}]) or "I need more information."

def get_hint(question, answers):
    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"
    prompt += "\nSuggest a helpful hint."
    return query_llm("MISTRAL", [{"role": "user", "content": prompt}]) or "Try to think about what it really means."

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)

def start_game():
    st.session_state.game_state.update({
        "active": True,
        "questions_asked": 1,
        "answers": [],
        "current_question": "Is it a living thing?"
    })
    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:
            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}")