Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,44 +5,62 @@ from docx import Document
 from docx.shared import Pt
 import os
 import tempfile
-import asyncio
 
-# === Load API Key ===
+# ===== Load API Key =====
 load_dotenv()
 client = Anthropic(api_key=os.getenv("ANTHROPIC_API_KEY"))
 
-# === Page Config ===
+# ===== Page Config =====
 st.set_page_config(page_title="Claude UI-HF", layout="wide")
 st.markdown("<h2 style='text-align:center;'>Claude UI-HF</h2>", unsafe_allow_html=True)
 
-# === Helper: Stream Claude Response ===
+# ===== Word/Character Counter =====
+def count_words(text):
+    words = len(text.split())
+    chars = len(text)
+    return f"📊 Words: {words:,} | Characters: {chars:,}"
+
+# ===== DOCX Creation =====
+def create_docx(text, doc_name):
+    cleaned_output = "\n".join(line for line in text.splitlines() if line.strip() != "---").strip()
+    if not doc_name.strip():
+        doc_name = "Claude_Output"
+    docx_path = os.path.join(tempfile.gettempdir(), f"{doc_name}.docx")
+    document = Document()
+    for line in cleaned_output.split("\n"):
+        clean_line = line.strip()
+        if not clean_line:
+            continue
+        if clean_line.startswith("# "):
+            document.add_heading(clean_line[2:], level=1)
+        elif clean_line.startswith("## "):
+            document.add_heading(clean_line[3:], level=2)
+        elif clean_line.startswith("### "):
+            document.add_heading(clean_line[4:], level=3)
+        else:
+            para = document.add_paragraph(clean_line)
+            para.style.font.size = Pt(11)
+    document.save(docx_path)
+    return docx_path
+
+# ===== Main Streaming Function =====
 def stream_claude_response(prompt, text):
     if not prompt.strip() and not text.strip():
         st.warning("⚠️ Please enter a prompt or text.")
         return ""
 
-    user_content = ""
-    if prompt.strip():
-        user_content += prompt.strip()
+    user_content = prompt.strip()
     if text.strip():
         user_content += f"\n\n{text.strip()}"
 
     output_text = ""
     progress_bar = st.progress(0, text="Starting...")
 
-    # Create a placeholder and one text_area (fixed key)
-    if "output_box_text" not in st.session_state:
-        st.session_state.output_box_text = ""
+    # Single placeholder for output
+    output_placeholder = st.empty()
+    output_placeholder.text_area("Claude Output", value="", height=400, key="output_box")
 
-    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
+    # Streaming from Claude
     with client.messages.stream(
         model="claude-sonnet-4-20250514",
         max_tokens=64000,
@@ -52,76 +70,35 @@ def stream_claude_response(prompt, text):
             if event.type == "content_block_delta" and event.delta.type == "text_delta":
                 delta = event.delta.text
                 output_text += delta
-                st.session_state.output_box_text = output_text
+                # Update output placeholder
+                output_placeholder.text_area("Claude Output", value=output_text, height=400)
                 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(
-        line for line in text.splitlines() if line.strip() != "---"
-    ).strip()
-
-    if not doc_name.strip():
-        doc_name = "Claude_Output"
-    docx_path = os.path.join(tempfile.gettempdir(), f"{doc_name}.docx")
-
-    document = Document()
-    for line in cleaned_output.split("\n"):
-        clean_line = line.strip()
-        if not clean_line:
-            continue
-        if clean_line.startswith("# "):
-            document.add_heading(clean_line[2:], level=1)
-        elif clean_line.startswith("## "):
-            document.add_heading(clean_line[3:], level=2)
-        elif clean_line.startswith("### "):
-            document.add_heading(clean_line[4:], level=3)
-        else:
-            para = document.add_paragraph(clean_line)
-            para.style.font.size = Pt(11)
-    document.save(docx_path)
-    return docx_path
-
-# === Word Counter ===
-def count_words(text):
-    words = len(text.split())
-    chars = len(text)
-    return f"📊 Words: {words:,} | Characters: {chars:,}"
-
-# === UI Layout ===
-with st.container():
-    prompt = st.text_area("Prompt", placeholder="Enter your instruction here", height=100)
-    text = st.text_area("Text", placeholder="Paste your large text here", height=300)
-    doc_name = st.text_input("📂 Document Name", placeholder="e.g. MyNotes")
+# ===== UI Inputs =====
+prompt = st.text_area("Prompt", placeholder="Enter your instruction here", height=100)
+text = st.text_area("Text", placeholder="Paste your large text here", height=300)
+doc_name = st.text_input("📂 Document Name", placeholder="e.g. MyNotes")
 
 run_col, cancel_col = st.columns([1, 1])
-run_clicked = run_col.button("🚀 Run", use_container_width=True)
-cancel_clicked = cancel_col.button("🛑 Cancel", use_container_width=True)
+run_clicked = run_col.button("🚀 Run")
+cancel_clicked = cancel_col.button("🛑 Cancel")
 
-output_box = st.empty()
 stats_placeholder = st.empty()
 download_placeholder = st.empty()
+copy_placeholder = st.empty()
 
-# === Main Logic ===
+# ===== Run Logic =====
 if run_clicked:
     try:
-        with st.spinner("Running Claude..."):
-            response_text = stream_claude_response(prompt, text)
+        response_text = stream_claude_response(prompt, text)
         if response_text:
-            # Word count
+            # Word/char stats
             stats_placeholder.write(count_words(response_text))
 
-            # Save DOCX
+            # Save DOCX & provide download
             docx_file = create_docx(response_text, doc_name)
             with open(docx_file, "rb") as f:
                 download_placeholder.download_button(
@@ -132,8 +109,8 @@ if run_clicked:
                     use_container_width=True
                 )
 
-            # Copy to clipboard (via JS)
-            st.markdown("""
+            # Copy to clipboard (JS)
+            copy_placeholder.markdown("""
             <button onclick="navigator.clipboard.writeText(document.querySelector('textarea[data-testid=stTextArea-input]').value); alert('✅ Copied to clipboard!');">
             📋 Copy Output
             </button>
@@ -143,5 +120,4 @@ if run_clicked:
         st.error(f"❌ Error: {e}")
 
 elif cancel_clicked:
-    st.warning("🛑 Cancel clicked — refresh to restart session.")
-
+    st.warning("🛑 Cancel clicked — refresh the page to restart.")