Update app.py

app.py

@@ -1,4 +1,4 @@
-import streamlit as st
+Import streamlit as st
 import os
 import time
 import re
@@ -78,16 +78,6 @@ with col2:
             with st.chat_message(msg["role"]):
                 st.write(msg["content"])
 
-                # ✅ Extract and render follow-up questions as buttons (from assistant)
-                if msg["role"] == "assistant":
-                    followups = re.findall(r"[-•*] (What|How|Are|Is|Does|Can|When|Where|Why)[^\n]+", msg["content"])
-                    if followups:
-                        st.markdown("#### 💡 Ask a Follow-up:")
-                        for i, q in enumerate(followups):
-                            if st.button(f"🔁 {q}", key=f"followup-{i}-{q[:10]}"):
-                                st.session_state.messages.append({"role": "user", "content": q})
-                                st.rerun()
-
     if prompt:
         st.session_state.messages.append({"role": "user", "content": prompt})
 
@@ -146,4 +136,4 @@ with col2:
             st.rerun()
 
         except Exception as e:
-            st.error(f"❌ Error: {str(e)}")
+            st.error(f"❌ Error: {str(e)}")