Update streamlit_app.py

streamlit_app.py

@@ -75,6 +75,7 @@ DEFAULT_STATE = {
     "analysis_out": "",
     "raw_output": "",          # full Gemini response before stripping
     "last_error_detail": "",  # traceback + raw output for debugging    
+    "show_raw_on_error": False,
 }
 for k, v in DEFAULT_STATE.items():
     st.session_state.setdefault(k, v)
@@ -381,6 +382,8 @@ if generate_now and not st.session_state["busy"]:
             # keep both traceback and whatever raw output we might have
             st.session_state["last_error_detail"] = f"{tb}\n\nRaw Gemini output:\n{st.session_state.get('raw_output','')}"
             st.session_state["last_error"] = f"Generation error: {exc}"
+            # indicate that raw output should be shown in the main area
+            st.session_state["show_raw_on_error"] = True
             st.error("An error occurred during generation.")
         finally:
             st.session_state["busy"] = False
@@ -390,15 +393,19 @@ if st.session_state["analysis_out"]:
     st.subheader("📝 Analysis")
     st.markdown(st.session_state["analysis_out"])
 
-    # NEW – show full Gemini response
-    if st.session_state["raw_output"]:
-        with st.expander("🔎 Full Gemini output (debug)"):
-            st.code(st.session_state["raw_output"], language="text")
+# Show raw Gemini output **above** the error expander when:
+#   • we have a normal analysis (so user can compare)  OR
+#   • an error occurred (flag set)
+if st.session_state["raw_output"] and (
+    st.session_state["analysis_out"] or st.session_state["show_raw_on_error"]
+):
+    with st.expander("🔎 Full Gemini output (debug)"):
+        st.code(st.session_state["raw_output"], language="text")
 
-    # ---------- Errors ----------
+# ---------- Errors ----------
 if st.session_state["last_error"]:
     with st.expander("❗️ Error details"):
-        # NEW – include raw output if present
+        # only the traceback – raw output already shown above
         st.code(st.session_state["last_error_detail"], language="text")
 
 if __name__ == "__main__":