Update app.py

app.py

@@ -1,5 +1,5 @@
-import gradio as gr
 import os
+import gradio as gr
 import pdfplumber
 import docx
 import pandas as pd
@@ -9,12 +9,18 @@ import base64
 import whisper
 from openai import OpenAI
 
-# Setup GROQ
-client = OpenAI(api_key=os.environ["GROQ_API_KEY"], base_url="https://api.groq.com/openai/v1")
-
 # Load Whisper model
 whisper_model = whisper.load_model("base")
 
+# Load Groq API key
+GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
+
+# Initialize OpenAI client with Groq base URL
+client = OpenAI(
+    api_key=GROQ_API_KEY,
+    base_url="https://api.groq.com/openai/v1"
+)
+
 def extract_text_from_file(file):
     if file.name.endswith(".pdf"):
         with pdfplumber.open(file.name) as pdf:
@@ -51,7 +57,8 @@ def generate_reply(history):
         messages=messages,
         temperature=0.7
     )
-    return response.choices[0].message.content
+    reply = response.choices[0].message.content
+    return reply
 
 def respond(message, history):
     reply = generate_reply(history + [[message, ""]])
@@ -75,7 +82,7 @@ with gr.Blocks(css="body { background-color: white; color: black }") as demo:
     chatbot = gr.Chatbot(label="Chat", elem_id="chatbox", height=450, type="messages")
 
     with gr.Row():
-        txt = gr.Textbox(placeholder="Type your message or review the file/audio content here…", scale=5, show_label=False)
+        txt = gr.Textbox(placeholder="Type a message or edit transcribed/file content here...", scale=5, show_label=False)
         send_btn = gr.Button("Send", scale=1)
 
     with gr.Row():
@@ -84,10 +91,7 @@ with gr.Blocks(css="body { background-color: white; color: black }") as demo:
 
     history = gr.State([])
 
-    # Message only sent when Send is clicked
     send_btn.click(respond, [txt, history], [chatbot, txt])
-
-    # File and audio just modify the message box
     upload_btn.change(handle_file_upload, [upload_btn, txt], txt)
     audio_in.change(handle_audio_upload, [audio_in, txt], txt)