Update app.py

app.py

@@ -71,46 +71,67 @@
 import os
 from groq import Groq
 import gradio as gr
+from PyPDF2 import PdfReader
+import tempfile
 
-# Initialize Groq
+# Groq client
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-# System prompt
 SYSTEM_PROMPT = {
     "role": "system",
-    "content": "You are a helpful AI assistant."
+    "content": "You are a smart, friendly AI assistant."
 }
 
-# Read uploaded file
+# -------- FILE HANDLING --------
 def read_file(file):
     if file is None:
         return ""
+
     try:
-        return file.read().decode("utf-8", errors="ignore")[:2000]
+        # TEXT FILE
+        if file.name.endswith(".txt"):
+            return file.read().decode("utf-8", errors="ignore")[:3000]
+
+        # PDF FILE
+        elif file.name.endswith(".pdf"):
+            with tempfile.NamedTemporaryFile(delete=False) as tmp:
+                tmp.write(file.read())
+                tmp_path = tmp.name
+
+            reader = PdfReader(tmp_path)
+            text = ""
+
+            for page in reader.pages:
+                text += page.extract_text() or ""
+
+            return text[:3000]
+
     except:
         return ""
 
-# Chatbot function (GRADIO 6 FORMAT)
+    return ""
+
+# -------- CHAT FUNCTION (STREAMING) --------
 def bot(history, file):
 
     messages = [SYSTEM_PROMPT]
 
-    # Add file content
+    # File content
     file_text = read_file(file)
     if file_text:
         messages.append({
             "role": "user",
-            "content": f"Here is a document:\n{file_text}"
+            "content": f"Use this document:\n{file_text}"
         })
 
-    # Convert history safely (REMOVE metadata)
+    # History
     for msg in history[:-1]:
         messages.append({
             "role": msg["role"],
             "content": str(msg["content"])
         })
 
-    # Latest user message
+    # Latest user msg
     user_msg = history[-1]["content"]
     messages.append({"role": "user", "content": user_msg})
 
@@ -127,12 +148,10 @@ def bot(history, file):
             if chunk.choices[0].delta.content:
                 response += chunk.choices[0].delta.content
 
-                # Update assistant message
                 history[-1] = {
                     "role": "assistant",
                     "content": response
                 }
-
                 yield history
 
     except Exception as e:
@@ -142,33 +161,55 @@ def bot(history, file):
         }
         yield history
 
+# -------- VOICE INPUT --------
+def speech_to_text(audio):
+    if audio is None:
+        return ""
+    return "Voice input received (speech-to-text can be added here)"
 
-# UI
+# -------- UI --------
 with gr.Blocks() as demo:
 
-    gr.Markdown("# 🚀 Groq AI Chatbot")
-    gr.Markdown("💬 Streaming + 📄 File Upload (Gradio 6.11)")
+    gr.Markdown("# 🚀 Advanced Groq AI Chatbot")
+    gr.Markdown("💬 Chat | 📄 PDF | 🎤 Voice | 🌗 Theme Toggle")
 
     chatbot = gr.Chatbot(height=400)
 
     with gr.Row():
-        msg = gr.Textbox(placeholder="Type your message...", scale=4)
-        file = gr.File(scale=1)
+        msg = gr.Textbox(placeholder="Type message...", scale=4)
+        file = gr.File(file_types=[".pdf", ".txt"], scale=1)
 
-    send = gr.Button("Send 🚀")
+    with gr.Row():
+        audio = gr.Audio(type="filepath", label="🎤 Voice Input")
+        send = gr.Button("Send 🚀")
+
+    # Theme toggle (Dark/Light)
+    theme_toggle = gr.Checkbox(label="🌗 Dark Mode")
+
+    def toggle_theme(dark):
+        return gr.themes.Soft() if not dark else gr.themes.Monochrome()
 
-    # Add user message
+    theme_toggle.change(toggle_theme, inputs=theme_toggle, outputs=[])
+
+    # User input
     def user_input(message, history):
         history = history or []
-        history.append({
-            "role": "user",
-            "content": message
-        })
+        history.append({"role": "user", "content": message})
         return "", history
 
+    # Voice input -> text
+    def voice_input(audio, history):
+        text = speech_to_text(audio)
+        if text:
+            history.append({"role": "user", "content": text})
+        return history
+
     send.click(user_input, [msg, chatbot], [msg, chatbot]) \
         .then(bot, [chatbot, file], chatbot)
 
-# Launch (theme here in v6)
+    audio.change(voice_input, [audio, chatbot], chatbot) \
+        .then(bot, [chatbot, file], chatbot)
+
+# Launch
 if __name__ == "__main__":
     demo.launch(theme=gr.themes.Soft())