Update app.py

app.py

@@ -1,102 +1,33 @@
-def main():
-    import streamlit as st
-    import pandas as pd
-    import plotly.express as px
-    from datetime import datetime, timedelta
-    import json
-    import os
+import streamlit as st
+import os
+from groq import Groq
 
-    st.set_page_config(page_title="Habit & To-Do Tracker", layout="wide")
-    st.title("📋 Habit & To-Do Tracker")
+# --- Setup Groq client using environment variable ---
+api_key = os.environ.get("GROQ_API_KEY")
+if not api_key:
+    st.error("Error: GROQ_API_KEY not set in environment variables.")
+else:
+    client = Groq(api_key=api_key)
 
-    DATA_FILE = "data.json"
+st.title("Habit & To-Do Tracker with Groq AI")
 
-    # Load data
-    if os.path.exists(DATA_FILE):
-        with open(DATA_FILE, "r") as f:
-            data = json.load(f)
-    else:
-        data = {"tasks": [], "habits": {}}
-
-    # Sidebar: Add Task
-    st.sidebar.header("➕ Add Task")
-    task_name = st.sidebar.text_input("Task Name")
-    task_completed = st.sidebar.checkbox("Completed")
-    if st.sidebar.button("Add Task") and task_name.strip():
-        today = datetime.today().strftime("%Y-%m-%d")
-        data["tasks"].append({
-            "Task": task_name,
-            "Completed": task_completed,
-            "Date": today
-        })
-        with open(DATA_FILE, "w") as f:
-            json.dump(data, f)
-        st.sidebar.success("Task added!")
-
-    # Sidebar: Track Habit
-    st.sidebar.header("🔥 Track Habit")
-    habit_name = st.sidebar.text_input("Habit Name")
-    if st.sidebar.button("Mark Habit Done") and habit_name.strip():
-        today = datetime.today().strftime("%Y-%m-%d")
-        if habit_name not in data["habits"]:
-            data["habits"][habit_name] = []
-        if today not in data["habits"][habit_name]:
-            data["habits"][habit_name].append(today)
-            with open(DATA_FILE, "w") as f:
-                json.dump(data, f)
-            st.sidebar.success("Habit marked for today!")
-        else:
-            st.sidebar.warning("Already marked today.")
-
-    # Display Tasks
-    st.subheader("📌 Tasks")
-    if data["tasks"]:
-        task_df = pd.DataFrame(data["tasks"])
-        st.dataframe(task_df, use_container_width=True)
-    else:
-        st.info("No tasks yet.")
+st.markdown("""
+Welcome! Enter a prompt and get insights from the Groq API.
+""")
 
-    # Display Habits
-    st.subheader("🔥 Habits")
-    habit_data = []
-    for habit, dates in data["habits"].items():
-        dates_sorted = sorted(dates)
-        streak = 0
-        today = datetime.today().date()
-        for i in range(21):
-            day = today - timedelta(days=i)
-            if day.strftime("%Y-%m-%d") in dates_sorted:
-                streak += 1
-            else:
-                break
-        habit_type = "🌟 Permanent Habit" if streak >= 21 else "⏳ Temporary Habit"
-        habit_data.append({"Habit": habit, "Days Followed": len(dates), "Streak": streak, "Type": habit_type})
+# --- User input ---
+user_prompt = st.text_area("Enter your prompt:", "")
 
-    if habit_data:
-        habit_df = pd.DataFrame(habit_data)
-        st.dataframe(habit_df, use_container_width=True)
-        st.subheader("📊 Habit Analytics")
-        st.plotly_chart(px.bar(habit_df, x="Habit", y="Days Followed", color="Type"), use_container_width=True)
-        st.plotly_chart(px.pie(habit_df, names="Habit", values="Days Followed"), use_container_width=True)
+if st.button("Submit"):
+    if not user_prompt.strip():
+        st.warning("Please enter a prompt!")
     else:
-        st.info("No habits tracked yet.")
-
-    # Daily Report
-    st.subheader("📑 Daily Report")
-    today_str = datetime.today().strftime("%Y-%m-%d")
-    today_tasks = [t for t in data["tasks"] if t["Date"] == today_str]
-    report_df = pd.DataFrame([{
-        "Date": today_str,
-        "Total Tasks Today": len(today_tasks),
-        "Completed Tasks Today": len([t for t in today_tasks if t["Completed"]]),
-        "Total Habits": len(data["habits"])
-    }])
-    st.table(report_df)
-
-    # Download CSV
-    csv_data = report_df.to_csv(index=False).encode("utf-8")
-    st.download_button("⬇ Download CSV", csv_data, "daily_report.csv", "text/csv")
-
-# Run the app
-if __name__ == "__main__":
-    main()
+        try:
+            with st.spinner("Processing..."):
+                response = client.chat.completions.create(
+                    messages=[{"role": "user", "content": user_prompt}]
+                )
+                st.subheader("Groq AI Response:")
+                st.write(response.choices[0].message.content)
+        except Exception as e:
+            st.error(f"Error calling Groq API: {e}")