Update main.py

main.py

@@ -109,34 +109,218 @@ class AddDocumentResponse(BaseModel):
 
 @app.get("/")
 async def root():
-    """Health check endpoint"""
+    """Health check endpoint with comprehensive API documentation"""
     return {
         "status": "running",
-        "service": "Event Social Media Embeddings & ChatbotRAG API",
-        "embedding_model": "Jina CLIP v2",
+        "service": "ChatbotRAG API",
+        "version": "2.0.0",
         "vector_db": "Qdrant",
-        "language_support": "Vietnamese + 88 other languages",
+        "document_db": "MongoDB",        
         "endpoints": {
-            "embeddings": {
-                "POST /index": "Index data với text/image",
-                "POST /search": "Hybrid search",
-                "POST /search/text": "Text search",
-                "POST /search/image": "Image search",
-                "DELETE /delete/{doc_id}": "Delete document",
-                "GET /document/{doc_id}": "Get document",
-                "GET /stats": "Collection statistics"
-            },
             "chatbot_rag": {
-                "POST /chat": "Chat với RAG",
-                "POST /documents": "Add document to knowledge base",
-                "POST /rag/search": "Search in knowledge base",
-                "GET /history": "Get chat history",
-                "DELETE /documents/{doc_id}": "Delete document from knowledge base"
+                "POST /chat": {
+                    "description": "Chat với AI sử dụng RAG (Retrieval-Augmented Generation)",
+                    "request": {
+                        "method": "POST",
+                        "content_type": "application/json",
+                        "body": {
+                            "message": "string (required) - User message/question",
+                            "use_rag": "boolean (optional, default: true) - Enable RAG context retrieval",
+                            "top_k": "integer (optional, default: 3) - Number of context documents to retrieve",
+                            "system_message": "string (optional) - Custom system prompt",
+                            "max_tokens": "integer (optional, default: 512) - Max response length",
+                            "temperature": "float (optional, default: 0.7, range: 0-1) - Creativity level",
+                            "top_p": "float (optional, default: 0.95) - Nucleus sampling",
+                            "hf_token": "string (optional) - Hugging Face token (fallback to env)"
+                        }
+                    },
+                    "response": {
+                        "response": "string - AI generated response",
+                        "context_used": [
+                            {
+                                "id": "string - Document ID",
+                                "confidence": "float - Relevance score",
+                                "metadata": {
+                                    "text": "string - Retrieved context"
+                                }
+                            }
+                        ],
+                        "timestamp": "string - ISO 8601 timestamp"
+                    },
+                    "example_request": {
+                        "message": "Dao có nguy hiểm không?",
+                        "use_rag": True,
+                        "top_k": 3,
+                        "temperature": 0.7
+                    },
+                    "example_response": {
+                        "response": "Dựa trên thông tin trong database, dao được phân loại là vũ khí nguy hiểm. Dao sắc có thể gây thương tích nghiêm trọng nếu không sử dụng đúng cách. Cần tuân thủ các quy định an toàn khi sử dụng.",
+                        "context_used": [
+                            {
+                                "id": "68a3fc14c853d7621e8977b5",
+                                "confidence": 0.92,
+                                "metadata": {
+                                    "text": "Vũ khí"
+                                }
+                            },
+                            {
+                                "id": "68a3fc4cc853d7621e8977b6",
+                                "confidence": 0.85,
+                                "metadata": {
+                                    "text": "Con dao sắc"
+                                }
+                            }
+                        ],
+                        "timestamp": "2025-10-13T10:30:45.123456"
+                    },
+                    "notes": [
+                        "RAG retrieves relevant context from vector DB before generating response",
+                        "LLM uses context to provide accurate, grounded answers",
+                        "Requires HUGGINGFACE_TOKEN environment variable or hf_token in request"
+                    ]
+                },
+                "POST /documents": {
+                    "description": "Add document to knowledge base for RAG",
+                    "request": {
+                        "method": "POST",
+                        "content_type": "application/json",
+                        "body": {
+                            "text": "string (required) - Document text content",
+                            "metadata": "object (optional) - Additional metadata (source, category, etc.)"
+                        }
+                    },
+                    "response": {
+                        "success": "boolean",
+                        "doc_id": "string - MongoDB ObjectId",
+                        "message": "string - Status message"
+                    },
+                    "example_request": {
+                        "text": "Để tạo event mới: Click nút 'Tạo Event' ở góc trên bên phải màn hình. Điền thông tin sự kiện bao gồm tên, ngày giờ, địa điểm. Click Lưu để hoàn tất.",
+                        "metadata": {
+                            "source": "user_guide.pdf",
+                            "section": "create_event",
+                            "page": 5,
+                            "category": "tutorial"
+                        }
+                    },
+                    "example_response": {
+                        "success": True,
+                        "doc_id": "67a9876543210fedcba98765",
+                        "message": "Document added successfully with ID: 67a9876543210fedcba98765"
+                    }
+                },
+                "POST /rag/search": {
+                    "description": "Search in knowledge base (similar to /search/text but for RAG documents)",
+                    "request": {
+                        "method": "POST",
+                        "content_type": "multipart/form-data",
+                        "body": {
+                            "query": "string (required) - Search query",
+                            "top_k": "integer (optional, default: 5) - Number of results",
+                            "score_threshold": "float (optional, default: 0.5) - Minimum relevance score"
+                        }
+                    },
+                    "response": [
+                        {
+                            "id": "string",
+                            "confidence": "float",
+                            "metadata": {
+                                "text": "string",
+                                "source": "string"
+                            }
+                        }
+                    ],
+                    "example_request": {
+                        "query": "cách tạo sự kiện mới",
+                        "top_k": 3,
+                        "score_threshold": 0.6
+                    }
+                },
+                "GET /history": {
+                    "description": "Get chat conversation history",
+                    "request": {
+                        "method": "GET",
+                        "query_params": {
+                            "limit": "integer (optional, default: 10) - Number of messages",
+                            "skip": "integer (optional, default: 0) - Pagination offset"
+                        }
+                    },
+                    "response": {
+                        "history": [
+                            {
+                                "user_message": "string",
+                                "assistant_response": "string",
+                                "context_used": "array",
+                                "timestamp": "string - ISO 8601"
+                            }
+                        ],
+                        "total": "integer - Total messages count"
+                    },
+                    "example_request": "GET /history?limit=5&skip=0",
+                    "example_response": {
+                        "history": [
+                            {
+                                "user_message": "Dao có nguy hiểm không?",
+                                "assistant_response": "Dao được phân loại là vũ khí...",
+                                "context_used": [],
+                                "timestamp": "2025-10-13T10:30:45.123456"
+                            }
+                        ],
+                        "total": 15
+                    }
+                },
+                "DELETE /documents/{doc_id}": {
+                    "description": "Delete document from knowledge base",
+                    "request": {
+                        "method": "DELETE",
+                        "path_params": {
+                            "doc_id": "string - MongoDB ObjectId"
+                        }
+                    },
+                    "response": {
+                        "success": "boolean",
+                        "message": "string"
+                    }
+                }
             }
+        },
+        "usage_examples": {
+            "curl_chat": "curl -X POST 'http://localhost:8000/chat' -H 'Content-Type: application/json' -d '{\"message\": \"Dao có nguy hiểm không?\", \"use_rag\": true}'",
+            "python_chat": """
+import requests
+
+response = requests.post(
+    'http://localhost:8000/chat',
+    json={
+        'message': 'Nút tạo event ở đâu?',
+        'use_rag': True,
+        'top_k': 3
+    }
+)
+print(response.json()['response'])
+            """
+        },
+        "authentication": {
+            "embeddings_apis": "No authentication required",
+            "chat_api": "Requires HUGGINGFACE_TOKEN (env variable or request body)"
+        },
+        "rate_limits": {
+            "embeddings": "No limit",
+            "chat_with_llm": "Limited by Hugging Face API (free tier: ~1000 requests/hour)"
+        },
+        "error_codes": {
+            "400": "Bad Request - Missing required fields or invalid input",
+            "401": "Unauthorized - Invalid Hugging Face token",
+            "404": "Not Found - Document ID not found",
+            "500": "Internal Server Error - Server or database error"
+        },
+        "links": {
+            "docs": "http://localhost:8000/docs",
+            "redoc": "http://localhost:8000/redoc",
+            "openapi": "http://localhost:8000/openapi.json"
         }
     }
 
-
 @app.post("/index", response_model=IndexResponse)
 async def index_data(
     id: str = Form(...),