Update app.py

app.py

@@ -1,21 +1,24 @@
+import os
+os.system("pip install -q git+https://github.com/openai/whisper.git")
+
 import gradio as gr
-import os, json, re, uuid
+import json, re, uuid
 from datetime import datetime
 from pydub import AudioSegment
 import whisper
 import requests
 import emoji
 
-# Load keys and models
+# Load API key
 groq_key = os.getenv("GROQ_API_KEY")
 whisper_model = whisper.load_model("base")
 
-# Emoji filter for assistant replies
+# ✅ Safe emoji filtering
 def filter_emojis(text):
-    allowed = {"😊", "😄", "👍", "🤖", "✨", "🎉", "💬", "🙌", "😎", "📢", "🧠", "✅"}
-    return "".join(char if char not in emoji.EMOJI_DATA or char in allowed else "" for char in text)
+    allowed_emojis = {"😊", "😄", "👍", "🤖", "✨", "🎉", "💬", "🙌", "😎", "📢", "🧠", "✅"}
+    return "".join(char if char not in emoji.EMOJI_DATA or char in allowed_emojis else "" for char in text)
 
-# Chat function using Groq API
+# ✅ Chat handler
 def chat_with_groq(message, history):
     messages = [{"role": "system", "content": "You are JAWERIA'SBOT 🤖 – cheerful, emoji-savvy, and sleek."}]
     messages += history + [{"role": "user", "content": message}]
@@ -24,17 +27,23 @@ def chat_with_groq(message, history):
         "Authorization": f"Bearer {groq_key}",
         "Content-Type": "application/json"
     }
-    payload = {"model": "llama3-70b-8192", "messages": messages}
-    res = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, json=payload)
-    reply = filter_emojis(res.json()["choices"][0]["message"]["content"])
+
+    payload = {
+        "model": "llama3-70b-8192",
+        "messages": messages
+    }
+
+    response = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, json=payload)
+    raw_reply = response.json()["choices"][0]["message"]["content"]
+    reply = filter_emojis(raw_reply)
 
     history += [{"role": "user", "content": message}, {"role": "assistant", "content": reply}]
     return "", history, history
 
-# Transcribe audio from mic or file
+# ✅ Transcription
 def transcribe_audio(audio_path):
-    if not audio_path or not os.path.exists(audio_path):
-        return "⚠️ No audio available."
+    if audio_path is None or not os.path.exists(audio_path):
+        return "⚠️ No audio input."
     try:
         temp_wav = f"{uuid.uuid4()}.wav"
         AudioSegment.from_file(audio_path).export(temp_wav, format="wav")
@@ -44,11 +53,13 @@ def transcribe_audio(audio_path):
     except Exception as e:
         return f"❌ Transcription error: {e}"
 
-# Save/load chat sessions
+# ✅ Save/load
 def save_session(history):
-    title = re.sub(r"[^\w\s]", "", next((m["content"] for m in history if m["role"] == "user"), "chat")).strip()
+    prompt = next((m["content"] for m in history if m["role"] == "user"), "chat")
+    title = re.sub(r"[^\w\s]", "", prompt).strip()
     title = " ".join(title.split()[:6])
-    filename = f"{title} - {datetime.now().strftime('%b %d %Y %H-%M')}.json"
+    timestamp = datetime.now().strftime("%b %d %Y %H-%M")
+    filename = f"{title} - {timestamp}.json"
     with open(filename, "w", encoding="utf-8") as f:
         json.dump(history, f, indent=2, ensure_ascii=False)
     return f"✅ Saved `{filename[:-5]}`"
@@ -57,69 +68,74 @@ def list_saved_files():
     return sorted([f[:-5] for f in os.listdir() if f.endswith(".json")])
 
 def load_chat(name):
+    filename = f"{name}.json"
     try:
-        with open(f"{name}.json", "r", encoding="utf-8") as f:
+        with open(filename, "r", encoding="utf-8") as f:
             history = json.load(f)
         return history, history, f"✅ Loaded `{name}`"
     except Exception as e:
         return [], [], f"❌ Load error: {e}"
 
-# Interface
+# ✅ Gradio Interface
 with gr.Blocks(css="""
-body { background: #111; color: white; font-family: 'Segoe UI'; }
+body { background-color: #111; font-family: 'Segoe UI', sans-serif; }
+.gr-chatbot { background-color: #1e1e1e; border-radius: 10px; border: 1px solid #333; }
+.gr-chatbot-message { background-color: #292929; color: #f1f1f1; border-radius: 10px; margin-bottom: 5px; }
 textarea, input[type='text'] {
-    background-color: #222; color: white; border-radius: 30px;
-    padding: 10px 15px; border: 1px solid #444; height: 48px;
+    background-color: #222; color: #fff; border-radius: 30px;
+    border: 1px solid #444; padding: 10px 15px; height: 48px;
 }
 .gr-button {
-    background-color: #333 !important; color: white !important;
-    border: 1px solid #444; border-radius: 10px; font-weight: bold;
+    background-color: #333 !important; color: #fff !important;
+    border: 1px solid #444 !important; border-radius: 8px;
+    font-weight: bold; transition: background-color 0.2s;
 }
-.gr-button:hover { background: #555 !important; }
-.gr-dropdown { background: #222; color: white; max-height: 200px; overflow-y: auto; }
-.gr-chatbot-message { background: #292929 !important; color: white; border-radius: 10px; }
+.gr-button:hover { background-color: #555 !important; cursor: pointer; }
+.gr-dropdown { background-color: #222 !important; color: #fff !important; max-height: 200px; overflow-y: auto; }
+h1, h2, h3, .gr-markdown { color: #f1f1f1 !important; text-align: center; }
 """) as demo:
 
     state = gr.State([])
+
+    gr.Markdown("# ✨ JAWERIA'SBOT 🤖")
+    gr.Markdown("Type or speak — your assistant listens and responds 💬")
+
+    chatbot = gr.Chatbot(label="JAWERIA'SBOT 🤖", height=400, type="messages")
+
     with gr.Row():
-        chat_input = gr.Textbox(placeholder="Type or speak...", label=None, scale=9)
-        record_btn = gr.Button("🎙️ Record", scale=1)
-        upload_btn = gr.Audio(source="upload", type="filepath", visible=False)
+        chat_input = gr.Textbox(placeholder="Type your message...", scale=8, show_label=False)
+        record_btn = gr.Button("🎙️", scale=1)
+        upload_btn = gr.File(file_types=[".mp3", ".wav", ".m4a"], label="📁 Upload", visible=False)
 
-    chatbot = gr.Chatbot(label="JAWERIA'SBOT 🤖", height=400)
     send_btn = gr.Button("Send 🚀")
 
+    recording = gr.Audio(type="filepath", visible=False)
+
     with gr.Row():
-        new_btn = gr.Button("🆕 New")
+        new_chat_btn = gr.Button("🆕 New")
         save_btn = gr.Button("💾 Save")
-        dropdown = gr.Dropdown(label="📂 Load Saved", choices=list_saved_files(), interactive=True)
+        saved_dropdown = gr.Dropdown(label="📂 Load Saved", choices=list_saved_files(), interactive=True)
         load_btn = gr.Button("📥 Load")
 
     save_msg = gr.Markdown()
     load_msg = gr.Markdown()
 
-    recording = gr.Audio(source="microphone", type="filepath", visible=False)
-
-    def toggle_recording(is_recording):
-        return not is_recording, gr.update(visible=not is_recording)
-
-    def start_record(): return gr.update(visible=True)
-    def stop_record(path): return transcribe_audio(path)
-
-    record_state = gr.State(False)
-    record_btn.click(start_record, outputs=recording).then(
-        fn=lambda: True, outputs=record_state)
-    recording.change(fn=stop_record, inputs=recording, outputs=chat_input)
-
-    upload_btn.change(fn=transcribe_audio, inputs=upload_btn, outputs=chat_input)
+    # Event bindings
     send_btn.click(chat_with_groq, inputs=[chat_input, state], outputs=[chat_input, chatbot, state])
     chat_input.submit(chat_with_groq, inputs=[chat_input, state], outputs=[chat_input, chatbot, state])
-    new_btn.click(lambda: ("", [], []), outputs=[chat_input, chatbot, state])
-    save_btn.click(save_session, inputs=[state], outputs=[save_msg])
-    save_btn.click(list_saved_files, outputs=[dropdown])
-    load_btn.click(load_chat, inputs=[dropdown], outputs=[chatbot, state, load_msg])
-    demo.load(list_saved_files, outputs=[dropdown])
+    recording.change(transcribe_audio, inputs=recording, outputs=chat_input)
+    upload_btn.change(transcribe_audio, inputs=upload_btn, outputs=chat_input)
 
-    gr.Markdown("<center><small>Made with 🤍 by Jaweria</small></center>")
+    def toggle_recording(current):
+        return gr.update(visible=not current)
+
+    record_btn.click(lambda: gr.update(visible=True), None, recording)
+
+    new_chat_btn.click(fn=lambda: ("", [], []), outputs=[chat_input, chatbot, state])
+    save_btn.click(fn=save_session, inputs=[state], outputs=[save_msg])
+    save_btn.click(fn=list_saved_files, outputs=[saved_dropdown])
+    load_btn.click(fn=load_chat, inputs=[saved_dropdown], outputs=[chatbot, state, load_msg])
+    demo.load(fn=list_saved_files, outputs=[saved_dropdown])
 
 demo.launch()
+