Update app.py

app.py

@@ -303,34 +303,24 @@ APA Citations:
         else:
             st.success("✅ No major overlaps found.")
 
-        # 🔹 Follow-up Interaction (Always Visible)
         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
         
-        # Always-visible input box
-        if "follow_input" not in st.session_state:
-            st.session_state["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_input"] = st.text_input("Ask a question about the report:", value=st.session_state["follow_input"])
-        
-        # Handle Submit Button
-        if st.button("🔄 Submit Follow-Up") and st.session_state["follow_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_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("⚠️ Please generate a report first.")
-
+            st.session_state.followup_question = ""
 
     except Exception as e:
         st.error(f"❌ Error occurred: {e}")