Fixed to keep state in a file

.gitignore

@@ -1 +1,2 @@
 config.toml
+replied_mentions.txt

mbot.py

@@ -1,59 +1,78 @@
 import toml
-from mastodon import Mastodon
+import requests
-import time
+import mastodon
+import os
 import html2text
-from ollama import Client
+import ollama
+import time
 
-# Load configuration from config.toml
+CONFIG_PATH = 'config.toml'
-with open('config.toml', 'r') as f:
+STATE_FILE = 'replied_mentions.txt'
-    config = toml.load(f)
 
-mastodon_config = config['mastodon']
+def load_config(path):
-ollama_config = config['ollama']
+    with open(path, 'r') as file:
-preamble = config.get('preamble', '')
+        return toml.load(file)
 
-# Initialize Mastodon client
+def load_state():
-mastodon = Mastodon(
+    if not os.path.exists(STATE_FILE):
-    access_token=mastodon_config['access_token'],
+        return set()
-    api_base_url=mastodon_config['host']
+    with open(STATE_FILE, 'r') as file:
-)
+        return set(line.strip() for line in file.readlines())
 
-# Function to send text to Ollama and get a response
+def save_state(state):
-def ask_ollama(prompt):
+    with open(STATE_FILE, 'w') as file:
-    client = Client(
+        for item in state:
-        host=f"http://{ollama_config['host']}:{ollama_config['port']}"
+            file.write(f"{item}\n")
+
+def send_to_ollama(prompt, config):
+    client = ollama.Client(
+        host=f"http://{config['ollama']['host']}:{config['ollama']['port']}"
     )
     try:
-        response = client.generate(f"{ollama_config['model']}", f"{preamble}\n{prompt}")
+        response = client.generate(f"{config['ollama']['model']}", f"{config['ollama']['preamble']}\n{prompt}")
-        print(f"{ollama_config['model']}", f"{preamble}\n{prompt}")
+        print(f"{config['ollama']['model']}", f"{config['ollama']['preamble']}\n{prompt}")
         return response['response']
     except Exception as e:
         print(f"HTTP error occurred: {e}")
         return "Error contacting Ollama server."
 
-# Function to process mentions
+def main():
-def process_mentions():
+    config = load_config(CONFIG_PATH)
-    since_id = None
+    state = load_state()
-    while True:
-        mentions = mastodon.notifications(since_id=since_id) if since_id else mastodon.notifications()
-        for mention in mentions:
-            if mention['type'] == 'mention':
-                content = html2text.html2text(mention['status']['content'])
-                user = mention['status']['account']['acct']
-                status_id = mention['status']['id']
-                print(f"Mention from {user}: {content}")
 
-                response_text = ask_ollama(content)
+    masto = mastodon.Mastodon(
-                print(f"Ollama response: {response_text}")
+        api_base_url=config['mastodon']['host'],
+        access_token=config['mastodon']['access_token']
+    )
 
-                mastodon.status_post(
+    mentions = masto.notifications(mentions_only=True)
-                    status=f"@{user} {response_text}",
-                    in_reply_to_id=status_id,
-                    visibility=mastodon_config.get('default_visibility', 'public')
-                )
-                since_id = mention['id']
 
-        time.sleep(30)
+    for mention in mentions:
+        if mention['id'] in state:
+            continue
+
+        content = mention['status']['content']
+        username = mention['status']['account']['username']
+        prompt = html2text.html2text(content)
+        
+        if username != "maxheadroom":
+            continue
+
+        response = send_to_ollama(prompt, config)
+        if not response or config['mastodon']['bot_username'] in response:
+            continue
+
+        masto.status_post(
+            status=f"@{username} {response}",
+            in_reply_to_id=mention['status']['id'],
+            visibility=config['mastodon']['default_visibility']
+        )
+
+        state.add(mention['id'])
+
+    save_state(state)
 
 if __name__ == "__main__":
-    process_mentions()
+    while True:
+        main()
+        time.sleep(30)