Update app.py

app.py

@@ -16,13 +16,13 @@ state = {
     "current_q": "",
     "question_num": 0,
     "hint_mode": False,
-    "final_answer": "",
-    "category": ""
+    "final_answer": ""
 }
 
 def convert_audio_to_text(audio, lang_choices):
     if audio is None:
         return ""
+
     try:
         sr_rate, audio_data = audio
         with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as fp:
@@ -48,6 +48,7 @@ def query_llm(api, messages, model=None):
         "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)
     if response.status_code == 200:
         return response.json()["choices"][0]["message"]["content"]
@@ -55,23 +56,27 @@ def query_llm(api, messages, model=None):
         return f"[{api} Error] {response.status_code} - {response.text}"
 
 def begin_game():
-    state.update({
-        "started": True,
-        "history": [],
-        "question_num": 1,
-        "hint_mode": False,
-        "final_answer": "",
-        "current_q": "Does it belong to the world of living things?"
-    })
-    return f"Question 1: {state['current_q']}", "", gr.update(visible=False)
+    state["started"] = True
+    state["history"] = []
+    state["question_num"] = 1
+    state["hint_mode"] = False
+    state["current_q"] = "Does it belong to the world of living things?"
+    state["final_answer"] = ""
+    return (
+        "🎮 Game started! Think of something and answer with yes/no.\n\n"
+        f"Question 1: {state['current_q']}",
+        gr.update(visible=False),
+        gr.update(visible=True),
+        gr.update(visible=False)  # Final Answer Box - Hidden initially
+    )
 
 def interpret_answer(user_answer):
     if not state["started"]:
-        return "⛔ Please start a new game.", "", gr.update(visible=False)
+        return "⛔ Please start a new game.", "", "", gr.update(visible=False)
 
     normalized = user_answer.strip().lower()
     if normalized not in ["yes", "no", "ہاں", "نہیں"]:
-        return "⚠️ Please reply with 'yes' or 'no'.", user_answer, gr.update(visible=False)
+        return "⚠️ Respond with 'yes' or 'no' (or ہاں/نہیں).", "", user_answer, gr.update(visible=False)
 
     state["history"].append((state["current_q"], normalized))
 
@@ -79,27 +84,26 @@ def interpret_answer(user_answer):
         state["started"] = False
         guess = generate_final_guess()
         state["final_answer"] = guess
-        return f"🎯 I think the answer is: {guess}", user_answer, gr.update(visible=True)
+        return f"🎯 Final guess: {guess}", "", user_answer, gr.update(visible=True)
 
     if state["question_num"] % 5 == 0:
         guess = generate_final_guess()
         if "i think" in guess.lower():
             state["current_q"] = f"{guess} Am I right?"
-            return f"{state['current_q']}", user_answer, gr.update(visible=False)
+            return f"🤔 {state['current_q']}", "", user_answer, gr.update(visible=False)
 
     state["question_num"] += 1
     state["current_q"] = get_next_question()
-    return f"Question {state['question_num']}: {state['current_q']}", "", gr.update(visible=False)
+    return f"❓ Question {state['question_num']}: {state['current_q']}", "", user_answer, gr.update(visible=False)
 
 def get_next_question():
     history_prompt = "\n".join([f"Q: {q}\nA: {a}" for q, a in state["history"]])
-    category_hint = f"\nCategory: {state['category']}" if state["category"] else ""
-    prompt = f"""You're playing a guessing game. Ask a smart yes/no question based on the following:\n{history_prompt}{category_hint}\nYour question:"""
+    prompt = f"""You're playing a guessing game. You can only ask yes/no questions. Based on the following history, ask the next smart question:  {history_prompt}  Your question:"""
     return query_llm("MISTRAL", [{"role": "user", "content": prompt}]).strip()
 
 def generate_final_guess():
     history = "\n".join([f"Q: {q}\nA: {a}" for q, a in state["history"]])
-    prompt = f"""Based on this Q&A, try to guess what the user is thinking of:\n{history}\nYour guess:"""
+    prompt = f"""Guess the secret concept based on these Q&A. If you're not sure, say "I need more clues."  {history}  Your guess:"""
     return query_llm("MISTRAL", [{"role": "user", "content": prompt}])
 
 def hint_response():
@@ -107,48 +111,42 @@ def hint_response():
         return "Hint mode is off or game not active."
     question = state["current_q"]
     history = "\n".join([f"{q} - {a}" for q, a in state["history"]])
-    prompt = f"""The user seems confused by this question: "{question}". Prior Q&A:\n{history}\n\nGive a short and helpful hint."""
+    prompt = f"""The player seems confused by this question: "{question}". Previous Q&A were:\n{history}\n\nGive a helpful, simple hint."""
     return query_llm("MISTRAL", [{"role": "user", "content": prompt}])
 
 def toggle_hint_mode():
     state["hint_mode"] = not state["hint_mode"]
-    return ("Hint Mode: ON" if state["hint_mode"] else "Hint Mode: OFF")
-
-def choose_category(category):
-    state["category"] = category
-    return f"📦 Category selected: {category}"
+    return ("Hint Mode: ON" if state["hint_mode"] else "Hint Mode: OFF"), gr.update(visible=state["hint_mode"])
 
 with gr.Blocks(title="Kasoti") as demo:
-    gr.Markdown("## 🎮 Kasoti - AI Guessing Game")
-    gr.Markdown("Think of something in your mind. I’ll try to guess it in 20 questions or less. Only reply **yes/no**.")
+    gr.Markdown("## 🌸 Kasoti")
+    gr.Markdown("Think of something. I'll guess it in 20 tries or less. Answer **yes/no** each time.")
 
     with gr.Row():
-        start = gr.Button("🎲 Start Game")
-        hint_toggle = gr.Button("💡 Toggle Hint Mode")
-        hint_status = gr.Textbox(value="Hint Mode: OFF", interactive=False, show_label=False)
-
-    with gr.Row():
-        gr.Markdown("### 📂 Choose a category to help the AI")
-        with gr.Row():
-            for cat in ["Animal", "Plant", "Thing", "Car", "Human", "Place", "Other"]:
-                gr.Button(cat).click(fn=choose_category, inputs=[], outputs=[hint_status], _js=None, _name=None, _id=None, _target=None)
+        start = gr.Button("🎲 Start Game", scale=1)
+        hint_toggle = gr.Button("💡Hint Mode", scale=1)
+        hint_status = gr.Textbox(value="Hint Mode: OFF", show_label=False, interactive=False)
 
     with gr.Row():
         with gr.Column(scale=1):
-            lang_sel = gr.CheckboxGroup(["English", "Urdu"], label="Select Language", value=["English"])
-            mic_input = gr.Audio(sources=["microphone"], type="numpy", label="🎙️ Record Answer")
-            transcribe = gr.Button("📝 Transcribe")
-            typed_ans = gr.Textbox(label="✍️ Or type answer here")
+            gr.Markdown("### 🎤 Input Zone")
+            lang_sel = gr.CheckboxGroup(
+                ["English", "Urdu"], label="Select Languages", value=["English"]
+            )
+            mic_input = gr.Audio(sources=["microphone"], type="numpy", label="Record Your Answer")
+            transcribe = gr.Button("🎙️ Transcribe")
+            typed_ans = gr.Textbox(label="✍️ Or type your answer here")
             submit = gr.Button("✅ Submit Answer")
 
         with gr.Column(scale=2):
-            game_text = gr.Textbox(label="🧠 AI Question", lines=3, interactive=False)
-            final_answer_box = gr.Textbox(label="🎯 Final Guess", visible=False, lines=2, interactive=False)
+            gr.Markdown("### 🧠 Game Engine")
+            game_text = gr.Textbox(label="📝 Game Status", lines=10, interactive=False)
+            final_answer_box = gr.Textbox(label="🎯 Final Answer", visible=False, lines=3, interactive=False)
 
-    start.click(fn=begin_game, outputs=[game_text, typed_ans, final_answer_box])
-    transcribe.click(fn=convert_audio_to_text, inputs=[mic_input, lang_sel], outputs=[typed_ans])
-    submit.click(fn=interpret_answer, inputs=[typed_ans], outputs=[game_text, typed_ans, final_answer_box])
-    hint_toggle.click(fn=toggle_hint_mode, outputs=[hint_status])
+    start.click(fn=begin_game, outputs=[game_text, hint_status, submit, final_answer_box])
+    hint_toggle.click(fn=toggle_hint_mode, outputs=[hint_status, hint_status])
     hint_toggle.click(fn=hint_response, outputs=[hint_status])
+    transcribe.click(fn=convert_audio_to_text, inputs=[mic_input, lang_sel], outputs=[typed_ans])
+    submit.click(fn=interpret_answer, inputs=[typed_ans], outputs=[game_text, typed_ans, typed_ans, final_answer_box])
 
-demo.launch()
+demo.launch()