Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import openai
 import gradio as gr
-import groq_gradio
+import groq_gradio  # Ensure this module exists and contains necessary functions
 
 # Set the environment variable for the Groq API key
 os.environ["GROQ_API_KEY"] = "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"
@@ -12,16 +12,25 @@ client = openai.OpenAI(
     api_key=os.environ.get("GROQ_API_KEY")  # Use the environment variable to fetch the API key
 )
 
+# Define the chat function
+def chat_function(message):
+    # Call Groq API for completion based on the given message
+    response = client.chat_completions.create(
+        model="llama3-8b-8192",
+        messages=[{"role": "user", "content": message}]
+    )
+    return response.choices[0].message["content"]
+
 # Gradio interface setup
 gr.Interface(
-    fn=groq_gradio.registry,  # Assuming the function used for the Groq model registry
-    inputs=gr.Textbox(placeholder="Ask something..."),  # Updated component usage
-    outputs="text",
+    fn=chat_function,  # Function that handles the chatbot
+    inputs=gr.Textbox(placeholder="Ask something..."),  # Input for user message
+    outputs="text",  # Output as text
     title="Groq-Gradio Chat",
-    theme="upsatwal/mlsc_tiet",
+    theme="upsatwal/mlsc_tiet",  # Ensure theme is valid
     examples=[
         "Tell me a short story about a puppy",
-        "Write a 14 line poem about travelling in Shakespeare style",
+        "Write a 14-line poem about travelling in Shakespeare style",
         "What are the wonders of the world?",
         "List the countries in Africa and their capitals"
     ]