Update app.py

app.py

@@ -2,158 +2,120 @@ import gradio as gr
 import requests
 import os
 
-# -----------------------------
-# GROK SETUP (xAI)
-# -----------------------------
+# =========================
+# CONFIG
+# =========================
 
 API_KEY = os.getenv("XAI_API_KEY")
 
-API_URL = "https://api.x.ai/v1/chat/completions"
-
-def call_llm(prompt):
-    try:
-        headers = {
-            "Authorization": f"Bearer {API_KEY}",
-            "Content-Type": "application/json"
-        }
-
-        payload = {
-            "model": "grok-beta",
-            "messages": [
-                {"role": "user", "content": prompt}
-            ],
-            "temperature": 0.7,
-            "max_tokens": 500
-        }
-
-        response = requests.post(API_URL, headers=headers, json=payload)
-        result = response.json()
-
-        return result["choices"][0]["message"]["content"]
-
-    except Exception as e:
-        return f"ERROR: {str(e)}"
-
-
-# -----------------------------
-# BLOCK LOGIC (same as before)
-# -----------------------------
-
-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])
+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"
 
-# -----------------------------
-# AI FUNCTIONS (GROK)
-# -----------------------------
 
-def ai_rewrite(blocks, selected):
-    if not blocks or not selected:
-        return blocks, gr.update()
+# =========================
+# GROK CALL
+# =========================
 
-    idx = int(selected.split(":")[0])
-    original = blocks[idx].split(": ", 1)[1]
+def call_grok(prompt):
+    headers = {
+        "Authorization": f"Bearer {API_KEY}",
+        "Content-Type": "application/json"
+    }
 
-    prompt = f"Rewrite this more clearly and professionally:\n\n{original}"
+    payload = {
+        "model": "grok-2-latest",  # fallback: grok-1
+        "messages": [
+            {"role": "system", "content": "You are a helpful AI assistant."},
+            {"role": "user", "content": prompt}
+        ],
+        "temperature": 0.7
+    }
 
-    new_text = call_llm(prompt)
+    try:
+        response = requests.post(
+            API_URL,
+            headers=headers,
+            json=payload,
+            timeout=30
+        )
 
-    blocks[idx] = f"{idx}: {new_text}"
+        data = response.json()
 
-    return blocks, gr.update(choices=blocks, value=blocks[idx])
+        # 🔴 Handle API errors cleanly
+        if "choices" not in data:
+            return f"❌ API Error:\n{data}"
 
+        return data["choices"][0]["message"]["content"]
 
-def ai_summarize(blocks, selected):
-    if not blocks or not selected:
-        return ""
+    except Exception as e:
+        return f"❌ Request Failed:\n{str(e)}"
 
-    idx = int(selected.split(":")[0])
-    text = blocks[idx].split(": ", 1)[1]
 
-    prompt = f"Summarize this in one concise sentence:\n\n{text}"
+# =========================
+# BLOCKSMITH LOGIC
+# =========================
 
-    return call_llm(prompt)
+def generate_blocks(user_input):
+    if not user_input.strip():
+        return "⚠️ Please enter something."
 
+    # Simulate your “block” concept
+    prompt = f"""
+Break this into structured reasoning blocks:
 
-# -----------------------------
-# UI
-# -----------------------------
+Input:
+{user_input}
 
-with gr.Blocks(title="Blocksmith + Grok") as app:
+Output format:
+1. Problem
+2. Key Ideas
+3. Solution Steps
+4. Final Answer
+"""
 
-    gr.Markdown("# 🧱 Blocksmith (Powered by Grok)")
+    result = call_grok(prompt)
+    return result
 
-    state_blocks = gr.State([])
 
-    input_text = gr.Textbox(label="Paste Text", lines=8)
-    create_btn = gr.Button("Create Blocks")
+# =========================
+# UI (CLEAN + MODERN)
+# =========================
 
-    block_selector = gr.Dropdown(choices=[], label="Blocks", interactive=True)
+with gr.Blocks(theme=gr.themes.Soft(), title="Blocksmith AI") as app:
 
-    block_editor = gr.Textbox(label="Edit Block")
+    gr.Markdown("""
+    # 🧠 Blocksmith AI
+    Turn any idea into structured reasoning blocks.
+    """)
 
     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")
-
-    # -----------------------------
-    # EVENTS
-    # -----------------------------
-
-    create_btn.click(
-        fn=create_blocks,
-        inputs=input_text,
-        outputs=[state_blocks, block_selector]
-    )
-
-    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]
+        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
     )
 
-    summarize_btn.click(
-        fn=ai_summarize,
-        inputs=[state_blocks, block_selector],
-        outputs=summary_output
-    )
+# =========================
+# RUN
+# =========================
 
-app.launch()
+if __name__ == "__main__":
+    app.launch()