Update app.py

app.py

@@ -1,55 +1,30 @@
-import gradio as gr
-import requests
-import json
 import os
+import openai
 
-# Ensure the environment variable is set securely (store your API key as an environment variable)
-api_key = os.getenv("GROQ_API_KEY")
-url = "https://api.groq.com/openai/v1/chat/completions"
+# Set the base URL to the Groq endpoint
+openai.api_base = "https://api.groq.com/openai/v1"
+openai.api_key = os.getenv("GROQ_API_KEY")  # Fetch the API key from environment variable
 
 def chat_function(message):
-    # Define the request payload
+    # Prepare the request payload
     payload = {
-        "model": "mixtral-8x7b-32768",
-        "messages": [
-            {"role": "system", "content": "You are a helpful assistant."},
-            {"role": "user", "content": message}
-        ],
+        "model": "mixtral-8x7b-32768",  # Adjust the model name as per Groq's available models
+        "messages": [{"role": "user", "content": message}],
         "max_tokens": 150,
-        "temperature": 1,
-        "presence_penalty": 0,
-        "frequency_penalty": 0
+        "temperature": 1
     }
 
-    # Set the headers including your API key
-    headers = {
-        "Authorization": f"Bearer {api_key}",
-        "Content-Type": "application/json"
-    }
-
-    # Make the POST request to the API
-    response = requests.post(url, headers=headers, data=json.dumps(payload))
-
-    # Check if the request was successful
-    if response.status_code == 200:
-        data = response.json()
-        return data["choices"][0]["message"]["content"]  # Return the model's response
-    else:
-        return f"Error: {response.status_code}, {response.text}"
-
-# Set up Gradio interface
-gr.Interface(
-    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",  # Title of the interface
-    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",
-        "What are the wonders of the world?",
-        "List the countries in Africa and their capitals"
-    ]
-).launch()
-
-
+    try:
+        # Make the API request to the Groq server
+        response = openai.ChatCompletion.create(**payload)
+        
+        # Check if the response is successful and return the model's response
+        return response["choices"][0]["message"]["content"]
+    
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Example usage
+message = "Tell me about the wonders of the world"
+response = chat_function(message)
+print(response)