Update app.py

app.py

@@ -303,28 +303,34 @@ APA Citations:
         else:
             st.success("✅ No major overlaps found.")
 
-        # 🔹 Follow-up Interaction
+        # 🔹 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["follow_up"] = follow_input
+        
+        # Always-visible input box
+        if "follow_input" not in st.session_state:
+            st.session_state["follow_input"] = ""
+        
+        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.")
 
-        if st.session_state.get("follow_up") and 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"]}
-                ]
-                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}")
-            finally:
-                st.session_state["follow_up"] = ""
 
     except Exception as e:
         st.error(f"❌ Error occurred: {e}")