Update app.py

app.py

@@ -1,19 +1,19 @@
-# Smart Study Planner - Enhanced Version
+# Smart Study Planner - FIXED VERSION
 
 import streamlit as st
 import numpy as np
 import pandas as pd
 
 # -------------------------------
-# PAGE CONFIG (Better UI)
+# PAGE CONFIG
 # -------------------------------
 st.set_page_config(page_title="Smart Study Planner", page_icon="📚", layout="centered")
 
-st.title("📚 Smart Study Planner with AI Prediction")
-st.markdown("Analyze your study habits and improve performance 🚀")
+st.title("📚 Smart Study Planner with Performance Prediction")
+st.markdown("Track your study progress and predict performance smartly 🚀")
 
 # -------------------------------
-# CLASSES (OOP)
+# OOP CLASSES
 # -------------------------------
 
 class Subject:
@@ -51,7 +51,7 @@ class Student:
         marks = [s.marks for s in self.subjects]
 
         if len(hours) > 1:
-            return np.corrcoef(hours, marks)[0][1]  # correlation
+            return np.corrcoef(hours, marks)[0][1]
         return 0
 
     def predict_performance(self):
@@ -59,8 +59,8 @@ class Student:
         predicted = []
 
         for s in self.subjects:
-            improvement = (2 * 4) + (trend * 5)  # smarter logic
-            predicted.append(min(s.marks + improvement, 100))
+            boost = (2 * 4) + (trend * 5)  # smart logic
+            predicted.append(min(s.marks + boost, 100))
 
         return np.mean(predicted) if predicted else 0
 
@@ -82,57 +82,68 @@ class Student:
         return min(self.subjects, key=lambda x: x.marks).name
 
     def suggest_improvement(self):
-        weak = self.weak_subject()
-        return f"Focus more on {weak} and increase study hours by at least 2 hours."
+        return f"Focus more on {self.weak_subject()} and increase study time by 2 hours."
 
 
 # -------------------------------
-# SESSION STATE
+# SESSION STATE FIX (IMPORTANT)
 # -------------------------------
+
 if "student" not in st.session_state:
     st.session_state.student = None
 
+if "student_name" not in st.session_state:
+    st.session_state.student_name = ""
+
 # -------------------------------
-# INPUT SECTION (Styled)
+# INPUT UI
 # -------------------------------
 
-with st.container():
-    st.subheader("👤 Student Details")
+st.subheader("👤 Student Information")
+
+name = st.text_input("Enter Student Name", value=st.session_state.student_name)
 
-    name = st.text_input("Enter Student Name")
+if name:
+    st.session_state.student_name = name
 
-    if name:
+    # FIX: only create once
+    if st.session_state.student is None:
         st.session_state.student = Student(name)
 
-with st.container():
-    st.subheader("📘 Add Subject")
 
-    col1, col2, col3 = st.columns(3)
+st.subheader("📘 Add Subject")
+
+col1, col2, col3 = st.columns(3)
+
+with col1:
+    subject_name = st.text_input("Subject")
 
-    with col1:
-        subject_name = st.text_input("Subject")
+with col2:
+    hours = st.number_input("Study Hours", min_value=0.0, step=0.5)
 
-    with col2:
-        hours = st.number_input("Study Hours", min_value=0.0, step=0.5)
+with col3:
+    marks = st.number_input("Marks", min_value=0, max_value=100)
 
-    with col3:
-        marks = st.number_input("Marks", min_value=0, max_value=100)
+# -------------------------------
+# ADD SUBJECT
+# -------------------------------
 
-    if st.button("➕ Add Subject"):
-        if not subject_name:
-            st.error("Enter subject name!")
-        elif st.session_state.student is None:
-            st.error("Enter student name first!")
-        else:
-            sub = Subject(subject_name, hours, marks)
-            st.session_state.student.add_subject(sub)
-            st.success(f"{subject_name} added successfully!")
+if st.button("➕ Add Subject"):
+    if not subject_name:
+        st.error("Please enter subject name")
+    elif st.session_state.student is None:
+        st.error("Please enter student name first")
+    else:
+        sub = Subject(subject_name, hours, marks)
+        st.session_state.student.add_subject(sub)
+        st.success(f"{subject_name} added successfully!")
 
 # -------------------------------
-# ANALYSIS SECTION
+# ANALYSIS
 # -------------------------------
 
 st.markdown("---")
+
 if st.button("📊 Analyze Performance"):
 
     student = st.session_state.student
@@ -142,7 +153,7 @@ if st.button("📊 Analyze Performance"):
     else:
         df = student.get_dataframe()
 
-        st.subheader("📋 Subject Data")
+        st.subheader("📋 Performance Table")
         st.dataframe(df, use_container_width=True)
 
         avg = student.calculate_average()
@@ -151,36 +162,26 @@ if st.button("📊 Analyze Performance"):
         predicted = student.predict_performance()
         predicted_grade = student.get_grade(predicted)
 
-        # -------------------------------
-        # METRICS UI
-        # -------------------------------
+        # ---------------- METRICS ----------------
         col1, col2, col3 = st.columns(3)
 
-        col1.metric("Average", f"{avg:.2f}")
-        col2.metric("Highest", max_m)
-        col3.metric("Lowest", min_m)
+        col1.metric("Average Marks", f"{avg:.2f}")
+        col2.metric("Highest Marks", max_m)
+        col3.metric("Lowest Marks", min_m)
 
-        # -------------------------------
-        # PROGRESS BAR (NEW FEATURE)
-        # -------------------------------
+        # ---------------- PROGRESS BAR ----------------
         st.subheader("📈 Performance Progress")
         st.progress(int(avg))
 
         st.write(f"🎯 Current Grade: **{grade}**")
 
-        # -------------------------------
-        # SMART PREDICTION
-        # -------------------------------
+        # ---------------- PREDICTION ----------------
         st.subheader("🔮 Smart Prediction")
 
-        st.write(f"If you study 2 more hours daily:")
-
-        st.success(f"📊 Expected Marks: {predicted:.2f}")
-        st.success(f"🏆 Predicted Grade: {predicted_grade}")
+        st.success(f"Expected Marks (if improved): {predicted:.2f}")
+        st.success(f"Predicted Grade: {predicted_grade}")
 
-        # -------------------------------
-        # INSIGHTS
-        # -------------------------------
+        # ---------------- INSIGHTS ----------------
         st.subheader("📌 Insights")
 
         st.write(f"🏆 Best Subject: **{student.best_subject()}**")
@@ -188,8 +189,6 @@ if st.button("📊 Analyze Performance"):
 
         st.info(student.suggest_improvement())
 
-        # -------------------------------
-        # CHART
-        # -------------------------------
-        st.subheader("📉 Study Trend")
+        # ---------------- CHART ----------------
+        st.subheader("📉 Study vs Performance Trend")
         st.line_chart(df.set_index("Subject"))