Update app.py

app.py

@@ -1,5 +1,3 @@
-# app.py
-
 import gradio as gr
 import requests
 import speech_recognition as sr
@@ -9,8 +7,8 @@ import os
 import numpy as np
 
 # Set your API keys as HF secrets or environment variables
-os.environ["MISTRAL_API_KEY"] = os.getenv("MISTRAL_API_KEY", "your-mistral-key")
-os.environ["GROQ_API_KEY"] = os.getenv("GROQ_API_KEY", "your-groq-key")
+os.environ["MISTRAL_API_KEY"] = "cNjUx79Hl0A2AeiAMf6yi7o7ah4APoZy"
+os.environ["GROQ_API_KEY"] = "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"
 
 game_state = {
     "active": False,
@@ -19,6 +17,7 @@ game_state = {
     "current_question": None,
     "consult_mode": False
 }
+consult_history = []
 
 def transcribe_audio(audio, language):
     if audio is None:
@@ -40,7 +39,20 @@ def transcribe_audio(audio, language):
             return text.lower()
     except Exception as e:
         print(f"Transcription error: {e}")
-        return ""
+        return "❌ Could not transcribe the audio."
+
+def handle_user_question(user_question):
+    if not game_state["consult_mode"]:
+        return "❗Consultant mode is off.", gr.update(value=""), gr.update(visible=True)
+
+    messages = [{"role": "user", "content": user_question}]
+    answer = query_llm("MISTRAL", messages)
+    if answer:
+        consult_history.append((user_question, answer))
+        formatted_history = "\n".join([f"**Q:** {q}\n**A:** {a}" for q, a in consult_history])
+        return answer, gr.update(value=formatted_history), gr.update(visible=True)
+    else:
+        return "⚠️ Failed to get a response.", gr.update(), gr.update()
 
 def query_llm(api, messages, model=None):
     headers = {
@@ -147,31 +159,40 @@ def get_consult_hint():
     return get_hint(game_state["current_question"], game_state["answers"])
 
 # Colorful and polished UI
-with gr.Blocks(css=".gr-button {background-color: #4CAF50; color: white} .gr-box {background: #f7f7ff; border: 1px solid #ccc; padding: 10px; border-radius: 10px;}") as demo:
+with gr.Blocks(css="") as demo:
     gr.Markdown("## 🧠 Kasoti: 20 Questions AI Game")
-    gr.Markdown("Think of a person, place, or thing. I'll try to guess it in 20 questions or less! Just answer 'yes' or 'no'.")
-    
+    gr.Markdown("Think of a person, place, or thing. I'll try to guess it in 20 questions or less!")
+
     with gr.Row():
-        start_btn = gr.Button("🚀 Start Game")
-        consult_btn = gr.Button("💬 Toggle Consult Mode")
+        start_btn = gr.Button("🚀 Start Game", elem_id="start-btn")
+        consult_btn = gr.Button("💬 Toggle Consult Mode", elem_id="consult-btn")
         consult_status = gr.Textbox(label="Consult Mode", interactive=False)
 
     with gr.Row():
         with gr.Column():
             language = gr.Dropdown(["English", "Urdu"], label="Audio Language", value="English")
             audio_input = gr.Audio(label="🎤 Answer via Microphone", type="numpy", sources=["microphone"])
-            transcribe_btn = gr.Button("📝 Transcribe Audio")
+            transcribe_btn = gr.Button("📝 Transcribe Audio", elem_id="transcribe-btn")
             transcribed_text = gr.Textbox(label="✍️ Answer Text", interactive=True)
-            submit_btn = gr.Button("✅ Submit Answer")
+            submit_btn = gr.Button("✅ Submit Answer", elem_id="submit-btn")
 
         with gr.Column():
             game_output = gr.Textbox(label="🎲 Game Progress", interactive=False)
             consult_output = gr.Textbox(label="💡 Consult Hint", visible=False)
 
+    # Consultant Interaction Section
+    with gr.Row(visible=False) as consult_row:
+        user_question = gr.Textbox(label="Ask a question to the LLM")
+        ask_btn = gr.Button("📤 Ask")
+        llm_answer = gr.Textbox(label="🧠 LLM Answer", interactive=False)
+        consult_history_box = gr.Textbox(label="🗂️ Session History", lines=8, interactive=False)
+
+    # Events
     start_btn.click(start_game, outputs=[game_output, transcribed_text, submit_btn, consult_status, consult_output])
-    consult_btn.click(toggle_consult_mode, outputs=[consult_status, consult_output])
+    consult_btn.click(toggle_consult_mode, outputs=[consult_status, consult_output, consult_row])
     consult_btn.click(get_consult_hint, outputs=[consult_output])
     transcribe_btn.click(transcribe_audio, inputs=[audio_input, language], outputs=[transcribed_text])
     submit_btn.click(process_answer, inputs=[transcribed_text], outputs=[game_output, transcribed_text, transcribed_text, consult_output])
+    ask_btn.click(handle_user_question, inputs=[user_question], outputs=[llm_answer, consult_history_box, consult_row])
 
-demo.launch()
+demo.launch()