Update app.py

app.py

@@ -6,7 +6,7 @@ import tempfile
 import scipy.io.wavfile
 import requests
 
-# Set your API keys here
+# Set your API keys
 os.environ["MISTRAL_API_KEY"] = "R1ISnVkHrj7fSd5Dh6ZSZHqCJhhct0ZR"
 os.environ["GROQ_API_KEY"] = "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"
 
@@ -51,7 +51,7 @@ def query_llm(api, messages, model=None):
 
     response = requests.post(endpoint, headers=headers, json=payload)
     if response.status_code == 200:
-        return response.json()["choices"][0]["message"]["content"]
+        return response.json()["choices"][0]["message"]["content"].strip()
     else:
         return f"[{api} Error] {response.status_code} - {response.text}"
 
@@ -60,23 +60,23 @@ def begin_game():
     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"] = ""
+    state["current_q"] = "Does it belong to the world of living things?"
     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
+        gr.update(visible=False)
     )
 
 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), "", gr.update(visible=False)
 
     normalized = user_answer.strip().lower()
     if normalized not in ["yes", "no", "ہاں", "نہیں"]:
-        return "⚠️ Respond with 'yes' or 'no' (or ہاں/نہیں).", "", user_answer, gr.update(visible=False)
+        return "⚠️ Respond with 'yes' or 'no' (or ہاں/نہیں).", user_answer, gr.update(visible=False), "", gr.update(visible=False)
 
     state["history"].append((state["current_q"], normalized))
 
@@ -84,26 +84,26 @@ def interpret_answer(user_answer):
         state["started"] = False
         guess = generate_final_guess()
         state["final_answer"] = guess
-        return f"🎯 Final guess: {guess}", "", user_answer, gr.update(visible=True)
+        return "Game Over!", user_answer, gr.update(visible=True), "", gr.update(value=guess, visible=True)
 
     if state["question_num"] % 5 == 0:
         guess = generate_final_guess()
-        if "i think" in guess.lower():
+        if "i think" in guess.lower() or "maybe" in guess.lower():
             state["current_q"] = f"{guess} Am I right?"
-            return f"🤔 {state['current_q']}", "", user_answer, gr.update(visible=False)
+            return state["current_q"], user_answer, gr.update(visible=False), "", gr.update(visible=False)
 
     state["question_num"] += 1
     state["current_q"] = get_next_question()
-    return f"❓ Question {state['question_num']}: {state['current_q']}", "", user_answer, gr.update(visible=False)
+    return f"Question {state['question_num']}: {state['current_q']}", user_answer, gr.update(visible=False), "", gr.update(visible=False)
 
 def get_next_question():
     history_prompt = "\n".join([f"Q: {q}\nA: {a}" for q, a in state["history"]])
-    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()
+    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:\n{history_prompt}\n\nYour question:"""
+    return query_llm("MISTRAL", [{"role": "user", "content": prompt}])
 
 def generate_final_guess():
     history = "\n".join([f"Q: {q}\nA: {a}" for q, a in state["history"]])
-    prompt = f"""Guess the secret concept based on these Q&A. If you're not sure, say "I need more clues."  {history}  Your guess:"""
+    prompt = f"""Guess the secret concept based on these Q&A. If you're not sure, say "I need more clues."\n\n{history}\n\nYour guess:"""
     return query_llm("MISTRAL", [{"role": "user", "content": prompt}])
 
 def hint_response():
@@ -116,37 +116,39 @@ def hint_response():
 
 def toggle_hint_mode():
     state["hint_mode"] = not state["hint_mode"]
-    return ("Hint Mode: ON" if state["hint_mode"] else "Hint Mode: OFF"), gr.update(visible=state["hint_mode"])
+    return (
+        "Hint Mode: ON" if state["hint_mode"] else "Hint Mode: OFF",
+        gr.update(visible=True)
+    )
 
 with gr.Blocks(title="Kasoti") as demo:
-    gr.Markdown("## 🌸 Kasoti")
-    gr.Markdown("Think of something. I'll guess it in 20 tries or less. Answer **yes/no** each time.")
+    gr.Markdown("## 🧠 Kasoti: 20 Questions Challenge")
+    gr.Markdown("Think of something. I will try to guess it in 20 yes/no questions. Answer with **yes/no** or **ہاں/نہیں**.")
 
     with gr.Row():
         start = gr.Button("🎲 Start Game", scale=1)
-        hint_toggle = gr.Button("💡Hint Mode", scale=1)
+        hint_toggle = gr.Button("💡 Toggle 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):
-            gr.Markdown("### 🎤 Input Zone")
-            lang_sel = gr.CheckboxGroup(
-                ["English", "Urdu"], label="Select Languages", value=["English"]
-            )
+            gr.Markdown("### 🎤 Input")
+            lang_sel = gr.CheckboxGroup(["English", "Urdu"], label="Select Language", 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")
+            typed_ans = gr.Textbox(label="✍️ Or type your answer")
             submit = gr.Button("✅ Submit Answer")
 
         with gr.Column(scale=2):
-            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, 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])
+            gr.Markdown("### 🎮 Game Status")
+            game_q_box = gr.Textbox(label="🧩 Current Question", interactive=False, lines=2)
+            game_history = gr.Textbox(label="🧾 Hint / Status", interactive=False, lines=2)
+            final_answer_box = gr.Textbox(label="🎯 Final Answer", visible=False, lines=2, interactive=False)
+
+    start.click(fn=begin_game, outputs=[game_q_box, game_history, final_answer_box, submit, final_answer_box])
+    hint_toggle.click(fn=toggle_hint_mode, outputs=[hint_status, game_history])
+    hint_toggle.click(fn=hint_response, outputs=[game_history])
     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])
+    submit.click(fn=interpret_answer, inputs=[typed_ans], outputs=[game_q_box, typed_ans, final_answer_box, game_history, final_answer_box])
 
-demo.launch() 
+demo.launch()