Update app.py

app.py

@@ -1,15 +1,37 @@
+import os
 import gradio as gr
+import openai
 import groq_gradio
-import os
 
+# Set up the API key
 os.environ["GROQ_API_KEY"] = "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"
-gr.load(
-    name = "llama-3.2-3b-preview",
-    src = groq_gradio.registry,
-    title = "Groq-Gradio Chat",
-    theme =  "upsatwal/mlsc_tiet",
-    examples = ["Tell me a short story about a puppy",
-                "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"]
-).launch()
+
+# Initialize the OpenAI client to use the Groq API endpoint
+client = openai.OpenAI(
+    base_url="https://api.groq.com/openai/v1",
+    api_key=os.environ.get("GROQ_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
+gr.Interface(
+    fn=generate_response,
+    inputs=gr.Textbox(label="Enter your prompt here:"),
+    outputs=gr.Textbox(label="Response from Groq model:"),
+    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",
+        "What are the wonders of the world?",
+        "List the countries in Africa and their capitals"
+    ]
+).launch()