Update category matching to 4 types: Ai BF, Ai GF, Ai Hug Video, Ai Kiss Video

README.md

@@ -98,14 +98,16 @@ Response:
 **Form Data:**
 - `image`: Image file (JPG, PNG, WebP)
 - `user_id`: User ID string (optional)
-- `prompt_text`: Prompt text for video generation (must contain 'kiss' or 'hug' to match category)
+- `prompt_text`: Prompt text for video generation (must contain keywords to match category)
 - `firebase_id_token`: Firebase ID token
 - `appname`: App name for database routing (optional)
 
-**Category Matching:**
-The prompt_text is automatically matched to categories:
-- Contains "kiss" → Pixverse Kiss Video
-- Contains "hug" → Pixverse Hug Video
+**Category Matching (4 video types):**
+The prompt_text is automatically matched to one of 4 categories:
+- Contains "kiss" → Ai Kiss Video
+- Contains "hug" → Ai Hug Video
+- Contains "bf" or "boyfriend" → Ai BF
+- Contains "gf" or "girlfriend" → Ai GF
 
 **Success Response:**
 ```json

main.py

@@ -131,7 +131,7 @@ async def get_firebase_user(
 async def generate_video(
     image: UploadFile = File(..., description="User image file (JPG, PNG, WebP)"),
     user_id: Optional[str] = Form(None, description="User ID (ObjectId string or integer)"),
-    prompt_text: str = Form(..., description="Prompt text for video generation (must contain 'kiss' or 'hug' to match category)"),
+    prompt_text: str = Form(..., description="Prompt text for video generation (must contain keywords: kiss, hug, bf/boyfriend, gf/girlfriend)"),
     firebase_id_token: str = Form(..., description="Firebase ID token"),
     appname: Optional[str] = Form(None, description="App name for database routing (collage-maker, AI-Enhancer, etc.)"),
     authorization: Optional[str] = Header(None, description="HF Token as Bearer token")
@@ -142,14 +142,16 @@ async def generate_video(
     - Uploads user image to Digital Ocean Spaces (milestone/valentine/source/)
     - Generates video using PixVerse API
     - Stores result video in Digital Ocean Spaces (milestone/valentine/results/)
-    - Automatically matches prompt_text to category (Pixverse Kiss Video / Pixverse Hug Video)
+    - Automatically matches prompt_text to one of 4 categories
     - Tracks media clicks in MongoDB based on appname
     - Logs request to MongoDB Logs database
     - Returns URLs for both source image and generated video
     
     Category Matching (based on prompt_text keywords):
-    - "kiss" → Pixverse Kiss Video
-    - "hug" → Pixverse Hug Video
+    - "kiss" → Ai Kiss Video
+    - "hug" → Ai Hug Video
+    - "bf" or "boyfriend" → Ai BF
+    - "gf" or "girlfriend" → Ai GF
     
     Database Routing (based on appname):
     - "collage-maker" → collage-maker MongoDB → adminPanel DB → media_clicks collection

services/mongodb_service.py

@@ -476,9 +476,14 @@ def get_category_name(category_id: str, appname: Optional[str] = None) -> Option
 
 
 # Pixverse category mapping - keyword to category name
+# 4 types of videos: AI BF, AI GF, AI Hug Video, AI Kiss Video
 PIXVERSE_CATEGORIES = {
-    "kiss": "Pixverse Kiss Video",
-    "hug": "Pixverse Hug Video"
+    "kiss": "Ai Kiss Video",
+    "hug": "Ai Hug Video",
+    "boyfriend": "Ai BF",
+    "bf": "Ai BF",
+    "girlfriend": "Ai GF",
+    "gf": "Ai GF"
 }