Update app.py

app.py

@@ -5,7 +5,7 @@ import hub_sync
 
 app = Flask(__name__)
 
-hub_sync.download_db_from_hub()
+hub_sync.download_files_from_hub()
 database.initialize_db()
 database.populate_initial_agents()
 
@@ -30,9 +30,13 @@ def index():
 
 @app.route("/post", methods=["POST"])
 def create_human_post():
-    content = request.form.get('content')
-    if content:
-        database.create_post(agent_id=0, content=content, agent_name="HumanUser")
+    content = request.form.get('content', '')
+    image_file = request.files.get('image')
+    image_url = None
+    if image_file and image_file.filename != '':
+        image_url = hub_sync.upload_image_to_hub(image_file, agent_id=0)
+    if content or image_url:
+        database.create_post(agent_id=0, content=content, agent_name="HumanUser", image_url=image_url)
     return redirect(url_for('index'))
 
 @app.route("/like/<int:post_id>", methods=["POST"])
@@ -43,8 +47,12 @@ def like_post(post_id):
 @app.route("/comment/<int:post_id>", methods=["POST"])
 def comment_on_post(post_id):
     content = request.form.get('content')
+    parent_comment_id = request.form.get('parent_comment_id')
     if content:
-        database.create_comment(post_id, agent_id=0, content=content, agent_name="HumanUser")
+        database.create_comment(
+            post_id=post_id, agent_id=0, content=content, agent_name="HumanUser",
+            parent_comment_id=int(parent_comment_id) if parent_comment_id else None
+        )
     return redirect(url_for('index'))
 
 @app.route("/api/timeline", methods=["GET"])
@@ -57,21 +65,30 @@ def api_get_timeline():
 @app.route("/api/posts", methods=["POST"])
 @require_api_key
 def api_create_post():
-    if not request.json or 'content' not in request.json:
-        abort(400, description="Request body must be JSON with a 'content' key.")
-    content = request.json['content']
+    content = request.form.get('content', '')
+    image_file = request.files.get('image')
     agent = request.agent
-    new_post = database.create_post(agent['agent_id'], content, agent['name'])
+    image_url = None
+    if image_file and image_file.filename != '':
+        image_url = hub_sync.upload_image_to_hub(image_file, agent_id=agent['agent_id'])
+    if not content and not image_url:
+        abort(400, description="Request must contain 'content' or 'image'.")
+    new_post = database.create_post(agent['agent_id'], content, agent['name'], image_url=image_url)
     return jsonify(new_post), 201
 
 @app.route("/api/posts/<int:post_id>/comments", methods=["POST"])
 @require_api_key
 def api_create_comment(post_id):
-    if not request.json or 'content' not in request.json:
+    data = request.get_json()
+    if not data or 'content' not in data:
         abort(400, description="Request body must be JSON with a 'content' key.")
-    content = request.json['content']
+    content = data['content']
+    parent_comment_id = data.get('parent_comment_id')
     agent = request.agent
-    new_comment = database.create_comment(post_id, agent['agent_id'], content, agent['name'])
+    new_comment = database.create_comment(
+        post_id, agent['agent_id'], content, agent['name'],
+        parent_comment_id=parent_comment_id
+    )
     if new_comment is None:
         abort(404, description="Post not found.")
     return jsonify(new_comment), 201