Update app.py

app.py

@@ -1,30 +1,135 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer
-
-# Load the Mistral model and tokenizer from Hugging Face Hub
-model_name = "mistralai/mistral-7b"  # Change this to the desired model name
-model = AutoModelForCausalLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-
-# Function to generate text
-def generate_text(prompt: str, max_length: int = 50):
-    # Tokenize input prompt
-    inputs = tokenizer(prompt, return_tensors="pt")
-    
-    # Generate text
-    outputs = model.generate(
-        input_ids=inputs['input_ids'],
-        max_length=max_length,  # Max tokens in output
-        num_return_sequences=1,  # Number of sequences to generate
-        temperature=0.7,  # Adjust for randomness (higher = more random)
-        top_p=0.9,  # Use top-p sampling for diversity
-        top_k=50  # Top-k sampling for diversity
-    )
-    
-    # Decode the generated text and return it
-    generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return generated_text
-
-# Example usage
-prompt = "The future of artificial intelligence is"
-generated_output = generate_text(prompt)
-print("Generated Text:", generated_output)
+import streamlit as st
+import speech_recognition as sr
+import tempfile
+import scipy.io.wavfile
+import os
+import requests
+import numpy as np
+
+# API Keys
+os.environ["MISTRAL_API_KEY"] = "your-mistral-key"
+os.environ["GROQ_API_KEY"] = "your-groq-key"
+
+# Game State
+if "state" not in st.session_state:
+    st.session_state.state = {
+        "active": False,
+        "questions_asked": 0,
+        "answers": [],
+        "current_question": None,
+        "consult_mode": False
+    }
+
+# Functions
+def transcribe_audio(uploaded_file, language):
+    if uploaded_file is None:
+        return ""
+    try:
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
+            tmp_file.write(uploaded_file.read())
+            tmp_file_path = tmp_file.name
+
+        recognizer = sr.Recognizer()
+        with sr.AudioFile(tmp_file_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()
+    except Exception as e:
+        return f"Transcription error: {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")
+    }
+    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"
+
+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
+
+def generate_question(answers):
+    prompt = "You are playing Kasoti. Ask the next best yes/no 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}])
+
+def make_guess(answers):
+    prompt = "Based on this history, make a best 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}])
+
+def get_hint(question, answers):
+    prompt = f"The player is unsure about: {question}\n\nHistory:\n"
+    for q, a in answers:
+        prompt += f"- Q: {q}\n  A: {a}\n"
+    return query_llm("MISTRAL", [{"role": "user", "content": prompt}])
+
+# 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.")
+
+if st.button("🔄 Start Game"):
+    st.session_state.state.update({
+        "active": True,
+        "questions_asked": 1,
+        "answers": [],
+        "current_question": "Is it a living thing?",
+        "consult_mode": False
+    })
+
+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 'ہاں/نہیں').")
+        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}")