Update app.py

app.py

@@ -1,22 +1,38 @@
-import os
-import openai
 import gradio as gr
+import requests
+import json
 
-os.environ["OPENAI_API_KEY"] = os.environ.get("GROQ_API_KEY")
+# Set the API key
+api_key = "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"
 
-client = openai.OpenAI(
-    base_url="https://api.groq.com/v1"
-)
+# Define the endpoint URL
+url = "https://api.groq.com/openai/v1/chat/completions"
 
-# Define the chat function
+# Define the request payload
 def chat_function(message):
-    # Call Groq API for completion using the correct method: completions.create
-    response = client.completions.create(
-        model="lama3-8b-8192",  # Specify the model
-        prompt=message,  # Send the user's message as the prompt
-        max_tokens=150  # Optionally, specify the maximum number of tokens in the response
-    )
-    return response.choices[0].text.strip()  # Extract and return the text of the response
+    payload = {
+        "model": "lama3-8b-8192", 
+        "messages": [
+            {"role": "system", "content": "You are a helpful assistant."},
+            {"role": "user", "content": message}
+        ],
+        "max_tokens": 150,
+        "temperature": 1,
+        "presence_penalty": 0,
+        "frequency_penalty": 0
+    }
+    
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+    
+    response = requests.post(url, headers=headers, data=json.dumps(payload))
+    if response.status_code == 200:
+        data = response.json()
+        return data["choices"][0]["message"]["content"]
+    else:
+        return f"Error: {response.status_code}, {response.text}"
 
 # Gradio interface setup
 gr.Interface(
@@ -24,11 +40,11 @@ gr.Interface(
     inputs=gr.Textbox(placeholder="Ask something..."),  # Input for user message
     outputs="text",  # Output as text
     title="Groq-Gradio Chat",
-    theme="upsatwal/mlsc_tiet",  # Ensure theme is valid
+    theme="gradio",  # Use a valid theme
     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()
+).launch()