Update app.py

app.py

@@ -21,27 +21,24 @@ def call_llm(prompt):
         }
 
         payload = {
-            # ✅ FIXED MODEL
             "model": "grok-4-1-fast-non-reasoning",
-
             "messages": [
                 {"role": "system", "content": "You are a helpful AI assistant."},
                 {"role": "user", "content": prompt}
             ],
             "temperature": 0.7,
-            "max_tokens": 300
+            "max_tokens": 400
         }
 
         response = requests.post(
             API_URL,
             headers=headers,
             json=payload,
-            timeout=30  # ✅ prevents hanging
+            timeout=30
         )
 
         data = response.json()
 
-        # ✅ CRITICAL FIX (no more 'choices' crash)
         if "choices" not in data:
             return f"❌ API Error:\n{data}"
 
@@ -86,6 +83,7 @@ def update_block(blocks, selected, new_text):
 # AI FUNCTIONS
 # -----------------------------
 
+# ✨ Rewrite selected block
 def ai_rewrite(blocks, selected):
     if not blocks or not selected:
         return blocks, gr.update()
@@ -102,16 +100,28 @@ def ai_rewrite(blocks, selected):
     return blocks, gr.update(choices=blocks, value=blocks[idx])
 
 
-def ai_summarize(blocks, selected):
-    if not blocks or not selected:
-        return ""
+# 🚀 Rewrite entire document
+def ai_rewrite_all(blocks):
+    if not blocks:
+        return blocks, gr.update()
 
-    idx = int(selected.split(":")[0])
-    text = blocks[idx].split(": ", 1)[1]
+    full_text = "\n".join([b.split(": ", 1)[1] for b in blocks])
+
+    prompt = f"""
+Rewrite the following text to be clearer, more professional, and well-structured:
+
+{full_text}
+"""
 
-    prompt = f"Summarize this in one concise sentence:\n\n{text}"
+    rewritten = call_llm(prompt)
 
-    return 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])
 
 
 # -----------------------------
@@ -134,9 +144,7 @@ with gr.Blocks(title="Blocksmith + Grok") as app:
     with gr.Row():
         update_btn = gr.Button("Update")
         rewrite_btn = gr.Button("✨ Rewrite (AI)")
-        summarize_btn = gr.Button("🧠 Summarize")
-
-    summary_output = gr.Textbox(label="Summary")
+        rewrite_all_btn = gr.Button("🚀 Rewrite All (AI)")
 
     # -----------------------------
     # EVENTS
@@ -166,10 +174,10 @@ with gr.Blocks(title="Blocksmith + Grok") as app:
         outputs=[state_blocks, block_selector]
     )
 
-    summarize_btn.click(
-        fn=ai_summarize,
-        inputs=[state_blocks, block_selector],
-        outputs=summary_output
+    rewrite_all_btn.click(
+        fn=ai_rewrite_all,
+        inputs=state_blocks,
+        outputs=[state_blocks, block_selector]
     )
 
 app.launch()