Update app.py

app.py

@@ -2,34 +2,30 @@ import os
 import gradio as gr
 from groq import Groq
 
-# Set up Groq API client (ensure GROQ_API_KEY is set in your environment or as a Hugging Face secret for deployment)
-apikey = os.getenv("apikey")
-print(f"API Key: {apikey}")  # Debugging line, remove it before pushing to Hugging Face
-
+# Set up the Groq API client
+apikey = os.getenv("apikey")  # Ensure your API key is set as a secret
+if not apikey:
+    raise ValueError("API Key is not set. Add it in the Secrets tab.")
 client = Groq(api_key=apikey)
 
-# Function to interact with the LLM using Groq's API
-def chatbot(messages):
+# Function to interact with the LLM
+def chatbot(user_input):
     try:
-        # Ensure the messages list is not empty
-        if not messages:
-            messages = [("System", "Hello! How can I assist you today?")]
-
-        user_input = messages[-1][0]  # Last user input message
-
-        if not user_input.strip():  # Check for empty input
-            messages.append(("System", "It seems like you may have accidentally sent an empty message. Please rephrase."))
-            return messages
-
-        # Sending request to Groq API
         chat_completion = client.chat.completions.create(
             messages=[{"role": "user", "content": user_input}],
-            model="llama3-8b-8192",  # Replace with the correct model name
+            model="llama3-8b-8192",
         )
-
-        response = chat_completion.choices[0].message.content
-        messages.append((user_input, response))  # Append user input and bot response as a tuple
-        return messages
+        return chat_completion.choices[0].message.content
     except Exception as e:
-        # Capture the specific error message for debugging
-        print
+        return f"An error occurred: {str(e)}"
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=chatbot,
+    inputs=gr.Textbox(label="Your Message", placeholder="Type your message here..."),
+    outputs=gr.Textbox(label="Response"),
+    title="Groq LLM Chatbot",
+    description="Interact with a powerful LLM via Groq's API."
+)
+
+iface.launch()