Create app.py

app.py

@@ -1,17 +1,34 @@
 import os
+import streamlit as st
 from groq import Groq
 
-# Set API key directly for testing (remove this after testing)
-os.environ["GROQ_API_KEY"] = "your_actual_api_key_here"
+# Load API key from environment variables
+api_key = os.environ.get("GROQ_API_KEY")
 
-client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+# Initialize Groq client
+client = Groq(api_key=api_key)
 
-# Test API authentication
-try:
-    chat_completion = client.chat.completions.create(
-        messages=[{"role": "user", "content": "Test message"}],
-        model="llama-3.3-70b-versatile",
-    )
-    print(chat_completion.choices[0].message.content)
-except Exception as e:
-    print("Error:", e)
+# Streamlit app setup
+st.title("AI-powered Routine Assistant")
+
+# Function to get response from the Groq model
+def get_groq_response(user_message):
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[{"role": "user", "content": user_message}],
+            model="llama-3.3-70b-versatile",
+        )
+        return chat_completion.choices[0].message.content
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Display the chatbot's response based on the user's input
+user_input = st.text_input("Ask me about your routine:")
+
+if user_input:
+    # Make sure the question is routine-related, otherwise guide the user
+    if "routine" in user_input.lower() or "schedule" in user_input.lower():
+        response = get_groq_response(user_input)
+        st.write(response)
+    else:
+        st.write("I'm here to help with your study routine. Please ask me about your schedule!")