Upload 12 files

auth.py

@@ -0,0 +1,14 @@
+
+from fastapi import Request, HTTPException
+from starlette.middleware.base import BaseHTTPMiddleware
+
+API_KEY = "super-secret-key"  # TODO: Secure this
+
+class APIKeyMiddleware(BaseHTTPMiddleware):
+    async def dispatch(self, request: Request, call_next):
+        if request.url.path.startswith("/"):
+            api_key = request.headers.get("x-api-key")
+            if api_key != API_KEY:
+                raise HTTPException(status_code=401, detail="Invalid API Key")
+        response = await call_next(request)
+        return response

logger.py

@@ -0,0 +1,13 @@
+
+import logging
+
+def setup_logger():
+    logger = logging.getLogger("autonomous_ai")
+    logger.setLevel(logging.INFO)
+    handler = logging.FileHandler("system.log")
+    formatter = logging.Formatter("[%(asctime)s] %(levelname)s: %(message)s")
+    handler.setFormatter(formatter)
+    logger.addHandler(handler)
+    return logger
+
+logger = setup_logger()

storage.py

@@ -0,0 +1,40 @@
+
+import sqlite3
+
+def init_db():
+    conn = sqlite3.connect("ai_data.db")
+    cursor = conn.cursor()
+    cursor.execute("CREATE TABLE IF NOT EXISTS goals (id INTEGER PRIMARY KEY AUTOINCREMENT, text TEXT)")
+    cursor.execute("CREATE TABLE IF NOT EXISTS feedback (id INTEGER PRIMARY KEY AUTOINCREMENT, score INTEGER, note TEXT)")
+    conn.commit()
+    conn.close()
+
+def save_goal(goal: str):
+    conn = sqlite3.connect("ai_data.db")
+    cursor = conn.cursor()
+    cursor.execute("INSERT INTO goals (text) VALUES (?)", (goal,))
+    conn.commit()
+    conn.close()
+
+def list_goals():
+    conn = sqlite3.connect("ai_data.db")
+    cursor = conn.cursor()
+    cursor.execute("SELECT text FROM goals")
+    rows = cursor.fetchall()
+    conn.close()
+    return "\n".join(row[0] for row in rows)
+
+def save_feedback(score: int, note: str = ""):
+    conn = sqlite3.connect("ai_data.db")
+    cursor = conn.cursor()
+    cursor.execute("INSERT INTO feedback (score, note) VALUES (?, ?)", (score, note))
+    conn.commit()
+    conn.close()
+
+def list_feedback():
+    conn = sqlite3.connect("ai_data.db")
+    cursor = conn.cursor()
+    cursor.execute("SELECT score, note FROM feedback")
+    rows = cursor.fetchall()
+    conn.close()
+    return "\n".join(f"{score} - {note}" for score, note in rows)