Update app.py

app.py

@@ -100,10 +100,9 @@ def ai_rewrite(blocks, selected):
     return blocks, gr.update(choices=blocks, value=blocks[idx])
 
 
-# 🚀 Rewrite entire document
 def ai_rewrite_all(blocks):
     if not blocks:
-        return blocks, gr.update()
+        return "⚠️ No blocks to rewrite."
 
     full_text = "\n".join([b.split(": ", 1)[1] for b in blocks])
 
@@ -113,15 +112,7 @@ Rewrite the following text to be clearer, more professional, and well-structured
 {full_text}
 """
 
-    rewritten = call_llm(prompt)
-
-    lines = [line.strip() for line in rewritten.split("\n") if line.strip()]
-    if not lines:
-        lines = ["EMPTY"]
-
-    new_blocks = [f"{i}: {line}" for i, line in enumerate(lines)]
-
-    return new_blocks, gr.update(choices=new_blocks, value=new_blocks[0])
+    return call_llm(prompt)
 
 
 # -----------------------------
@@ -146,6 +137,11 @@ with gr.Blocks(title="Blocksmith + Grok") as app:
         rewrite_btn = gr.Button("✨ Rewrite (AI)")
         rewrite_all_btn = gr.Button("🚀 Rewrite All (AI)")
 
+    full_output = gr.Textbox(
+        label="Rewritten Full Text",
+        lines=10
+    )
+
     # -----------------------------
     # EVENTS
     # -----------------------------
@@ -177,7 +173,7 @@ with gr.Blocks(title="Blocksmith + Grok") as app:
     rewrite_all_btn.click(
         fn=ai_rewrite_all,
         inputs=state_blocks,
-        outputs=[state_blocks, block_selector]
+        outputs=full_output
     )
 
 app.launch()