feat: add rename and delete conversation APIs

database.py

@@ -177,3 +177,44 @@ def get_messages_paginated(conv_id: int, start_row: int, end_row: int):
     finally:
         if conn:
             conn.close()
+
+def rename_conversation(conv_id: int, new_name: str):
+    """Update a conversation's title."""
+    conn = None
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor()
+        cursor.execute(query_constants.RENAME_CONVERSATION, (new_name, conv_id))
+        updated = cursor.fetchone()
+        conn.commit()
+        return dict(updated) if updated else None
+    except Exception as e:
+        logger.error(f"Error renaming conversation: {e}")
+        if conn:
+            conn.rollback()
+        raise
+    finally:
+        if conn:
+            conn.close()
+
+def delete_conversation(conv_id: int):
+    """Delete a conversation and all its messages."""
+    conn = None
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor()
+        # Delete messages first (also handled by CASCADE, but explicit is safer)
+        cursor.execute(query_constants.DELETE_MESSAGES_BY_CONVERSATION, (conv_id,))
+        # Delete the conversation itself
+        cursor.execute(query_constants.DELETE_CONVERSATION, (conv_id,))
+        deleted = cursor.fetchone()
+        conn.commit()
+        return dict(deleted) if deleted else None
+    except Exception as e:
+        logger.error(f"Error deleting conversation: {e}")
+        if conn:
+            conn.rollback()
+        raise
+    finally:
+        if conn:
+            conn.close()

main.py

@@ -9,7 +9,7 @@ from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 
 from fastapi.responses import StreamingResponse
-from models.schemas import CreateConversationRequest, LoadMessagesRequest, AskQuestionRequest
+from models.schemas import CreateConversationRequest, LoadMessagesRequest, AskQuestionRequest, RenameConversationRequest, DeleteConversationRequest
 import database
 import model
 import time
@@ -184,6 +184,43 @@ async def ask_question(req: AskQuestionRequest):
         logger.error(f"Error processing question: {e}")
         raise HTTPException(status_code=500, detail="Internal server error")
 
+@app.put("/conversations/rename")
+async def rename_conversation(req: RenameConversationRequest):
+    """Renames an existing conversation."""
+    if not req.new_name or not req.new_name.strip():
+        raise HTTPException(status_code=400, detail="new_name must not be empty")
+
+    try:
+        conv = database.get_conversation(req.conversation_id)
+        if not conv:
+            raise HTTPException(status_code=404, detail="conversation_id not found")
+
+        updated = database.rename_conversation(req.conversation_id, req.new_name.strip())
+        if updated and 'created_at' in updated and updated['created_at']:
+            updated['created_at'] = updated['created_at'].isoformat()
+        return updated
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error(f"Error renaming conversation: {e}")
+        raise HTTPException(status_code=500, detail="Internal server error")
+
+@app.delete("/conversations/delete")
+async def delete_conversation(req: DeleteConversationRequest):
+    """Deletes a conversation and all its messages."""
+    try:
+        conv = database.get_conversation(req.conversation_id)
+        if not conv:
+            raise HTTPException(status_code=404, detail="conversation_id not found")
+
+        deleted = database.delete_conversation(req.conversation_id)
+        return {"message": "Conversation deleted successfully", "deleted_id": deleted["id"]}
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error(f"Error deleting conversation: {e}")
+        raise HTTPException(status_code=500, detail="Internal server error")
+
 @app.get("/health")
 async def health_check():
     """Health check endpoint."""

models/schemas.py

@@ -12,3 +12,10 @@ class AskQuestionRequest(BaseModel):
     conversation_id: int
     query: str
     max_tokens: int = 500
+
+class RenameConversationRequest(BaseModel):
+    conversation_id: int
+    new_name: str
+
+class DeleteConversationRequest(BaseModel):
+    conversation_id: int

query_constants.py

@@ -51,3 +51,9 @@ GET_PAGINATED_MESSAGES = """
             ORDER BY created_at ASC 
             OFFSET %s LIMIT %s;
             """
+
+RENAME_CONVERSATION = "UPDATE conversations SET title = %s WHERE id = %s RETURNING id, title, created_at;"
+
+DELETE_MESSAGES_BY_CONVERSATION = "DELETE FROM messages WHERE conversation_id = %s;"
+
+DELETE_CONVERSATION = "DELETE FROM conversations WHERE id = %s RETURNING id;"

requests.txt

@@ -102,3 +102,39 @@ curl --location 'https://adeebjamal-private-ai-backend.hf.space/conversations/me
 #         }
 #     ]
 # }
+
+
+# -----------------------------------------------------------------------------
+
+# 6. Rename a Conversation
+# Updates the title of an existing conversation.
+curl --location --request PUT 'https://adeebjamal-private-ai-backend.hf.space/conversations/rename' \
+--header 'Content-Type: application/json' \
+--data '{
+    "conversation_id": <REPLACE_WITH_CONVERSATION_ID>,
+    "new_name": "<REPLACE_WITH_NEW_NAME>"
+}'
+
+# Example Response:
+# {
+#     "id": 2,
+#     "title": "Updated Chat Title",
+#     "created_at": "2026-04-21T17:09:16.741250"
+# }
+
+
+# -----------------------------------------------------------------------------
+
+# 7. Delete a Conversation
+# Deletes a conversation and all its associated messages permanently.
+curl --location --request DELETE 'https://adeebjamal-private-ai-backend.hf.space/conversations/delete' \
+--header 'Content-Type: application/json' \
+--data '{
+    "conversation_id": <REPLACE_WITH_CONVERSATION_ID>
+}'
+
+# Example Response:
+# {
+#     "message": "Conversation deleted successfully",
+#     "deleted_id": 2
+# }