Spaces:
Sleeping
Sleeping
| import os | |
| import gradio as gr | |
| from groq import Groq | |
| # Get API key from environment variable set in Hugging Face secrets | |
| GROQ_API_KEY = os.getenv("GROQ_API_KEY") | |
| # Initialize the Groq client | |
| client = Groq(api_key=GROQ_API_KEY) | |
| # Chatbot logic | |
| def chat_with_groq(message): | |
| try: | |
| chat_completion = client.chat.completions.create( | |
| messages=[{"role": "user", "content": message}], | |
| model="llama-3.3-70b-versatile", | |
| stream=False, | |
| ) | |
| return chat_completion.choices[0].message.content | |
| except Exception as e: | |
| return f"Error: {str(e)}" | |
| # Gradio interface | |
| def chatbot_interface(user_input): | |
| return chat_with_groq(user_input) | |
| iface = gr.Interface( | |
| fn=chatbot_interface, | |
| inputs="text", | |
| outputs="text", | |
| title="🤖 Your Conversational Agent", | |
| description="Chat with the Groq-powered chatbot." | |
| ) | |
| iface.launch() | |