Update src/streamlit_app.py

src/streamlit_app.py

@@ -29,7 +29,18 @@ def stream_claude_response(prompt, text):
 
     output_text = ""
     progress_bar = st.progress(0, text="Starting...")
-    progress_step = 0
+
+    # Create a placeholder and one text_area (fixed key)
+    if "output_box_text" not in st.session_state:
+        st.session_state.output_box_text = ""
+
+    output_box = st.empty()
+    output_box.text_area(
+        "Claude Output",
+        value=st.session_state.output_box_text,
+        height=400,
+        key="output_box_display",
+    )
 
     # Streaming response
     with client.messages.stream(
@@ -37,18 +48,23 @@ def stream_claude_response(prompt, text):
         max_tokens=64000,
         messages=[{"role": "user", "content": user_content}],
     ) as stream:
-        placeholder = st.empty()
-        for event in stream:
+        for i, event in enumerate(stream):
             if event.type == "content_block_delta" and event.delta.type == "text_delta":
                 delta = event.delta.text
                 output_text += delta
-                progress_step += 1
-                progress_bar.progress(min(progress_step / 400, 0.95), text="Generating...")
-                placeholder.text_area("Claude Output", value=output_text, height=400, key="output_box", label_visibility="collapsed")
+                st.session_state.output_box_text = output_text
+                progress_bar.progress(min(i / 400, 0.95), text="Generating...")
+                output_box.text_area(
+                    "Claude Output",
+                    value=st.session_state.output_box_text,
+                    height=400,
+                    key="output_box_display",
+                )
 
     progress_bar.progress(1.0, text="✅ Done")
     return output_text
 
+
 # === DOCX Creation ===
 def create_docx(text, doc_name):
     cleaned_output = "\n".join(