Update app.py

app.py

@@ -1,47 +1,29 @@
 import os
 import streamlit as st
-from groq import Groq
+from groq import *  # Import everything from groq, so you can access the chat functionality
 
-# Function to initialize Groq client
-def get_groq_client():
-    api_key = "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"
-    if not api_key:
-        st.error("API key is missing. Please set the GROQ_API_KEY environment variable.")
-        return None
-    return Client(api_key=api_key)
+# Ensure you have the correct API key set in the environment variable
+api_key = "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"
 
-# Function to get chatbot response
-def get_response(client, user_input):
+# Streamlit interface
+st.title("AI Chatbot with Groq")
+
+user_input = st.text_input("Ask something:", "")
+
+if user_input:
     try:
-        response = client.chat.completions.create(
+        # Create the Groq client using the API key
+        client = Groq(api_key=api_key)
+        
+        # Create chat completion with user input
+        chat_completion = client.chat.completions.create(
             messages=[{"role": "user", "content": user_input}],
             model="llama-3.3-70b-versatile"
         )
-        return response['choices'][0]['message']['content']
+        
+        # Display the response
+        st.write("AI's response: ")
+        st.write(chat_completion.choices[0].message.content)
+
     except Exception as e:
         st.error(f"Error: {e}")
-        return None
-
-# Streamlit UI
-def main():
-    st.title("AI Chatbot using Groq")
-
-    # Get the Groq client
-    client = get_groq_client()
-    if not client:
-        return
-
-    # User input
-    user_input = st.text_input("Ask a question:")
-    if st.button("Get Answer"):
-        if user_input:
-            with st.spinner("Thinking..."):
-                answer = get_response(client, user_input)
-            if answer:
-                st.success("Response:")
-                st.write(answer)
-        else:
-            st.warning("Please enter a message.")
-
-if __name__ == "__main__":
-    main()