Update app.py

app.py

@@ -1,4 +1,4 @@
-# Tech Trends Curator with Audio (Hugging Face Compatible)
+# Tech Trends Curator Chatbot with Audio (Hugging Face Compatible)
 
 import os
 import uuid
@@ -10,7 +10,7 @@ from gtts import gTTS
 GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
 
 # Function to query Groq LLM via REST API
-def query_groq(user_input):
+def query_groq(messages):
     url = "https://api.groq.com/openai/v1/chat/completions"
     headers = {
         "Authorization": f"Bearer {GROQ_API_KEY}",
@@ -19,47 +19,48 @@ def query_groq(user_input):
 
     data = {
         "model": "llama3-8b-8192",
-        "messages": [
-            {"role": "system", "content": (
-                "You are a tech trend curator. Summarize the latest AI tools, GitHub projects, "
-                "and startup news in an engaging tone. If asked for fake/funny trends, be creative and witty."
-            )},
-            {"role": "user", "content": user_input}
-        ]
+        "messages": messages
     }
 
     response = requests.post(url, headers=headers, json=data)
     result = response.json()
     return result["choices"][0]["message"]["content"]
 
-# Main function for chatbot + audio generation
-def get_tech_trends_with_audio(user_input, history):
-    output_text = query_groq(user_input)
+# Chatbot function with audio output
+def chat_with_audio(user_input, history):
+    system_prompt = {
+        "role": "system",
+        "content": (
+            "You are a tech trend curator. Summarize the latest AI tools, GitHub projects, "
+            "and startup news in an engaging tone. If asked for fake/funny trends, be creative and witty."
+        )
+    }
+
+    messages = [system_prompt] + [{"role": "user", "content": inp} if i % 2 == 0 else {"role": "assistant", "content": inp} for i, inp in enumerate(sum(history, []))]
+    messages.append({"role": "user", "content": user_input})
+
+    response_text = query_groq(messages)
 
     # Generate audio
-    tts = gTTS(output_text)
-    filename = f"/tmp/{uuid.uuid4()}.mp3"
-    tts.save(filename)
+    tts = gTTS(response_text)
+    audio_file = f"/tmp/{uuid.uuid4()}.mp3"
+    tts.save(audio_file)
+
+    return response_text, history + [[user_input, response_text]], audio_file
+
+# Gradio ChatInterface with audio output
+with gr.Blocks(theme="soft") as demo:
+    gr.Markdown("# 📈 Tech Trends Curator\nChat about trending AI tools, GitHub projects, and startup news — with audio!")
+    chatbot = gr.Chatbot(height=400)
+    msg = gr.Textbox(placeholder="Ask for top trends, tools, or fake news")
+    audio = gr.Audio(label="Audio Summary")
+    state = gr.State([])
 
-    return output_text, filename
+    def user_message(user_input, history):
+        reply, new_history, audio_path = chat_with_audio(user_input, history)
+        return reply, new_history, audio_path
 
-# Gradio interface
-demo = gr.Interface(
-    fn=get_tech_trends_with_audio,
-    inputs=gr.Textbox(placeholder="Ask for top trends, tools, or fake news", label="Your Question"),
-    outputs=[
-        gr.Textbox(label="Trend Summary"),
-        gr.Audio(label="Audio Summary")
-    ],
-    title="📈 Tech Trends Curator",
-    description="Get AI, GitHub, and startup trend summaries — now with audio!",
-    examples=[
-        ["Top 3 AI tools this week"],
-        ["Funny fake trends in tech"],
-        ["What’s hot on GitHub in AI?"]
-    ],
-    theme="soft"
-)
+    msg.submit(user_message, [msg, state], [chatbot, state, audio])
 
-# Launch app
+# Launch the app
 demo.launch(share=True)