Update app.py

app.py

@@ -14,43 +14,43 @@ load_dotenv()
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 model = genai.GenerativeModel("gemini-1.5-flash")
 
-st.set_page_config(page_title="LearnMate - AI Buddy", page_icon="📚")
-st.title("🎓 LearnMate - AI Learning Companion")
+st.set_page_config(page_title="LearnMate - AI Buddy", page_icon="📚")
+st.title("🎓 LearnMate - AI Learning Companion")
 
 # Sidebar: Enhanced Sidebar with Goals and Tasks
-st.sidebar.title("📌 LearnMate Dashboard")
+st.sidebar.title("📌 LearnMate Dashboard")
 
 # Learning Goals Section
-st.sidebar.subheader("🎯 Your Learning Goals")
+st.sidebar.subheader("🎯 Your Learning Goals")
 learning_goal = st.sidebar.text_input("Add a Goal")
-if st.sidebar.button("âž• Add Goal") and learning_goal:
+if st.sidebar.button("➕ Add Goal") and learning_goal:
     if "goals" not in st.session_state:
         st.session_state.goals = []
     st.session_state.goals.append(learning_goal)
 
 if "goals" in st.session_state:
     for goal in st.session_state.goals:
-        st.sidebar.markdown(f"✅ {goal}")
+        st.sidebar.markdown(f"✅ {goal}")
 
 # Project Tracker
-st.sidebar.subheader("📋😎Task Tracker")
+st.sidebar.subheader("📋😎Task Tracker")
 if "todo" not in st.session_state:
     st.session_state.todo = []
 if "done" not in st.session_state:
     st.session_state.done = []
 
-new_task = st.sidebar.text_input("🆕👉 New Task")
-if st.sidebar.button("📌🎯 Add Task") and new_task:
+new_task = st.sidebar.text_input("🆕👉 New Task")
+if st.sidebar.button("📌🎯 Add Task") and new_task:
     st.session_state.todo.append(new_task)
 
 for i, task in enumerate(st.session_state.todo):
-    if st.sidebar.checkbox(f"⬜ {task}", key=f"todo_{i}_{task}"):
+    if st.sidebar.checkbox(f"⬜ {task}", key=f"todo_{i}_{task}"):
         st.session_state.todo.remove(task)
         st.session_state.done.append(task)
 
-st.sidebar.subheader("✅🙌Task Completed")
+st.sidebar.subheader("✅🙌Task Completed")
 for i, task in enumerate(st.session_state.done):
-    st.sidebar.checkbox(f"✔️ {task}", value=True, disabled=True, key=f"done_{i}_{task}")
+    st.sidebar.checkbox(f"✔️ {task}", value=True, disabled=True, key=f"done_{i}_{task}")
 
 # Translation helper
 
@@ -61,19 +61,19 @@ def safe_translate(text, lang):
 
 # Tabs
 
-TABS = st.tabs(["📘 Learning Path", "💬 Study Twin", "🧪 Quiz Generator", "🎧 Audio Summary", "🌐 Regional Buddy"])
+TABS = st.tabs(["📘 Learning Path", "💬 Study Twin", "🧪 Quiz Generator", "🎧 Audio Summary", "🌐 Regional Buddy"])
 
-# ------------------------ 📘 Learning Path ------------------------# 
+# ------------------------ 📘 Learning Path ------------------------# 
 with TABS[0]:
-    st.header("📘 Build Your Learning Roadmap")
+    st.header("📘 Build Your Learning Roadmap")
     
-    lang = st.selectbox("🌐 Language", ["english", "hindi", "tamil", "telugu"])
-    knowledge = st.text_area("🧠 Your Current Knowledge")
-    goal = st.text_area("🎯 Learning Goal")
-    style = st.selectbox("🧩 Learning Style", ["Visual", "Reading", "Hands-on", "Mixed"])
+    lang = st.selectbox("🌐 Language", ["english", "hindi", "tamil", "telugu"])
+    knowledge = st.text_area("🧠 Your Current Knowledge")
+    goal = st.text_area("🎯 Learning Goal")
+    style = st.selectbox("🧩 Learning Style", ["Visual", "Reading", "Hands-on", "Mixed"])
 
-    if st.button("🚀 Generate Plan"):
-        with st.spinner("🧠 Crafting your custom roadmap..."):
+    if st.button("🚀 Generate Plan"):
+        with st.spinner("🧠 Crafting your custom roadmap..."):
             prompt = f"""
             You are LearnMate, an expert AI tutor.
             The user has the following:
@@ -82,11 +82,11 @@ with TABS[0]:
             - Preferred learning style: {style}
 
             Please generate a full markdown learning roadmap that includes:
-            1. 📘 Stage-by-stage steps with estimated timelines.
-            2. 🎨 Visual-style flow or layout described in text if user chose 'Visual'.
-            3. 📺 Three **specific YouTube videos** including titles and real video **hyperlinks**.
-            4. 📚 Recommended resources, tools or tutorials related to the goal.
-            5. 🧠 Personalized study tips matching the selected learning style.
+            1. 📘 Stage-by-stage steps with estimated timelines.
+            2. 🎨 Visual-style flow or layout described in text if user chose 'Visual'.
+            3. 📺 Three **specific YouTube videos** including titles and real video **hyperlinks**.
+            4. 📚 Recommended resources, tools or tutorials related to the goal.
+            5. 🧠 Personalized study tips matching the selected learning style.
 
             Format all sections clearly with markdown headers (##) and bullet points.
             Example for video: [How Neural Networks Learn](https://www.youtube.com/watch?v=aircAruvnKk)
@@ -100,33 +100,33 @@ with TABS[0]:
             if lang != "english":
                 plan = safe_translate(plan, lang)
 
-            st.markdown("### 📜 Your Learning Plan")
+            st.markdown("### 📜 Your Learning Plan")
             st.markdown(plan)
 
             # Enable download
             st.download_button(
-                label="⬇️ Download Plan as .txt",
+                label="⬇️ Download Plan as .txt",
                 data=plan,
                 file_name="learning_plan.txt",
                 mime="text/plain"
             )
 
             st.markdown("---")
-            st.success("✅ Video links are now clickable. Save this roadmap and start learning!")
-# ------------------------ 💬 Study Twin ------------------------
-# ------------------------ 💬 Study Twin ------------------------
+            st.success("✅ Video links are now clickable. Save this roadmap and start learning!")
+# ------------------------ 💬 Study Twin ------------------------
+# ------------------------ 💬 Study Twin ------------------------
 with TABS[1]:
-    st.header("💬 AI Study Twin👯")
+    st.header("💬 AI Study Twin👯")
     if "study_step" not in st.session_state:
         st.session_state.study_step = 1
     if "chat_history" not in st.session_state:
         st.session_state.chat_history = []
 
     if st.session_state.study_step == 1:
-        st.write("Let's get started ✨")
-        st.session_state.study_topic = st.text_input("📘 What topic are you studying?")
+        st.write("Let's get started ✨")
+        st.session_state.study_topic = st.text_input("📘 What topic are you studying?")
         st.session_state.confidence_level = st.slider("Confidence (0-10)", 0, 10)
-        if st.button("➡️ Continue"):
+        if st.button("➡️ Continue"):
             st.session_state.study_step = 2
 
     elif st.session_state.study_step == 2:
@@ -134,29 +134,29 @@ with TABS[1]:
         score = st.session_state.confidence_level
         prompt = f"User is studying: {topic}, confidence: {score}/10. Suggest action plan, style-based activities & encouragement."
         reply = model.generate_content(prompt).text
-        st.markdown("### 🎯 Suggestion")
+        st.markdown("### 🎯 Suggestion")
         st.markdown(reply)
-        if st.button("💬 Ask a Question🌟"):
+        if st.button("💬 Ask a Question🌟"):
             st.session_state.study_step = 3
 
     elif st.session_state.study_step == 3:
-        st.subheader("🤖 Chat with Your Twin")
+        st.subheader("🤖 Chat with Your Twin")
         user_msg = st.text_input("You:", key="twin_input")
-        if st.button("📨 Send"):
+        if st.button("📨 Send"):
             chat = model.start_chat(history=st.session_state.chat_history)
             reply = chat.send_message(user_msg)
             st.session_state.chat_history.append({"role": "user", "parts": [user_msg]})
             st.session_state.chat_history.append({"role": "model", "parts": [reply.text]})
 
         for msg in st.session_state.chat_history:
-            role = "🧑 You" if msg["role"] == "user" else "🤖 Twin"
+            role = "🧑 You" if msg["role"] == "user" else "🤖 Twin"
             st.markdown(f"**{role}:** {msg['parts'][0]}")
-# ------------------------ 🧪 Quiz Generator ------------------------
+# ------------------------ 🧪 Quiz Generator ------------------------
 with TABS[2]:
-    st.header("🧪 Test Yourself!")
+    st.header("🧪 Test Yourself!")
 
-    topic = st.text_input("📘 Enter a topic to quiz yourself:")
-    if st.button("🎯 Generate Quiz"):
+    topic = st.text_input("📘 Enter a topic to quiz yourself:")
+    if st.button("🎯 Generate Quiz"):
         prompt = f"""
         You are a quiz master.
         Generate 5 multiple choice questions (MCQs) for the topic: {topic}.
@@ -177,7 +177,7 @@ with TABS[2]:
         st.session_state.full_quiz_text = quiz_text
 
     if "quiz_data" in st.session_state:
-        st.markdown("### 📝 Your Quiz")
+        st.markdown("### 📝 Your Quiz")
         for i, q_block in enumerate(st.session_state.quiz_data):
             lines = q_block.strip().split("\n")
             q_line = next((l for l in lines if l.strip().lower().startswith("q:")), None)
@@ -185,26 +185,26 @@ with TABS[2]:
             ans_line = next((l for l in lines if "Answer:" in l), None)
 
             if not (q_line and opts and ans_line):
-                st.warning(f"❌ Skipping malformed Q{i+1}")
+                st.warning(f"❌ Skipping malformed Q{i+1}")
                 continue
 
             correct = ans_line.split(":")[-1].strip().lower()
             selected = st.radio(f"Q{i+1}: {q_line[2:].strip()}", opts, key=f"quiz_{i}")
 
-            if st.button(f"✔️ Check Q{i+1}", key=f"btn_{i}"):
+            if st.button(f"✔️ Check Q{i+1}", key=f"btn_{i}"):
                 if selected.lower().startswith(correct):
-                    st.success("✅ Correct!")
+                    st.success("✅ Correct!")
                 else:
-                    st.error(f"❌ Wrong. Correct answer is: {correct}")
+                    st.error(f"❌ Wrong. Correct answer is: {correct}")
 
         # Download full quiz
         st.markdown("---")
-        st.download_button("⬇️ Download Full Quiz (.txt)", st.session_state.full_quiz_text, file_name="quiz.txt")
-# ------------------------ 🎧 Audio Summary ------------------------
+        st.download_button("⬇️ Download Full Quiz (.txt)", st.session_state.full_quiz_text, file_name="quiz.txt")
+# ------------------------ 🎧 Audio Summary ------------------------
 with TABS[3]:
-    st.header("🎧 Audio Summary")
+    st.header("🎧 Audio Summary")
     text = st.text_area("Enter content:")
-    if st.button("🔊 Generate Audio"):
+    if st.button("🔊 Generate Audio"):
         tts = gTTS(text)
         with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as fp:
             tts.save(fp.name)
@@ -212,16 +212,16 @@ with TABS[3]:
                 audio_data = f.read()
                 b64 = base64.b64encode(audio_data).decode()
                 st.audio(f"data:audio/mp3;base64,{b64}", format='audio/mp3')
-                st.download_button("⬇️ Download Audio", audio_data, file_name="audio_summary.mp3")
+                st.download_button("⬇️ Download Audio", audio_data, file_name="audio_summary.mp3")
 
-# ------------------------ 🌐 Regional Buddy ------------------------
+# ------------------------ 🌐 Regional Buddy ------------------------
 with TABS[4]:
-    st.header("🌐 Speak in Your Language")
+    st.header("🌐 Speak in Your Language")
     lang = st.selectbox("Choose Language", ["hindi", "tamil", "telugu"])
     msg = st.text_area("Type your message:")
-    if st.button("🔁 Translate"):
+    if st.button("🔁 Translate"):
         try:
             translated = GoogleTranslator(source="en", target=lang).translate(msg)
             st.success(f"Translated ({lang.upper()}): {translated}")
         except Exception as e:
-            st.error(f"Error: {e}")
+            st.error(f"Error: {e}")