Update app.py

app.py

@@ -1,30 +1,39 @@
-import os
+import gradio as gr
 import openai
+import os
 
-# Set the base URL to the Groq endpoint
+# Set the base URL for the Groq API
 openai.api_base = "https://api.groq.com/openai/v1"
-openai.api_key = os.getenv("GROQ_API_KEY")  # Fetch the API key from environment variable
+openai.api_key = os.getenv("GROQ_API_KEY")  # Fetch API key from environment variable
 
 def chat_function(message):
-    # Prepare the request payload
+    # Define the request payload
     payload = {
-        "model": "mixtral-8x7b-32768",  # Adjust the model name as per Groq's available models
+        "model": "mixtral-8x7b-32768",  # Specify your model here (e.g., Groq's supported model)
         "messages": [{"role": "user", "content": message}],
         "max_tokens": 150,
         "temperature": 1
     }
-
+    
     try:
-        # Make the API request to the Groq server
+        # Send the request to the Groq API and get the response
         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)
+# Set up the Gradio interface
+gr.Interface(
+    fn=chat_function,  # The function that handles user input
+    inputs=gr.Textbox(placeholder="Ask something..."),  # User input text box
+    outputs="text",  # Output text box for the response
+    title="Groq-Gradio Chat",  # Interface title
+    theme="huggingface",  # Use a valid Gradio 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"
+    ]  # Example prompts
+).launch()