Update app.py

app.py

@@ -1,17 +1,41 @@
 import os
-
+import streamlit as st
 from groq import Groq
 
+# Set your Groq API key (preferably via environment variable for security)
+GROQ_API_KEY = os.environ.get("GROQ_API_KEY") or "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"  # Fallback only for testing
+
 # Initialize Groq client
-client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+try:
+    client = Groq(api_key=GROQ_API_KEY)
+except Exception as e:
+    st.error(f"Error initializing Groq client: {e}")
+    st.stop()  # Stop execution if client initialization fails
 
 def get_response(prompt):
-  # Create chat completion request
-  chat_completion = client.chat.completions.create(
-      messages=[{"role": "user", "content": prompt}],
-      model="your_desired_model"  # Replace with your preferred model
-  )
-  # Return the first choice's message content
-  return chat_completion.choices[0].message.content
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[{"role": "user", "content": prompt}],
+            model="llama-3.3-70b-versatile" # Or any other available model
+        )
+        return chat_completion.choices[0].message.content
+    except Exception as e:
+        st.error(f"Error generating response: {e}")
+        return "An error occurred while generating the response."
+
+# Streamlit app layout
+st.title("Interactive Chatbot")
+
+user_input = st.text_input("You: ", key="user_input")
+
+if user_input:
+    bot_response = get_response(user_input)
+    st.write("Bot: ", bot_response)
 
-# Rest of your Streamlit app code using get_response function
+if __name__ == "__main__":
+    st.sidebar.title("Settings")
+    st.sidebar.write("API Key Source: " + ("Environment Variable" if os.environ.get("GROQ_API_KEY") else "Directly in Code (Not Recommended)"))
+    if os.environ.get("GROQ_API_KEY"):
+        st.sidebar.success("Groq Client Initialized Successfully!")
+    else:
+        st.sidebar.warning("Using API Key Directly in Code (Not Recommended). Set GROQ_API_KEY Environment Variable for Security.")