Update app.py

app.py

@@ -1,46 +1,42 @@
 import gradio as gr
-import boto3
-import json
+import requests
+import os
 
 # -----------------------------
-# AWS BEDROCK SETUP
+# GROK SETUP (xAI)
 # -----------------------------
 
-bedrock = boto3.client(
-    service_name="bedrock-runtime",
-    region_name="us-east-1"  # change if needed
-)
-
-MODEL_ID = "anthropic.claude-3-5-sonnet-20240620-v1:0"
+API_KEY = os.getenv("XAI_API_KEY")
 
+API_URL = "https://api.x.ai/v1/chat/completions"
 
 def call_llm(prompt):
     try:
-        body = json.dumps({
-            "anthropic_version": "bedrock-2023-05-31",
+        headers = {
+            "Authorization": f"Bearer {API_KEY}",
+            "Content-Type": "application/json"
+        }
+
+        payload = {
+            "model": "grok-beta",
             "messages": [
                 {"role": "user", "content": prompt}
             ],
-            "max_tokens": 500,
-            "temperature": 0.7
-        })
-
-        response = bedrock.invoke_model(
-            body=body,
-            modelId=MODEL_ID,
-            contentType="application/json"
-        )
+            "temperature": 0.7,
+            "max_tokens": 500
+        }
 
-        response_body = json.loads(response["body"].read())
+        response = requests.post(API_URL, headers=headers, json=payload)
+        result = response.json()
 
-        return response_body["content"][0]["text"]
+        return result["choices"][0]["message"]["content"]
 
     except Exception as e:
         return f"ERROR: {str(e)}"
 
 
 # -----------------------------
-# BLOCK LOGIC
+# BLOCK LOGIC (same as before)
 # -----------------------------
 
 def create_blocks(text):
@@ -70,6 +66,10 @@ def update_block(blocks, selected, new_text):
     return blocks, gr.update(choices=blocks, value=blocks[idx])
 
 
+# -----------------------------
+# AI FUNCTIONS (GROK)
+# -----------------------------
+
 def ai_rewrite(blocks, selected):
     if not blocks or not selected:
         return blocks, gr.update()
@@ -93,7 +93,7 @@ def ai_summarize(blocks, selected):
     idx = int(selected.split(":")[0])
     text = blocks[idx].split(": ", 1)[1]
 
-    prompt = f"Summarize this in 1 sentence:\n\n{text}"
+    prompt = f"Summarize this in one concise sentence:\n\n{text}"
 
     return call_llm(prompt)
 
@@ -102,9 +102,9 @@ def ai_summarize(blocks, selected):
 # UI
 # -----------------------------
 
-with gr.Blocks(title="Blocksmith + Bedrock") as app:
+with gr.Blocks(title="Blocksmith + Grok") as app:
 
-    gr.Markdown("# 🧱 Blocksmith (AI Powered via Bedrock)")
+    gr.Markdown("# 🧱 Blocksmith (Powered by Grok)")
 
     state_blocks = gr.State([])