Update database.py

database.py

@@ -18,13 +18,14 @@ def initialize_db():
     cursor.execute("""
     CREATE TABLE IF NOT EXISTS posts (
         post_id INTEGER PRIMARY KEY, agent_id INTEGER NOT NULL,
-        content TEXT NOT NULL, timestamp TEXT NOT NULL
+        content TEXT, timestamp TEXT NOT NULL, image_url TEXT
     )""")
     cursor.execute("""
     CREATE TABLE IF NOT EXISTS comments (
         comment_id INTEGER PRIMARY KEY, post_id INTEGER NOT NULL, agent_id INTEGER NOT NULL,
-        content TEXT NOT NULL, timestamp TEXT NOT NULL,
-        FOREIGN KEY (post_id) REFERENCES posts (post_id)
+        content TEXT NOT NULL, timestamp TEXT NOT NULL, parent_comment_id INTEGER,
+        FOREIGN KEY (post_id) REFERENCES posts (post_id),
+        FOREIGN KEY (parent_comment_id) REFERENCES comments (comment_id)
     )""")
     cursor.execute("""
     CREATE TABLE IF NOT EXISTS likes (
@@ -83,7 +84,7 @@ def get_timeline(limit: int = 20) -> List[Dict[str, Any]]:
     conn.row_factory = sqlite3.Row
     cursor = conn.cursor()
     query = """
-    SELECT p.post_id, p.content, p.timestamp, p.agent_id, a.name AS author_name,
+    SELECT p.post_id, p.content, p.timestamp, p.agent_id, p.image_url, a.name AS author_name,
            (SELECT COUNT(*) FROM likes WHERE post_id = p.post_id) AS likes_count,
            (SELECT COUNT(*) FROM comments WHERE post_id = p.post_id) AS comments_count
     FROM posts p LEFT JOIN agents a ON p.agent_id = a.agent_id
@@ -96,6 +97,7 @@ def get_timeline(limit: int = 20) -> List[Dict[str, Any]]:
             "post_id": row["post_id"],
             "author": {"agent_id": row["agent_id"], "name": author_name},
             "content": row["content"],
+            "image_url": row["image_url"],
             "timestamp": datetime.datetime.fromisoformat(row["timestamp"]),
             "stats": {"likes": row["likes_count"], "comments": row["comments_count"]}
         })
@@ -111,37 +113,50 @@ def get_posts_with_details(limit: int = 20) -> List[Dict[str, Any]]:
     cursor = conn.cursor()
     query = f"SELECT c.*, a.name AS author_name FROM comments c LEFT JOIN agents a ON c.agent_id = a.agent_id WHERE c.post_id IN ({','.join('?' for _ in post_ids)}) ORDER BY c.timestamp ASC"
     cursor.execute(query, post_ids)
-    comments_by_post_id = {}
+    
+    comments_map = {}
     for row in cursor.fetchall():
-        post_id = row['post_id']
-        if post_id not in comments_by_post_id: comments_by_post_id[post_id] = []
+        comment_id = row['comment_id']
         author_name = row['author_name'] if row['author_name'] else "HumanUser"
-        comments_by_post_id[post_id].append({
-            "comment_id": row['comment_id'],
-            "author": {"agent_id": row['agent_id'], "name": author_name},
-            "content": row['content']
-        })
+        comments_map[comment_id] = {
+            "comment_id": comment_id, "post_id": row['post_id'], "parent_comment_id": row['parent_comment_id'],
+            "author": {"agent_id": row['agent_id'], "name": author_name}, "content": row['content'], "replies": []
+        }
+    
+    nested_comments = {}
+    for cid, comment in comments_map.items():
+        if comment['parent_comment_id'] and comment['parent_comment_id'] in comments_map:
+            comments_map[comment['parent_comment_id']]['replies'].append(comment)
+        else:
+            post_id = comment['post_id']
+            if post_id not in nested_comments: nested_comments[post_id] = []
+            nested_comments[post_id].append(comment)
+
     conn.close()
     for post in posts:
-        post['comments'] = comments_by_post_id.get(post['post_id'], [])
+        post['comments'] = nested_comments.get(post['post_id'], [])
     return posts
 
-def create_post(agent_id: int, content: str, agent_name: str) -> Dict[str, Any]:
+def create_post(agent_id: int, content: str, agent_name: str, image_url: Optional[str] = None) -> Dict[str, Any]:
     conn = sqlite3.connect(DB_FILE, check_same_thread=False)
     cursor = conn.cursor()
     timestamp = datetime.datetime.utcnow().isoformat()
-    cursor.execute("INSERT INTO posts (agent_id, content, timestamp) VALUES (?, ?, ?)",
-                   (agent_id, content, timestamp))
+    cursor.execute("INSERT INTO posts (agent_id, content, timestamp, image_url) VALUES (?, ?, ?, ?)",
+                   (agent_id, content, timestamp, image_url))
     post_id = cursor.lastrowid
     conn.commit()
     conn.close()
-    hub_sync.sync_db_to_hub()
+    
+    event = {"event_type": "create_post", "agent_id": agent_id, "post_id": post_id, "content": content, "image_url": image_url, "timestamp": timestamp}
+    hub_sync.log_interaction_to_dataset(event)
+    hub_sync.sync_files_to_hub()
+    
     return {
-        "post_id": post_id, "author": {"agent_id": agent_id, "name": agent_name}, "content": content,
+        "post_id": post_id, "author": {"agent_id": agent_id, "name": agent_name}, "content": content, "image_url": image_url,
         "timestamp": datetime.datetime.fromisoformat(timestamp), "stats": {"likes": 0, "comments": 0}
     }
 
-def create_comment(post_id: int, agent_id: int, content: str, agent_name: str) -> Optional[Dict[str, Any]]:
+def create_comment(post_id: int, agent_id: int, content: str, agent_name: str, parent_comment_id: Optional[int] = None) -> Optional[Dict[str, Any]]:
     conn = sqlite3.connect(DB_FILE, check_same_thread=False)
     cursor = conn.cursor()
     cursor.execute("SELECT post_id FROM posts WHERE post_id = ?", (post_id,))
@@ -149,12 +164,16 @@ def create_comment(post_id: int, agent_id: int, content: str, agent_name: str) -
         conn.close()
         return None
     timestamp = datetime.datetime.utcnow().isoformat()
-    cursor.execute("INSERT INTO comments (post_id, agent_id, content, timestamp) VALUES (?, ?, ?, ?)",
-                   (post_id, agent_id, content, timestamp))
+    cursor.execute("INSERT INTO comments (post_id, agent_id, content, timestamp, parent_comment_id) VALUES (?, ?, ?, ?, ?)",
+                   (post_id, agent_id, content, timestamp, parent_comment_id))
     comment_id = cursor.lastrowid
     conn.commit()
     conn.close()
-    hub_sync.sync_db_to_hub()
+    
+    event = {"event_type": "create_comment", "agent_id": agent_id, "post_id": post_id, "comment_id": comment_id, "parent_comment_id": parent_comment_id, "content": content, "timestamp": timestamp}
+    hub_sync.log_interaction_to_dataset(event)
+    hub_sync.sync_files_to_hub()
+
     return {
         "comment_id": comment_id, "author": {"agent_id": agent_id, "name": agent_name},
         "content": content, "timestamp": datetime.datetime.fromisoformat(timestamp)
@@ -167,11 +186,13 @@ def create_like(post_id: int, agent_id: int) -> bool:
     try:
         cursor.execute("INSERT INTO likes (post_id, agent_id) VALUES (?, ?)", (post_id, agent_id))
         conn.commit()
-        success = True
+        success = cursor.rowcount > 0
     except sqlite3.IntegrityError:
         success = True
     finally:
         conn.close()
     if success:
-        hub_sync.sync_db_to_hub()
+        event = {"event_type": "create_like", "agent_id": agent_id, "post_id": post_id, "timestamp": datetime.datetime.utcnow().isoformat()}
+        hub_sync.log_interaction_to_dataset(event)
+        hub_sync.sync_files_to_hub()
     return success