Update app.py

app.py

@@ -1,19 +1,23 @@
-import os
 import gradio as gr
+from together import Together
 from gtts import gTTS
 import tempfile
-from together import Together
+import os
 
-# Load API key from Hugging Face Secrets
+# Get your Together API key from Hugging Face secret
 TOGETHER_API_KEY = os.getenv("TOGETHER_API_KEY")
-client = Together(api_key=os.getenv("TOGETHER_API_KEY"))
 
+# Initialize Together client
+client = Together(api_key=TOGETHER_API_KEY)
+
+# Text-to-speech with gTTS
 def speak_text(text):
     tts = gTTS(text)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as fp:
         tts.save(fp.name)
         return fp.name
 
+# Main chat function
 def respond(message, history):
     history = history or []
     prompt = ""
@@ -21,6 +25,7 @@ def respond(message, history):
         prompt += f"<|user|> {user_msg}\n<|assistant|> {bot_msg}\n"
     prompt += f"<|user|> {message}\n<|assistant|>"
 
+    # Call Together API
     response = client.chat.completions.create(
         model="meta-llama/Llama-3-8B-Instruct",
         messages=[{"role": "user", "content": prompt}],
@@ -29,14 +34,16 @@ def respond(message, history):
         top_p=0.95
     )
 
-    answer = response.choices[0].message.content.strip()
-    history.append((message, answer))
-    audio_path = speak_text(answer)
+    reply = response.choices[0].message.content.strip()
+    history.append((message, reply))
+    audio_path = speak_text(reply)
     return history, history, audio_path
 
+# Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("# Sambit AI 🤖")
-    chatbot = gr.Chatbot([], label="Chat with AI", type="messages")
+    gr.Markdown("## Sambit AI 🤖 — Powered by Together & LLaMA 3")
+
+    chatbot = gr.Chatbot(label="Chat with AI")
     msg = gr.Textbox(label="Type your message here")
     audio_output = gr.Audio(label="AI Voice", interactive=False)
     clear = gr.Button("Clear Chat")