Update app.py

app.py

@@ -2,120 +2,174 @@ import gradio as gr
 import requests
 import os
 
-# =========================
-# CONFIG
-# =========================
+# -----------------------------
+# GROK SETUP (xAI)
+# -----------------------------
 
 API_KEY = os.getenv("XAI_API_KEY")
-
-if not API_KEY:
-    raise ValueError("❌ XAI_API_KEY not set. Add it in Hugging Face Secrets.")
-
 API_URL = "https://api.x.ai/v1/chat/completions"
 
+if not API_KEY:
+    raise ValueError("❌ XAI_API_KEY not set in Hugging Face Secrets")
 
-# =========================
-# GROK CALL
-# =========================
-
-def call_grok(prompt):
-    headers = {
-        "Authorization": f"Bearer {API_KEY}",
-        "Content-Type": "application/json"
-    }
-
-    payload = {
-        "model": "grok-4-1-fast-non-reasoning",
-        "messages": [
-            {"role": "system", "content": "You are a helpful AI assistant."},
-            {"role": "user", "content": prompt}
-        ],
-        "temperature": 0.7
-    }
 
+def call_llm(prompt):
     try:
+        headers = {
+            "Authorization": f"Bearer {API_KEY}",
+            "Content-Type": "application/json"
+        }
+
+        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
+        }
+
         response = requests.post(
             API_URL,
             headers=headers,
             json=payload,
-            timeout=30
+            timeout=30  # ✅ prevents hanging
         )
 
         data = response.json()
 
-        # 🔴 Handle API errors cleanly
+        # ✅ CRITICAL FIX (no more 'choices' crash)
         if "choices" not in data:
             return f"❌ API Error:\n{data}"
 
         return data["choices"][0]["message"]["content"]
 
     except Exception as e:
-        return f"❌ Request Failed:\n{str(e)}"
+        return f"❌ Request Failed: {str(e)}"
+
+
+# -----------------------------
+# BLOCK LOGIC
+# -----------------------------
+
+def create_blocks(text):
+    lines = [line.strip() for line in text.split("\n") if line.strip()]
+    if not lines:
+        lines = ["EMPTY"]
+
+    blocks = [f"{i}: {line}" for i, line in enumerate(lines)]
+    return blocks, gr.update(choices=blocks, value=blocks[0])
+
+
+def select_block(blocks, selected):
+    if not blocks or not selected:
+        return ""
+
+    idx = int(selected.split(":")[0])
+    return blocks[idx].split(": ", 1)[1]
+
+
+def update_block(blocks, selected, new_text):
+    if not blocks or not selected:
+        return blocks, gr.update()
+
+    idx = int(selected.split(":")[0])
+    blocks[idx] = f"{idx}: {new_text}"
+
+    return blocks, gr.update(choices=blocks, value=blocks[idx])
+
 
+# -----------------------------
+# AI FUNCTIONS
+# -----------------------------
 
-# =========================
-# BLOCKSMITH LOGIC
-# =========================
+def ai_rewrite(blocks, selected):
+    if not blocks or not selected:
+        return blocks, gr.update()
 
-def generate_blocks(user_input):
-    if not user_input.strip():
-        return "⚠️ Please enter something."
+    idx = int(selected.split(":")[0])
+    original = blocks[idx].split(": ", 1)[1]
 
-    # Simulate your “block” concept
-    prompt = f"""
-Break this into structured reasoning blocks:
+    prompt = f"Rewrite this more clearly and professionally:\n\n{original}"
 
-Input:
-{user_input}
+    new_text = call_llm(prompt)
 
-Output format:
-1. Problem
-2. Key Ideas
-3. Solution Steps
-4. Final Answer
-"""
+    blocks[idx] = f"{idx}: {new_text}"
 
-    result = call_grok(prompt)
-    return result
+    return blocks, gr.update(choices=blocks, value=blocks[idx])
 
 
-# =========================
-# UI (CLEAN + MODERN)
-# =========================
+def ai_summarize(blocks, selected):
+    if not blocks or not selected:
+        return ""
 
-with gr.Blocks(theme=gr.themes.Soft(), title="Blocksmith AI") as app:
+    idx = int(selected.split(":")[0])
+    text = blocks[idx].split(": ", 1)[1]
 
-    gr.Markdown("""
-    # 🧠 Blocksmith AI
-    Turn any idea into structured reasoning blocks.
-    """)
+    prompt = f"Summarize this in one concise sentence:\n\n{text}"
+
+    return call_llm(prompt)
+
+
+# -----------------------------
+# UI
+# -----------------------------
+
+with gr.Blocks(title="Blocksmith + Grok") as app:
+
+    gr.Markdown("# 🧱 Blocksmith (Powered by Grok)")
+
+    state_blocks = gr.State([])
+
+    input_text = gr.Textbox(label="Paste Text", lines=8)
+    create_btn = gr.Button("Create Blocks")
+
+    block_selector = gr.Dropdown(choices=[], label="Blocks", interactive=True)
+
+    block_editor = gr.Textbox(label="Edit Block")
 
     with gr.Row():
-        with gr.Column(scale=1):
-            input_box = gr.Textbox(
-                label="Your Input",
-                placeholder="Enter a problem, idea, or question...",
-                lines=5
-            )
-
-            generate_btn = gr.Button("⚡ Generate Blocks", variant="primary")
-
-        with gr.Column(scale=1):
-            output_box = gr.Textbox(
-                label="Structured Output",
-                lines=15
-            )
-
-    # 🔥 Button wiring (this is what usually breaks)
-    generate_btn.click(
-        fn=generate_blocks,
-        inputs=input_box,
-        outputs=output_box
+        update_btn = gr.Button("Update")
+        rewrite_btn = gr.Button("✨ Rewrite (AI)")
+        summarize_btn = gr.Button("🧠 Summarize")
+
+    summary_output = gr.Textbox(label="Summary")
+
+    # -----------------------------
+    # EVENTS
+    # -----------------------------
+
+    create_btn.click(
+        fn=create_blocks,
+        inputs=input_text,
+        outputs=[state_blocks, block_selector]
     )
 
-# =========================
-# RUN
-# =========================
+    block_selector.change(
+        fn=select_block,
+        inputs=[state_blocks, block_selector],
+        outputs=block_editor
+    )
+
+    update_btn.click(
+        fn=update_block,
+        inputs=[state_blocks, block_selector, block_editor],
+        outputs=[state_blocks, block_selector]
+    )
+
+    rewrite_btn.click(
+        fn=ai_rewrite,
+        inputs=[state_blocks, block_selector],
+        outputs=[state_blocks, block_selector]
+    )
+
+    summarize_btn.click(
+        fn=ai_summarize,
+        inputs=[state_blocks, block_selector],
+        outputs=summary_output
+    )
 
-if __name__ == "__main__":
-    app.launch()
+app.launch()