Add latest posts feature

app.py

@@ -52,6 +52,10 @@ async def retrieve_random_text_post() -> str:
     post = await db.get_text_post_random()
     return post
 
+async def retrieve_latest_text_posts() -> str:
+    """Retrieves latest 5 text posts with their ids from the database"""
+    posts = await db.get_text_posts_latest()
+    return posts
 
 async def retrieve_similar_text_post(query: str) -> str:
     """Retrieves a text post and its id semantically similar to the query through Vector Similarity"""
@@ -121,6 +125,12 @@ with socialnet:
             submit_btn = gr.Button("Retrieve")
             submit_btn.click(retrieve_random_text_post, inputs=None, outputs=text_output)
 
+        with gr.TabItem("Retrieve Latest"):
+            gr.Markdown("Retrieve latest 5 posts!")
+            text_output = gr.Textbox(placeholder="Posts will appear here!", label="Output")
+            submit_btn = gr.Button("Retrieve")
+            submit_btn.click(retrieve_latest_text_posts, inputs=None, outputs=text_output)
+
         with gr.TabItem("Retrieve Advanced"):
             gr.Markdown(
                 "Retrieve using query, uses semantic search using Vector Similarity"

database.py

@@ -36,10 +36,29 @@ class NetworkDB:
             )
             await conn.close()
             if post is not None:
-                formatted_post = f"Post id: {id}\n{post}"
+                formatted_post = f"<|PostId_{id}|>\n{post}"
                 return formatted_post
             return "[Internal Message: No post found!]"
         except Exception as e:
+            print(f"Unexpected Error: {e}")
+            return "[Internal Message: Server Error]"
+
+    async def get_text_posts_latest(self) -> str:
+        try:
+            conn = await asyncpg.connect(self.database_url)
+            posts = await conn.fetch("SELECT (id, content) from text_posts ORDER BY uploaded_at DESC LIMIT 5")
+            await conn.close()
+            if len(posts) == 0:
+                return "[Internal Message: No posts in the database]"
+            formatted_posts = "" 
+            for i, post in enumerate(posts):
+                post = post[0]
+                if i > 0:
+                    formatted_posts += "\n\n"
+                formatted_posts += f'<|PostId_{post[0]}|>\n{post[1]}'
+            return formatted_posts
+        except Exception as e:
+            print(f"Unexpected Error: {e}")
             return "[Internal Message: Server Error]"
 
     async def get_text_post_similar(self, query_embedding: list[float]) -> str: