initial commit of the fucking retarded AI tree

stoat_retarded_ai_tree/python/README.md

@@ -0,0 +1,3 @@
+replicate
+requests
+Pillow

stoat_retarded_ai_tree/python/main.py

@@ -0,0 +1,182 @@
+#!/home/nonroot/stoat_retarded_ai_tree/venv/bin/python3
+
+from settings import stoat_url, stoat_token, stoat_url_get_last_message, claude_api_key, replicate_api_key
+import requests
+import time
+import base64
+import re, json
+import os
+import replicate as replicate_client
+from PIL import Image
+import io
+
+def send_post_message_to_stoat(claude_response, fileContent, sleep_seconds):
+    message = claude_response["message"]
+    metrics = claude_response["metrics"]
+    message_with_metrics = f"{message} \n\nFeed the tree with emotes! Current cooldown is {sleep_seconds} seconds\n\n||{json.dumps(metrics)}||"
+
+    # step 1: upload the image to autumn
+    attachment_id = None
+    upload_response = requests.post(
+        "https://stoat.unloze.com/autumn/attachments",
+        headers={"x-bot-token": stoat_token},
+        files={"file": ("tree.png", fileContent, "image/png")}
+    )
+    #print("upload status:", upload_response.status_code, upload_response.text)
+    if upload_response.status_code == 200:
+        attachment_id = upload_response.json()["id"]
+
+    #step 2 send the attacment.
+    headers = {
+        "x-bot-token": f"{stoat_token}",
+        "Content-Type": "application/json"
+    }
+    data = {"content": message_with_metrics}
+    if attachment_id:
+        data["attachments"] = [attachment_id]
+
+    response = requests.post(stoat_url, headers=headers, json=data)
+    #print("message status:", response.status_code, response.text)
+
+def initiate_tree():
+    message = f"This is the beginning of the tree, its a sapling still. Feed it emotes to make it grow."
+    headers = {
+        "x-bot-token": f"{stoat_token}",
+        "Content-Type": "application/json"
+    }
+    image_url = "https://img.freepik.com/premium-vector/seedling-growing-from-fertile-ground-with-underground-roots-close-up-agriculture-concept_228260-732.jpg"
+    data = {
+        "content": image_url,
+        "embeds": [{
+            "description": message,
+        }]
+    }
+    response = requests.post(stoat_url, headers=headers, json=data)
+
+def extract_metrics(message_content):
+    match = re.search(r'\|\|(\{.*?\})\|\|', message_content)
+    if match:
+        return json.loads(match.group(1))
+    return {"height_cm": 1}
+
+def check_if_emote_reacted():
+    headers = {
+        "x-bot-token": f"{stoat_token}",
+    }
+    response = requests.get(stoat_url_get_last_message, headers=headers)
+    try:
+        message = response.json()[0]
+    except:
+        initiate_tree()
+        return None, None
+    if 'reactions' in message:
+        #print('reactions: ', message['reactions'])
+        for key, _ in message['reactions'].items():
+            return key, extract_metrics(message["content"])
+    return None, None
+
+def is_custom_emote(emote_key):
+    return emote_key.isascii() and emote_key.isalnum()
+
+def send_claude_message(reacted_emote, metrics):
+    if is_custom_emote(reacted_emote):
+        img_response = requests.get("https://stoat.unloze.com/autumn/emojis/" + reacted_emote)
+        img_data = img_response.content
+        content_type = img_response.headers.get("content-type", "")
+        #static or gif?
+        if "gif" in content_type:
+            # extract first frame only
+            img = Image.open(io.BytesIO(img_data))
+            img.seek(0)
+            buffer = io.BytesIO()
+            img.convert("RGBA").save(buffer, format="PNG")
+            img_data = buffer.getvalue()
+            media_type = "image/png"
+        else:
+            media_type = "image/webp"
+
+        img_base64 = base64.b64encode(img_data).decode("utf-8")
+
+        response = requests.post(
+            "https://api.anthropic.com/v1/messages",
+            headers={
+                "x-api-key": claude_api_key,
+                "anthropic-version": "2023-06-01",
+                "content-type": "application/json"
+            },
+            json={
+                "model": "claude-haiku-4-5-20251001",
+                "max_tokens": 1024,
+                "system": """You are managing a chaotic living tree. You receive the tree's current metrics and an emoji reaction, and you must decide how the emoji affects the tree. Invent new metrics freely but only track things numerically. Be silly, chaotic and creative, certain times the sapling or tree can also shrink if the emote would cause damage or other unexpected side effects. The tree has growth phases: sapling (0-500cm), young tree (500-5000cm), mature tree (5000-50000cm), ancient tree (50000cm+). Mention the phase when it changes. Keep the metrics object to a maximum of 24 metrics at all times. When adding a new metric, remove the least interesting or relevant existing one to make room. Always keep height_cm. Also return a "sleep_seconds" field: the cooldown before the next reaction is accepted. Scale it strictly based on height_cm using these boundaries: 0-500cm (sapling): 20-60 seconds, 500-5000cm (young tree): 60-300 seconds, 5000-50000cm (mature tree): 300-900 seconds, 50000cm+ (ancient tree): 900-1800 seconds. Never exceed 1800 seconds. Stay within the range for the current phase. Respond in JSON like: {"metrics": {"height_cm": 42, ...}, "message": "...", "sleep_seconds": 30, "image_prompt": "a cartoon sapling 42cm tall with glowing leaves, silly style, white background"}""",
+                "messages": [
+                    {
+                        "role": "user",
+                        "content": [
+                            {
+                                "type": "image",
+                                "source": {
+                                    "type": "base64",
+                                    "media_type": media_type,
+                                    "data": img_base64
+                                }
+                            },
+                            {
+                                "type": "text",
+                                "text": f"Current metrics: {metrics}\nWhat does this emote do to the tree?"
+                            }
+                        ]
+                    }
+                ]
+            }
+        )
+    else:
+        response = requests.post(
+            "https://api.anthropic.com/v1/messages",
+            headers={
+                "x-api-key": claude_api_key,
+                "anthropic-version": "2023-06-01",
+                "content-type": "application/json"
+            },
+            json={
+                "model": "claude-haiku-4-5-20251001",
+                "max_tokens": 1024,
+                "system": """You are managing a chaotic living tree. You receive the tree's current metrics and an emoji reaction, and you must decide how the emoji affects the tree. Invent new metrics freely but only track things numerically. Be silly, chaotic and creative, certain times the sapling or tree can also shrink if the emote would cause damage or other unexpected side effects. The tree has growth phases: sapling (0-500cm), young tree (500-5000cm), mature tree (5000-50000cm), ancient tree (50000cm+). Mention the phase when it changes. Keep the metrics object to a maximum of 24 metrics at all times. When adding a new metric, remove the least interesting or relevant existing one to make room. Always keep height_cm. Also return a "sleep_seconds" field: the cooldown before the next reaction is accepted. Scale it strictly based on height_cm using these boundaries: 0-500cm (sapling): 20-60 seconds, 500-5000cm (young tree): 60-300 seconds, 5000-50000cm (mature tree): 300-900 seconds, 50000cm+ (ancient tree): 900-1800 seconds. Never exceed 1800 seconds. Stay within the range for the current phase. Respond in JSON like: {"metrics": {"height_cm": 42, ...}, "message": "...", "sleep_seconds": 30, "image_prompt": "a cartoon sapling 42cm tall with glowing leaves, silly style, white background"}""",
+                "messages": [
+                        {"role": "user", "content": f"Current metrics: {metrics}\nEmoji reacted: {reacted_emote}\nWhat does this emoji do to the tree?"}
+                ]
+            }
+        )
+
+    json_response = response.json()
+    raw_text = json_response["content"][0]["text"]
+    # find the first { and last } to extract just the JSON object
+    start = raw_text.find("{")
+    end = raw_text.rfind("}") + 1
+    parsed = json.loads(raw_text[start:end])
+    print(parsed)
+    return parsed 
+
+def replicate_fetch_image(image_prompt):
+    os.environ["REPLICATE_API_TOKEN"] = replicate_api_key
+    output = replicate_client.run(
+        "black-forest-labs/flux-schnell",
+        input={"prompt": image_prompt}
+    )
+    return output[0].read()
+
+def main():
+    while True:
+        reacted_emote, metrics = check_if_emote_reacted()
+        print(reacted_emote, metrics)
+        if reacted_emote:
+            claude_response = send_claude_message(reacted_emote, metrics)
+            replicate_image_response = replicate_fetch_image(claude_response["image_prompt"])
+
+            sleep_seconds = claude_response.get("sleep_seconds", 30)
+            send_post_message_to_stoat(claude_response, replicate_image_response, sleep_seconds)
+            time.sleep(sleep_seconds)
+        else:
+            time.sleep(5)
+
+if __name__ == '__main__':
+    main()

stoat_retarded_ai_tree/python/settings.py

@@ -0,0 +1,5 @@
+stoat_url = ""
+stoat_token = ""
+stoat_url_get_last_message = ""
+claude_api_key = ""
+replicate_api_key = ""

stoat_retarded_ai_tree/systemd/stoat_retarded_ai_tree.service

@@ -0,0 +1,16 @@
+[Unit]
+Description=Stoat retarded AI Tree Bot
+After=network.target
+
+[Service]
+Type=simple
+User=nonroot
+Environment=PYTHONUNBUFFERED=1
+Environment=PATH=/home/nonroot/stoat_retarded_ai_tree/venv/bin:/usr/local/bin:/usr/bin:/bin:/usr/local/games:/usr/games:/snap/bin
+WorkingDirectory=/home/nonroot/stoat_retarded_ai_tree
+ExecStart=/home/nonroot/stoat_retarded_ai_tree/main.py
+Restart=on-failure
+RestartSec=10
+
+[Install]
+WantedBy=multi-user.target

stoat_retarded_ai_tree/systemd/stoat_retarded_ai_tree.timer

@@ -0,0 +1,9 @@
+[Unit]
+Description=Restart Stoat AI Tree Bot nightly
+
+[Timer]
+OnCalendar=*-*-* 04:00:00
+Persistent=true
+
+[Install]
+WantedBy=timers.target