Update app.py

app.py

@@ -190,12 +190,11 @@ def generate_download_button(file, label, mime_type):
 # --- Streamlit UI ---
 st.set_page_config("Deep Research Assistant", layout="centered")
 # 🧠 Initialize session state
-if "last_report" not in st.session_state:
-    st.session_state.last_report = ""
-if "mindmap_triggered" not in st.session_state:
-    st.session_state.mindmap_triggered = False
-if "followup_question" not in st.session_state:
-    st.session_state.followup_question = ""
+if "follow_up_input" not in st.session_state:
+    st.session_state["follow_up_input"] = ""
+if "methodology_recommended" not in st.session_state:
+    st.session_state["methodology_recommended"] = False
+
 
 # Mermaid.js for mind map rendering
 st.markdown("""
@@ -304,33 +303,37 @@ APA Citations:
             st.success("✅ No major overlaps found.")
 
         st.subheader("💬 Ask a Follow-Up")
-        follow_input = st.text_input("Ask a question about the report:")
-        if st.button("🔄 Submit Follow-Up") and follow_input:
-            st.session_state.followup_question = follow_input
-        
-        if st.session_state.followup_question and st.session_state.last_report:
-            follow_prompt = [
-                {"role": "system", "content": "You are a helpful academic assistant."},
-                {"role": "user", "content": st.session_state.last_report},
-                {"role": "user", "content": st.session_state.followup_question}
-            ]
-            follow_output = ""
-            follow_box = st.empty()
-            for chunk in call_llm(follow_prompt):
-                follow_output += chunk
-                follow_box.markdown(follow_output, unsafe_allow_html=True)
+
+        st.session_state["follow_up_input"] = st.text_input(
+            "Ask a question about the report:", value=st.session_state["follow_up_input"]
+        )
         
-            st.session_state.followup_question = ""
+        if st.button("🔄 Submit Follow-Up") and st.session_state["follow_up_input"]:
+            if st.session_state["last_report"]:
+                try:
+                    follow_prompt = [
+                        {"role": "system", "content": "You are a helpful academic assistant."},
+                        {"role": "user", "content": st.session_state["last_report"]},
+                        {"role": "user", "content": st.session_state["follow_up_input"]}
+                    ]
+                    follow_output = ""
+                    follow_box = st.empty()
+                    for chunk in call_llm(follow_prompt):
+                        follow_output += chunk
+                        follow_box.markdown(follow_output, unsafe_allow_html=True)
+                except Exception as e:
+                    st.error(f"❌ Follow-up failed: {e}")
+            else:
+                st.warning("⚠️ Generate the research report first.")
 
-        # 📌 Methodology Recommender
         st.subheader("🧪 Methodology Recommender")
-        
+
         if st.button("🧠 Suggest Research Methodologies"):
-            if st.session_state.last_report:
+            if st.session_state["last_report"]:
                 try:
                     method_prompt = [
-                        {"role": "system", "content": "You are a research advisor. Based on the report, suggest appropriate research methodologies or experimental frameworks (qualitative, quantitative, statistical models, AI/ML techniques, etc.). Provide concise and specific recommendations in bullet points."},
-                        {"role": "user", "content": st.session_state.last_report}
+                        {"role": "system", "content": "You are a research advisor. Based on the following research report, suggest suitable research methodologies (quantitative, qualitative, ML/AI techniques, etc.). Be concise and give bullet-point suggestions."},
+                        {"role": "user", "content": st.session_state["last_report"]}
                     ]
                     method_output = ""
                     method_box = st.empty()
@@ -340,7 +343,8 @@ APA Citations:
                 except Exception as e:
                     st.error(f"❌ Methodology suggestion failed: {e}")
             else:
-                st.warning("⚠️ Generate the research report first to get methodology suggestions.")
+                st.warning("⚠️ Generate the research report first.")
+
 
 
     except Exception as e: