add comment feature

.gitignore

@@ -0,0 +1,2 @@
+.env
+client.py

app.py

@@ -4,6 +4,12 @@ import requests
 import orjson
 import os
 
+# IMPORTANT: REMOVE THIS WHEN PUSHING TO GIT
+from dotenv import load_dotenv
+
+load_dotenv()
+
+
 db = NetworkDB(os.getenv("DATABASE_URL"))
 
 
@@ -18,7 +24,7 @@ def get_query_embeddings(content: str) -> list[float]:
     return embeddings
 
 
-async def post(content: str) -> bool:
+async def post_text(content: str) -> bool:
     """Posts a text post in the database, and returns True if it was successfuly posted"""
     content = content.strip(" ").strip("\n")
     try:
@@ -33,12 +39,7 @@ async def post(content: str) -> bool:
         )
         res = orjson.loads(embeddings.content)
         embeddings = res["embeddings"][0]  # A list
-        # await ctx.request_context.session.send_log_message(
-        #     level="info",
-        #     data=f"{embeddings}",
-        #     )
         res = await db.post_text(content, embeddings)
-        # Add to database.
         return res
     except gr.Error as e:
         raise e
@@ -46,15 +47,14 @@ async def post(content: str) -> bool:
         return False
 
 
-async def retrieve_post() -> str:
-    """Retrieves a random text post from database"""
-    # Retrive a post from the database.
+async def retrieve_random_text_post() -> str:
+    """Retrieves a random text post and its id from the database"""
     post = await db.get_text_post_random()
     return post
 
 
-async def retrieve_similar_post(query: str) -> str:
-    """Retrieves a post semantically similar to the query through Vector Similarity"""
+async def retrieve_similar_text_post(query: str) -> str:
+    """Retrieves a text post and its id semantically similar to the query through Vector Similarity"""
     query = query.strip(" ").strip("\n")
     try:
         if query == "":
@@ -68,7 +68,22 @@ async def retrieve_similar_post(query: str) -> str:
         return f"Unexpected Error. Are you using the correct API?"
 
 
-# socialnet = gr.Interface(retrieve_post, inputs=None, outputs="textbox")
+async def get_text_post_comments(post_id: int) -> str:
+    """Retrieves latest 5 comments from the text post with id post_id"""
+    try:
+        comments = await db.get_text_post_comments(post_id)
+        return comments
+    except Exception as e:
+        return f"Unexpected Error!"
+
+async def comment_on_text_post(post_id: int, content: str) -> bool:
+    """Adds a text comment to the text post with id post_id. Returns True if successful"""
+    try:
+        success = await db.comment_on_text_post(post_id, content)
+        return success 
+    except Exception as e:
+        return False
+
 socialnet = gr.Blocks()
 with socialnet:
     gr.Markdown(
@@ -85,16 +100,18 @@ with socialnet:
                 label="Your Post (`Shift + Enter` for new line)",
                 max_length=2000,
             )
-            outputs = gr.Checkbox(label="Success")
+            outputs = gr.Checkbox(value=False, label="Success")
             submit_btn = gr.Button(value="Post")
-            submit_btn.click(post, inputs=text_input, outputs=outputs)
+            submit_btn.click(post_text, inputs=text_input, outputs=outputs)
+
         with gr.TabItem("Retrieve Simple"):
             gr.Markdown("Retrieve a Random Post!")
             text_output = gr.Textbox(
                 placeholder="Post will appear here!", label="Output"
             )
             submit_btn = gr.Button("Retrieve")
-            submit_btn.click(retrieve_post, inputs=None, outputs=text_output)
+            submit_btn.click(retrieve_random_text_post, inputs=None, outputs=text_output)
+
         with gr.TabItem("Retrieve Advanced"):
             gr.Markdown(
                 "Retrieve using query, uses semantic search using Vector Similarity"
@@ -107,8 +124,28 @@ with socialnet:
             )
             submit_btn = gr.Button("Retrieve")
             submit_btn.click(
-                retrieve_similar_post, inputs=text_input, outputs=text_output
+                retrieve_similar_text_post, inputs=text_input, outputs=text_output
             )
+
+        with gr.TabItem("View Comments"):
+            gr.Markdown("Get Comments of a Post")
+            id_input = gr.Number(label="Post id")
+            text_output = gr.Textbox(
+                placeholder="Comments will appear here!", label="Output"
+            )
+            submit_btn = gr.Button("Retrieve")
+            submit_btn.click(
+                get_text_post_comments, inputs=id_input, outputs=text_output
+            )
+
+        with gr.TabItem("Post Comment"):
+            gr.Markdown("Post a comment!")
+            id_input = gr.Number(label="Post id")
+            text_input = gr.Textbox(placeholder="Type your comment here", label="Comment")
+            success = gr.Checkbox(value=False, label="Success")
+            submit_btn = gr.Button(value="Comment")
+            submit_btn.click(comment_on_text_post, inputs=[id_input, text_input], outputs=success)
+
         with gr.TabItem("Usage in Clients"):
             gr.Markdown(
                 "To add this MCP to clients that support SSE (eg. Cursor, Windsurf, Cline), add the following to your MCP Config"
@@ -158,7 +195,9 @@ with socialnet:
 </div>
 """
             )
-            gr.Markdown("""Want to use it in your Claude Desktop? Add this to your **claude_desktop_config.json**""")
+            gr.Markdown(
+                """Want to use it in your Claude Desktop? Add this to your **claude_desktop_config.json**"""
+            )
             gr.Code(
                 """{
   "mcpServers": {
@@ -176,6 +215,5 @@ with socialnet:
             )
 
 
-
 if __name__ == "__main__":
     socialnet.launch(mcp_server=True)

database.py

@@ -31,30 +31,61 @@ class NetworkDB:
     async def get_text_post_random(self) -> str:
         try:
             conn = await asyncpg.connect(self.database_url)
-            post = await conn.fetchval(
-                "SELECT content from text_posts ORDER BY random() LIMIT 1"
+            id, post = await conn.fetchval(
+                "SELECT (id, content) from text_posts ORDER BY random() LIMIT 1"
             )
             await conn.close()
-            return post if post is not None else "[Internal Message: No post found!]"
+            if post is not None:
+                formatted_post = f"Post id: {id}\n{post}"
+                return formatted_post
+            return "[Internal Message: No post found!]"
         except Exception as e:
             return "[Internal Message: Server Error]"
 
     async def get_text_post_similar(self, query_embedding: list[float]) -> str:
         try:
             conn = await asyncpg.connect(self.database_url)
-            post = await conn.fetchval(
-                "SELECT content FROM text_posts ORDER BY embedding <-> $1 LIMIT 1",
+            id, post = await conn.fetchval(
+                "SELECT (id, content) FROM text_posts ORDER BY embedding <-> $1 LIMIT 1",
                 f"{query_embedding}",
             )
             await conn.close()
-            return (
-                post
-                if post is not None
-                else "[Internal Message: No similar post found!]"
-            )
+            if post is not None:
+                formatted_post = f"Post id: {id}\n{post}"
+                return formatted_post
+            return "[Internal Message: No similar post found!]"
         except Exception as e:
             return "[Internal Message: Server Error]"
 
+    async def get_text_post_comments(self, post_id: int) -> str:
+        try:
+            conn = await asyncpg.connect(self.database_url)
+            comments = await conn.fetch(
+                "SELECT content FROM text_posts_comments WHERE post_id = $1 ORDER BY uploaded_at DESC LIMIT 5",
+                post_id
+            )
+            await conn.close()
+            if len(comments) == 0:
+                return "[Internal Message: No Comments on this post]"
+            formatted_comments = ""
+            for i, comment in enumerate(comments):
+                # Only add new lines before the comments. So last comment won't have extra new lines. Don't add before first comment obviously
+                if i > 0:
+                    formatted_comments += "\n\n"
+                formatted_comments += f"<|Comment_{i}|>\n{comment['content']}"
+            return formatted_comments
+        except Exception as e:
+            return ["Internal Message: Server Error"]
+    
+    async def comment_on_text_post(self, post_id: int, content: str) -> bool:
+        try:
+            conn = await asyncpg.connect(self.database_url)
+            success = await conn.fetchval("INSERT INTO text_posts_comments (post_id, content) VALUES ($1, $2) RETURNING id", post_id, content)
+            await conn.close()
+            return False if success is None else True
+        except Exception as e:
+            return False
+
     async def disconnect(self) -> None:
         if self.pool:
             self.pool.close()