Update frontend.py

frontend.py

@@ -185,19 +185,21 @@ with tab1:
 
                         if custom_q:
                             with st.spinner("Thinking..."):
-                                payload["follow_up"] = custom_q
-                                res = requests.post(f"{backend_url}/analyze/", json=payload, headers=headers, params=params)
-                                if res.status_code == 200:
-                                    follow = res.json().get("follow_up")
-                                    if follow:
+                                try:
+                                    follow_payload = {
+                                        "text": review,
+                                        "question": custom_q,
+                                        "verbosity": verbosity
+                                    }
+                                    res = requests.post(f"{backend_url}/followup/", json=follow_payload, headers=headers)
+                                    if res.status_code == 200:
+                                        follow = res.json().get("answer")
                                         st.subheader("🔁 Follow-Up Answer")
-                                        if isinstance(follow, list):
-                                            for q in follow:
-                                                st.write("➡️", q)
-                                        else:
-                                            st.warning(follow)
-                                else:
-                                    st.error(f"❌ Follow-up failed: {res.json().get('detail')}")
+                                        st.warning(follow)
+                                    else:
+                                        st.error(f"❌ Follow-up failed: {res.json().get('detail', 'Unknown error')}")
+                                except Exception as e:
+                                    st.error(f"⚠️ Follow-up error: {e}")
                     else:
                         st.error(f"❌ API Error {res.status_code}: {res.json().get('detail', 'Unknown error')}")
                 except Exception as e: