Update app.py

app.py

@@ -1,30 +1,155 @@
+# import streamlit as st
+# import pandas as pd
+# import os
+# from fpdf import FPDF
+# import datetime
+
+# DATA_FILE = "study_data.csv"
+# CALENDAR_FILE = "calendar_data.csv"
+
+# # Load or initialize data
+# def load_data():
+#     if os.path.exists(DATA_FILE):
+#         return pd.read_csv(DATA_FILE)
+#     else:
+#         return pd.DataFrame(columns=["Subject", "Topic", "Goal", "Progress (%)"])
+
+# def save_data(df):
+#     df.to_csv(DATA_FILE, index=False)
+
+# def load_calendar():
+#     if os.path.exists(CALENDAR_FILE):
+#         return pd.read_csv(CALENDAR_FILE)
+#     else:
+#         return pd.DataFrame(columns=["Day", "Planned Topic"])
+
+# def save_calendar(df):
+#     df.to_csv(CALENDAR_FILE, index=False)
+
+# def generate_pdf(df):
+#     pdf = FPDF()
+#     pdf.add_page()
+#     pdf.set_font("Arial", size=12)
+#     pdf.cell(200, 10, txt="Study Plan Report", ln=True, align='C')
+#     pdf.ln(10)
+
+#     for i, row in df.iterrows():
+#         pdf.multi_cell(0, 10, f"Subject: {row['Subject']}\nTopic: {row['Topic']}\nGoal: {row['Goal']}\nProgress: {row['Progress (%)']}%\n")
+#         pdf.ln(1)
+
+#     file_path = "study_report.pdf"
+#     pdf.output(file_path)
+#     return file_path
+
+# def main():
+#     st.set_page_config(page_title="Study Planner", layout="wide")
+#     st.title("📚 Course Organizer & Study Planner")
+
+#     df = load_data()
+#     calendar_df = load_calendar()
+
+#     with st.sidebar:
+#         st.header("➕ Add Study Task")
+#         subject = st.text_input("Subject")
+#         topic = st.text_input("Topic")
+#         goal = st.text_area("Goal Description")
+#         progress = st.slider("Progress (%)", 0, 100, 0)
+
+#         if st.button("Add Task"):
+#             if subject and topic:
+#                 new_row = {"Subject": subject, "Topic": topic, "Goal": goal, "Progress (%)": progress}
+#                 df = pd.concat([df, pd.DataFrame([new_row])], ignore_index=True)
+#                 save_data(df)
+#                 st.success("Task added!")
+#             else:
+#                 st.error("Subject and Topic are required.")
+
+#     st.subheader("🎯 Study Summary by Subject")
+#     if not df.empty:
+#         subjects = df["Subject"].unique()
+#         cols = st.columns(len(subjects))
+#         colors = ["#fca311", "#a1c181", "#f28482", "#8ecae6", "#ffb703"]
+
+#         for i, subject in enumerate(subjects):
+#             sub_df = df[df["Subject"] == subject]
+#             avg_progress = int(sub_df["Progress (%)"].mean())
+#             with cols[i % len(cols)]:
+#                 st.markdown(f"""
+#                     <div style='padding: 1em; border-radius: 10px; background-color: {colors[i % len(colors)]}; color: black'>
+#                         <h4>{subject}</h4>
+#                         <p>Topics: {len(sub_df)}</p>
+#                         <p>Avg. Progress: {avg_progress}%</p>
+#                     </div>
+#                 """, unsafe_allow_html=True)
+
+#     st.subheader("📅 Weekly Study Calendar")
+#     days = ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"]
+#     calendar_data = []
+#     for day in days:
+#         topic = st.text_input(f"{day}", value=calendar_df.set_index("Day").get("Planned Topic", {}).get(day, ""))
+#         calendar_data.append({"Day": day, "Planned Topic": topic})
+
+#     if st.button("Save Weekly Plan"):
+#         calendar_df = pd.DataFrame(calendar_data)
+#         save_calendar(calendar_df)
+#         st.success("Weekly calendar saved.")
+
+#     st.markdown("---")
+#     st.subheader("📋 All Study Tasks (Editable)")
+#     if not df.empty:
+#         edited_df = st.data_editor(df, num_rows="dynamic", use_container_width=True)
+#         save_data(edited_df)
+
+#         st.markdown("---")
+#         st.subheader("📊 Progress Summary")
+#         summary = edited_df.groupby("Subject")["Progress (%)"].mean().reset_index()
+#         st.bar_chart(summary.set_index("Subject"))
+
+#         st.markdown("---")
+#         st.subheader("📤 Download Study Plan Report")
+#         if st.button("Generate PDF Report"):
+#             file_path = generate_pdf(edited_df)
+#             with open(file_path, "rb") as f:
+#                 st.download_button("📄 Download PDF", f, file_name="study_plan_report.pdf")
+#     else:
+#         st.info("No tasks added yet.")
+
+# if __name__ == "__main__":
+#     main()
+
+
 import streamlit as st
 import pandas as pd
 import os
 from fpdf import FPDF
 import datetime
+import io
 
-DATA_FILE = "study_data.csv"
-CALENDAR_FILE = "calendar_data.csv"
+DEFAULT_FILE = "study_data.csv"
+DEFAULT_CALENDAR = "calendar_data.csv"
 
-# Load or initialize data
-def load_data():
-    if os.path.exists(DATA_FILE):
-        return pd.read_csv(DATA_FILE)
+# Load default or uploaded data
+def load_data(uploaded_file):
+    if uploaded_file:
+        return pd.read_csv(uploaded_file)
+    elif os.path.exists(DEFAULT_FILE):
+        return pd.read_csv(DEFAULT_FILE)
     else:
         return pd.DataFrame(columns=["Subject", "Topic", "Goal", "Progress (%)"])
 
 def save_data(df):
-    df.to_csv(DATA_FILE, index=False)
+    df.to_csv(DEFAULT_FILE, index=False)
 
-def load_calendar():
-    if os.path.exists(CALENDAR_FILE):
-        return pd.read_csv(CALENDAR_FILE)
+def load_calendar(uploaded_file):
+    if uploaded_file:
+        return pd.read_csv(uploaded_file)
+    elif os.path.exists(DEFAULT_CALENDAR):
+        return pd.read_csv(DEFAULT_CALENDAR)
     else:
         return pd.DataFrame(columns=["Day", "Planned Topic"])
 
 def save_calendar(df):
-    df.to_csv(CALENDAR_FILE, index=False)
+    df.to_csv(DEFAULT_CALENDAR, index=False)
 
 def generate_pdf(df):
     pdf = FPDF()
@@ -45,8 +170,13 @@ def main():
     st.set_page_config(page_title="Study Planner", layout="wide")
     st.title("📚 Course Organizer & Study Planner")
 
-    df = load_data()
-    calendar_df = load_calendar()
+    st.sidebar.markdown("---")
+    st.sidebar.header("📂 Upload Your Study Plan")
+    uploaded_plan = st.sidebar.file_uploader("Upload Study CSV", type=["csv"])
+    uploaded_calendar = st.sidebar.file_uploader("Upload Calendar CSV", type=["csv"])
+
+    df = load_data(uploaded_plan)
+    calendar_df = load_calendar(uploaded_calendar)
 
     with st.sidebar:
         st.header("➕ Add Study Task")
@@ -106,13 +236,18 @@ def main():
         st.bar_chart(summary.set_index("Subject"))
 
         st.markdown("---")
-        st.subheader("📤 Download Study Plan Report")
-        if st.button("Generate PDF Report"):
+        st.subheader("📤 Export Options")
+
+        csv = df.to_csv(index=False).encode("utf-8")
+        st.download_button("⬇️ Download CSV", data=csv, file_name="study_data.csv", mime="text/csv")
+
+        if st.button("📄 Generate PDF Report"):
             file_path = generate_pdf(edited_df)
             with open(file_path, "rb") as f:
-                st.download_button("📄 Download PDF", f, file_name="study_plan_report.pdf")
+                st.download_button("📥 Download PDF", f, file_name="study_plan_report.pdf")
     else:
         st.info("No tasks added yet.")
 
 if __name__ == "__main__":
     main()
+