Update app.py

app.py

@@ -1,34 +1,111 @@
 import gradio as gr
-import google.generativeai as genai
-import os
+import os, json, re, uuid
+from datetime import datetime
+from pydub import AudioSegment
+import whisper
+import requests
 
-# Get API key from Hugging Face secrets
-genai.configure(api_key=os.environ["GEMINI_API_KEY"])
-model = genai.GenerativeModel("gemini-pro")
+# 🔐 API key
+groq_key = "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"  # Replace with your actual Groq key
 
-def gemini_chat(user_input):
+# 🧠 Load Whisper locally
+whisper_model = whisper.load_model("base")  # Use "small"/"medium"/"large" for better results
+
+# 💬 Chat using Groq
+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}]
+
+    headers = {
+        "Authorization": f"Bearer {groq_key}",
+        "Content-Type": "application/json"
+    }
+
+    payload = {
+        "model": "llama3-70b-8192",
+        "messages": messages
+    }
+
+    response = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, json=payload)
+    reply = response.json()["choices"][0]["message"]["content"]
+
+    history += [{"role": "user", "content": message}, {"role": "assistant", "content": reply}]
+    return "", history, history
+
+# 🎙️ Transcribe voice to text
+def transcribe_audio(audio_path):
+    if audio_path is None or not os.path.exists(audio_path):
+        return "⚠️ No audio recorded."
+    try:
+        temp_wav = f"{uuid.uuid4()}.wav"
+        AudioSegment.from_file(audio_path).export(temp_wav, format="wav")
+        result = whisper_model.transcribe(temp_wav)
+        os.remove(temp_wav)
+        return result["text"]
+    except Exception as e:
+        return f"❌ Transcription error: {e}"
+
+# 💾 Save/load
+def save_session(history):
+    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])
+    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]}`"
+
+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:
-        if not user_input.strip():
-            return "Please type something."
-        response = model.generate_content(user_input)
-        return response.text
+        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"⚠️ Error: {str(e)}"
+        return [], [], f"❌ Load error: {e}"
 
+# 🌙 Gradio UI
 with gr.Blocks(css="""
-    .gradio-container { max-width: 700px; margin: auto; font-family: 'Segoe UI', sans-serif; }
-    .title { font-size: 2em; color: white; background-color: #1f2937; padding: 20px; border-radius: 12px; text-align: center; margin-bottom: 10px; }
-    .textbox { border-radius: 12px; }
+body {background-color: #111 !important; color: white;}
+.gr-chatbot-message {background-color: #222 !important; color: white !important; border-radius: 6px;}
+input[type='text'], textarea, select, .gr-textbox {background-color: #222 !important; color: white !important;}
+.gr-button {background-color: #007acc !important; color: white !important;}
 """) as demo:
-    
-    gr.Markdown("<div class='title'>🤖 Welcome to Jaweria'sBot</div>")
-    
+
+    state = gr.State([])
+
+    gr.Markdown("<h1 style='text-align:center; color:#00ccff;'>✨ JAWERIA'SBOT 🤖</h1>")
+    gr.Markdown("<div style='text-align:center;'>Speak or type — your assistant listens and replies with text 💬</div>")
+
+    chatbot = gr.Chatbot(type="messages", height=350)
+    chat_input = gr.Textbox(label="💬 Message", placeholder="Type or speak your message...")
+
+    send_btn = gr.Button("Send 🚀")
+
+    with gr.Row():
+        voice_input = gr.Audio(label="🎤 Speak", type="filepath", interactive=True)
+        voice_btn = gr.Button("🎙️ Transcribe to Text")
+
     with gr.Row():
-        user_input = gr.Textbox(placeholder="Type your message here...", lines=2, label="You", elem_classes=["textbox"])
-    
-    submit_btn = gr.Button("Send", variant="primary")
-    response_output = gr.Textbox(label="Jaweria'sBot", interactive=False, lines=6, elem_classes=["textbox"])
+        new_chat_btn = gr.Button("🆕 New")
+        save_btn = gr.Button("💾 Save")
+        saved_dropdown = gr.Dropdown(label="📂 Load Saved", choices=list_saved_files())
+        load_btn = gr.Button("📥 Load")
+        save_msg = gr.Markdown()
+        load_msg = gr.Markdown()
 
-    submit_btn.click(fn=gemini_chat, inputs=user_input, outputs=response_output)
+    # 🔗 Bind actions
+    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])
+    voice_btn.click(transcribe_audio, inputs=[voice_input], outputs=[chat_input])
+    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()