Merge pull request #3 from YSocialTwin/multimodal

Pages, Multimodal posts and Dynamic Interests

Author: GiulioRossetti

README.md

@@ -6,6 +6,8 @@ This repository contains the code for the **server-side** of the application.
 
 The client-side code can be found [here](https://github.com/YSocialTwin/YClient)
 
+For more information, please refer to the official [documentation](https://ysocialtwin.github.io/)
+
 #### Features
 
 - Realistic agent simulations using LLMs

config_files/exp_config.json

@@ -4,5 +4,5 @@
   "port": 5010,
   "debug": "True",
   "reset_db": "True",
-  "modules": ["news", "voting"]
+  "modules": ["news", "voting", "image"]
 }

data_schema/database_clean_server.db

@@ -25,6 +25,8 @@ class User_mgmt(UserMixin, db.Model):
     nationality = db.Column(db.String(15), default=None)
     round_actions = db.Column(db.Integer, default=3)
     toxicity = db.Column(db.String(10), default="no")
+    is_page = db.Column(db.Integer, default=0)
+    left_on = db.Column(db.Integer, default=None)
 
     posts = db.relationship("Post", backref="author", lazy=True)
     liked = db.relationship("Reactions", backref="liked_by", lazy=True)
@@ -39,6 +41,7 @@ class Post(db.Model):
     comment_to = db.Column(db.Integer, default=-1)
     thread_id = db.Column(db.Integer)
     news_id = db.Column(db.String(50), db.ForeignKey("articles.id"), default=None)
+    image_id = db.Column(db.Integer(), db.ForeignKey("images.id"), default=None)
     shared_from = db.Column(db.Integer, default=-1)
 
 
@@ -146,3 +149,16 @@ class Post_topics(db.Model):
     id = db.Column(db.Integer, primary_key=True)
     post_id = db.Column(db.Integer, db.ForeignKey("post.id"), nullable=False)
     topic_id = db.Column(db.Integer, db.ForeignKey("interests.iid"), nullable=False)
+
+
+class Images(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    url = db.Column(db.String(200), nullable=True)
+    description = db.Column(db.String(400), nullable=True)
+    article_id = db.Column(db.Integer, db.ForeignKey("articles.id"), nullable=True)
+
+
+class Article_topics(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    article_id = db.Column(db.Integer, db.ForeignKey("articles.id"), nullable=False)
+    topic_id = db.Column(db.Integer, db.ForeignKey("interests.iid"), nullable=False)

requirements_server.txt

@@ -16,7 +16,6 @@
     Emotions,
     Post_emotions,
     Post_topics,
-    Interests,
 )
 
 
@@ -33,9 +32,11 @@ def read():
     vround = int(data["visibility_rounds"])
     try:
         uid = int(data["uid"])
-        fratio = float(data["followers_ratio"])
     except:
         uid = None
+    try:
+        fratio = float(data["followers_ratio"])
+    except:
         fratio = 1
     articles = False
     if "article" in data:
@@ -50,14 +51,14 @@ def read():
         if articles:
             posts = [
                 db.session.query(Post)
-                .filter(Post.round >= visibility, Post.news_id != -1)
+                .filter(Post.round >= visibility, Post.news_id != -1, Post.user_id != uid)
                 .order_by(desc(Post.id))
                 .limit(10)
             ]
         else:
             posts = [
                 db.session.query(Post)
-                .filter(Post.round >= visibility)
+                .filter(Post.round >= visibility, Post.user_id != uid)
                 .order_by(desc(Post.id))
                 .limit(10)
             ]
@@ -107,6 +108,7 @@ def read():
                 Post.query.filter(
                     Post.round >= visibility,
                     Post.news_id != -1,
+                    Post.user_id != uid,
                     Post.user_id.in_(follower_ids),
                 )
                 .order_by(desc(Post.id))
@@ -180,33 +182,33 @@ def read():
         if additional_posts_limit != 0:
             if articles:
                 additional_posts = (
-                    Post.query.filter(Post.round >= visibility, Post.news_id != -1)
+                    Post.query.filter(Post.round >= visibility, Post.news_id != -1, Post.user_id != uid)
                     .order_by(desc(Post.id))
                     .limit(additional_posts_limit)
                 )
             else:
                 additional_posts = (
-                    Post.query.filter(Post.round >= visibility)
+                    Post.query.filter(Post.round >= visibility, Post.user_id != uid)
                     .order_by(desc(Post.id))
                     .limit(additional_posts_limit)
                 )
 
-        posts = [posts, additional_posts]
+            posts = [posts, additional_posts]
 
     else:
         # get posts in random order
         if articles:
             posts = [
                 (
-                    Post.query.filter(Post.round >= visibility, Post.news_id != -1)
+                    Post.query.filter(Post.round >= visibility, Post.news_id != -1, Post.user_id != uid)
                     .order_by(func.random())
                     .limit(limit)
                 )
             ]
         else:
             posts = [
                 (
-                    Post.query.filter(Post.round >= visibility)
+                    Post.query.filter(Post.round >= visibility, Post.user_id != uid)
                     .order_by(func.random())
                     .limit(limit)
                 )

y_server/modals.py

@@ -16,6 +16,8 @@
     Interests,
     Post_topics,
     User_interest,
+    Images,
+    Article_topics,
 )
 
 
@@ -43,5 +45,7 @@ def reset_experiment():
     db.session.query(User_interest).delete()
     db.session.query(Voting).delete()
     db.session.query(Post_topics).delete()
+    db.session.query(Images).delete()
+    db.session.query(Article_topics).delete()
     db.session.commit()
     return {"status": 200}

y_server/routes/content_management.py

@@ -0,0 +1,84 @@
+import json
+from flask import request
+from y_server import app, db
+from y_server.modals import (
+    Images,
+    Post,
+    Emotions,
+    Post_emotions,
+    Hashtags,
+    Post_hashtags,
+)
+
+
+@app.route("/comment_image", methods=["POST"])
+def post_image():
+    """
+    Comment on an image.
+
+    :return:
+    """
+    data = json.loads(request.get_data())
+    account_id = int(data["user_id"])
+    text = data["text"].strip('"')
+    emotions = data["emotions"]
+    hashtags = data["hashtags"]
+    tid = int(data["tid"])
+    image_url = data["image_url"]
+    image_description = data["image_description"]
+    article_id = int(data["article_id"])
+
+    # check if image exists
+    image = Images.query.filter_by(url=image_url).first()
+    if image is None:
+        image = Images(
+            url=image_url,
+            description=image_description,
+            article_id=article_id,
+        )
+        db.session.add(image)
+        db.session.commit()
+
+    # get image id
+    image_id = Images.query.filter_by(url=image_url).first().id
+
+    post = Post(
+        tweet=text,
+        round=tid,
+        user_id=account_id,
+        image_id=image_id,
+        comment_to=-1,
+    )
+
+    db.session.add(post)
+    db.session.commit()
+
+    post.thread_id = post.id
+    db.session.commit()
+
+    for emotion in emotions:
+        if len(emotion) < 1:
+            continue
+
+        em = Emotions.query.filter_by(emotion=emotion).first()
+        if em is not None:
+            post_emotion = Post_emotions(post_id=post.id, emotion_id=em.id)
+            db.session.add(post_emotion)
+            db.session.commit()
+
+    for tag in hashtags:
+        if len(tag) < 1:
+            continue
+
+        ht = Hashtags.query.filter_by(hashtag=tag).first()
+        if ht is None:
+            ht = Hashtags(hashtag=tag)
+            db.session.add(ht)
+            db.session.commit()
+            ht = Hashtags.query.filter_by(hashtag=tag).first()
+
+        post_tag = Post_hashtags(post_id=post.id, hashtag_id=ht.id)
+        db.session.add(post_tag)
+        db.session.commit()
+
+    return json.dumps({"status": 200})