Update app.py

app.py

@@ -1,36 +1,27 @@
 import os
-import gradio as gr
 import openai
+import gradio as gr
 import groq_gradio
 
-# Set up the API key
+# Set the environment variable for the Groq API key
 os.environ["GROQ_API_KEY"] = "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"
 
-# Initialize the OpenAI client to use the Groq API endpoint
+# Initialize OpenAI client with Groq's API base URL
 client = openai.OpenAI(
     base_url="https://api.groq.com/openai/v1",
-    api_key=os.environ.get("GROQ_API_KEY")
+    api_key=os.environ.get("GROQ_API_KEY")  # Use the environment variable to fetch the API key
 )
 
-# Define a function to generate a response using Groq API
-def generate_response(prompt):
-    response = openai.Completion.create(
-        model="llama-3.2-3b-preview",
-        prompt=prompt,
-        max_tokens=100
-    )
-    return response.choices[0].text.strip()
-
-# Define the Gradio interface to use the Groq model through Gradio and Groq API
+# Gradio interface setup
 gr.Interface(
-    fn=generate_response,
-    inputs=gr.Textbox(label="Enter your prompt here:"),
-    outputs=gr.Textbox(label="Response from Groq model:"),
+    fn=groq_gradio.registry,  # Assuming the function used for the Groq model registry
+    inputs=gr.inputs.Textbox(placeholder="Ask something..."),
+    outputs="text",
     title="Groq-Gradio Chat",
     theme="upsatwal/mlsc_tiet",
     examples=[
         "Tell me a short story about a puppy",
-        "Write a 14-line poem about traveling 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"
     ]