Update main.py

main.py

@@ -171,15 +171,31 @@ async def root():
                     "description": "Index multiple texts and images (NEW: up to 10 each)",
                     "content_type": "multipart/form-data",
                     "body": {
-                        "id": "string (required) - Document ID",
+                        "id": "string (required) - Document ID (primary)",
                         "texts": "List[string] (optional) - Up to 10 texts",
-                        "images": "List[UploadFile] (optional) - Up to 10 images"
+                        "images": "List[UploadFile] (optional) - Up to 10 images",
+                        "id_use": "string (optional) - ID của SocialMedia hoặc EventCode",
+                        "id_user": "string (optional) - ID của User"
                     },
-                    "example": "curl -X POST '/index' -F 'id=doc1' -F 'texts=Text 1' -F 'texts=Text 2' -F 'images=@img1.jpg'",
+                    "example": "curl -X POST '/index' -F 'id=doc1' -F 'id_use=social_123' -F 'id_user=user_789' -F 'texts=Text 1' -F 'images=@img1.jpg'",
                     "response": {
                         "success": True,
                         "id": "doc1",
                         "message": "Indexed successfully with 2 texts and 1 images"
+                    },
+                    "use_cases": {
+                        "social_media_post": {
+                            "id": "post_uuid_123",
+                            "id_use": "social_media_456",
+                            "id_user": "user_789",
+                            "description": "Link post to social media account and user"
+                        },
+                        "event_post": {
+                            "id": "post_uuid_789",
+                            "id_use": "event_code_ABC123",
+                            "id_user": "user_101",
+                            "description": "Link post to event and user"
+                        }
                     }
                 },
                 "POST /documents": {
@@ -417,20 +433,35 @@ async def root():
 async def index_data(
     id: str = Form(...),
     texts: Optional[List[str]] = Form(None),
-    images: Optional[List[UploadFile]] = File(None)
+    images: Optional[List[UploadFile]] = File(None),
+    id_use: Optional[str] = Form(None),
+    id_user: Optional[str] = Form(None)
 ):
     """
     Index data vào vector database (hỗ trợ nhiều texts và images)
 
     Body:
-    - id: Document ID (event ID, post ID, etc.)
+    - id: Document ID (primary ID)
     - texts: List of text contents (tiếng Việt supported) - Tối đa 10 texts
     - images: List of image files (optional) - Tối đa 10 images
+    - id_use: ID của SocialMedia hoặc EventCode (optional)
+    - id_user: ID của User (optional)
 
     Returns:
     - success: True/False
     - id: Document ID
     - message: Status message
+
+    Example:
+    ```bash
+    curl -X POST '/index' \
+      -F 'id=doc123' \
+      -F 'id_use=social_media_456' \
+      -F 'id_user=user_789' \
+      -F 'texts=Post content 1' \
+      -F 'texts=Post content 2' \
+      -F 'images=@image1.jpg'
+    ```
     """
     try:
         # Validation
@@ -490,7 +521,9 @@ async def index_data(
             "texts": texts if texts else [],
             "text_count": len(texts) if texts else 0,
             "image_count": len(images) if images else 0,
-            "image_filenames": [img.filename for img in images] if images else []
+            "image_filenames": [img.filename for img in images] if images else [],
+            "id_use": id_use if id_use else None,  # ID của SocialMedia hoặc EventCode
+            "id_user": id_user if id_user else None  # ID của User
         }
 
         result = qdrant_service.index_data(