add: prodctpage batch cto, mto, nto, video and static file

.gitignore

@@ -0,0 +1,5 @@
+.idea
+__pycache__
+*.pyc
+*.log
+logs

Dockerfile

@@ -0,0 +1,15 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+EXPOSE 7860
+
+VOLUME ["/app/logs"]
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,38 @@
+from pathlib import Path
+
+from fastapi import FastAPI
+from starlette.middleware.cors import CORSMiddleware
+from starlette.staticfiles import StaticFiles
+
+from src.api.api import router
+
+app = FastAPI()
+
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+DIR = Path(__file__).parent
+static_folder = DIR / "static"
+app.mount("/static", StaticFiles(directory=static_folder, html=True), name="static")
+
+app.include_router(router)
+
+
+@app.get("/{full_path:path}")
+async def serve_html(full_path: str):
+    file_path = static_folder / full_path
+    if file_path.exists() and file_path.is_file():
+        return StaticFiles(directory=static_folder).lookup_path(file_path)
+    return StaticFiles(directory=static_folder).lookup_path(static_folder / "index.html")
+
+
+if __name__ == "__main__":
+    import uvicorn
+
+    uvicorn.run(app, host="0.0.0.0", port=7860)

requirements.txt

@@ -0,0 +1,48 @@
+aiohappyeyeballs==2.4.4
+aiohttp==3.11.11
+aiosignal==1.3.2
+annotated-types==0.7.0
+anyio==4.8.0
+async-timeout==5.0.1
+attrs==24.3.0
+blinker==1.9.0
+certifi==2024.12.14
+charset-normalizer==3.4.1
+click==8.1.8
+deprecation==2.1.0
+exceptiongroup==1.2.2
+fastapi==0.115.6
+frozenlist==1.5.0
+gotrue==2.11.1
+h11==0.14.0
+h2==4.1.0
+hpack==4.0.0
+httpcore==1.0.7
+httpx==0.27.2
+hyperframe==6.0.1
+idna==3.10
+itsdangerous==2.2.0
+Jinja2==3.1.5
+MarkupSafe==3.0.2
+multidict==6.1.0
+packaging==24.2
+postgrest==0.19.1
+propcache==0.2.1
+pydantic==2.10.4
+pydantic_core==2.27.2
+python-dateutil==2.9.0.post0
+realtime==2.1.0
+requests==2.32.3
+six==1.17.0
+sniffio==1.3.1
+starlette==0.41.3
+storage3==0.11.0
+StrEnum==0.4.15
+supabase==2.11.0
+supafunc==0.9.0
+typing_extensions==4.12.2
+urllib3==2.3.0
+uvicorn==0.34.0
+websockets==13.1
+Werkzeug==3.1.3
+yarl==1.18.3

src/__init__.py

@@ -0,0 +1,5 @@
+"""
+project @ batch_generation_support_team
+created @ 2025-01-10
+author  @ github.com/ishworrsubedii
+"""

src/api/__init__.py

@@ -0,0 +1,5 @@
+"""
+project @ batch_generation_support_team
+created @ 2025-01-10
+author  @ github.com/ishworrsubedii
+"""

src/api/api.py

@@ -0,0 +1,76 @@
+"""
+project @ batch_generation_support_team
+created @ 2025-01-09
+author  @ github.com/ishworrsubedii
+"""
+import json
+
+from fastapi import APIRouter
+
+from src.components.supabase_information_fetch import supabase_image_fetch_product_page, supabase_product_page_approval, \
+    supabase_fetch_not_approved_necklaces
+from src.pipeline.main import combined_image_and_video_generation
+
+router = APIRouter()
+
+
+@router.get('/ping')
+def ping():
+    return "Running"
+
+
+@router.post('/generate')
+def generate(storename, image_url):
+    response = combined_image_and_video_generation(storename=storename, image_url=image_url)
+    return response
+
+
+from fastapi.responses import JSONResponse
+
+
+@router.post('/image_fetch_product_page')
+def image_fetch_product_page(necklace_id, model_name):
+    response = supabase_image_fetch_product_page(necklace_id, model_name)
+    return response
+
+
+@router.post('/approve')
+def image_fetch_product_page(necklace_id, model_name):
+    response = supabase_image_fetch_product_page(necklace_id, model_name)
+
+    response_nto = json.loads(response["nto"]) if isinstance(response["nto"], str) else response["nto"]
+    response_cto = json.loads(response["cto"]) if isinstance(response["cto"], str) else response["cto"]
+    response_mto = json.loads(response["mto"]) if isinstance(response["mto"], str) else response["mto"]
+
+    comma_seperated_nto = ",".join(response_nto)
+    comma_seperated_cto = ",".join(response_cto)
+    comma_seperated_mto = ",".join(response_mto)
+    responsee = supabase_product_page_approval(necklace_id, comma_seperated_nto, comma_seperated_cto,
+                                               comma_seperated_mto,
+                                               response["video"], model_name=model_name)
+
+    return JSONResponse(content=responsee)
+
+
+@router.get('/list_necklace_id')
+def list_not_approved_necklaces():
+    response = supabase_fetch_not_approved_necklaces()
+
+    necklace_models = {}
+
+    for item in response:
+        necklace_id = item.get('Id')
+        approve_dict = item.get('approve', {})
+
+        if necklace_id and approve_dict:
+            # Get list of models that have False approval status
+            unapproved_models = [
+                model_name
+                for model_name, status in approve_dict.items()
+                if status is False
+            ]
+
+            if unapproved_models:  # Only include if there are unapproved models
+                necklace_models[necklace_id] = unapproved_models
+
+    return JSONResponse(content=necklace_models)

src/components/__init__.py

@@ -0,0 +1,5 @@
+"""
+project @ batch_generation_support_team
+created @ 2025-01-10
+author  @ github.com/ishworrsubedii
+"""

src/components/image_generation.py

@@ -0,0 +1,105 @@
+"""
+project @ batch_generation_support_team
+created @ 2025-01-07
+author  @ github.com/ishworrsubedii
+"""
+from dataclasses import dataclass
+from typing import List, Dict
+from urllib.parse import quote
+
+import requests
+
+
+@dataclass
+class MakeupColors:
+    lipstick: str
+    eyeliner: str
+    eyeshadow: str
+
+
+@dataclass
+class ImageGenerationResult:
+    nto_results: List[Dict[str, str]]
+    cto_results: List[Dict[str, str]]
+    mto_results: List[Dict[str, Dict]]
+    status: str
+    message: str
+
+
+def batch_image_generation(
+        model_image: str,
+        necklace_id: str,
+        necklace_category: str,
+        storename: str,
+        clothing_list: List[str],
+        makeup_colors: Dict[str, str],
+        x_offset: float = 0.2,
+        y_offset: float = 0.35
+) -> ImageGenerationResult:
+    base_url = "https://techconspartners-video-gen.hf.space/product_page_image_generation"
+
+    encoded_model_image = quote(model_image)
+
+    url = f"{base_url}?model_image={encoded_model_image}&necklace_id={necklace_id}&necklace_category={quote(necklace_category)}&storename={quote(storename)}&x_offset={x_offset}&y_offset={y_offset}"
+
+    payload = {
+        "clothing_list": clothing_list,
+        "makeup_colors": makeup_colors
+    }
+
+    # Set up headers
+    headers = {
+        "accept": "application/json",
+        "Content-Type": "application/json"
+    }
+
+    try:
+        response = requests.post(url, headers=headers, json=payload)
+        response.raise_for_status()  # Raise an exception for bad status codes
+
+        data = response.json()
+
+        if data.get("status") != "success":
+            raise ValueError(f"API returned error: {data.get('message', 'Unknown error')}")
+
+        result = ImageGenerationResult(
+            status=data["status"],
+            message=data["message"],
+            nto_results=data["nto_results"],
+            cto_results=data["cto_results"],
+            mto_results=data["mto_results"]
+        )
+
+        return result
+
+    except requests.exceptions.RequestException as e:
+        raise RuntimeError(f"Failed to make API request: {str(e)}")
+    except (KeyError, ValueError) as e:
+        raise ValueError(f"Error processing API response: {str(e)}")
+
+
+if __name__ == "__main__":
+    test_params = {
+        "model_image": "https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/JewelmirrorModelImages/M0X75f993d5c22ab292b647e3c9J.png",
+        "necklace_id": "CJM002",
+        "necklace_category": "Gold Necklaces",
+        "storename": "ChamundiJewelsMandir",
+        "clothing_list": ["Red Silk saree", "Green south indian saree", "Purple kurti"],
+        "makeup_colors": {
+            "lipstick": "Carmine Red",
+            "eyeliner": "Black",
+            "eyeshadow": "Maroon"
+        }
+    }
+
+    try:
+        result = batch_image_generation(**test_params)
+        with open("image_generation_response.json", "w") as f:
+            f.write(str(result))
+        print(f"Status: {result.status}")
+        print(f"Message: {result.message}")
+        print(f"Number of NTO results: {len(result.nto_results)}")
+        print(f"Number of CTO results: {len(result.cto_results)}")
+        print(f"Number of MTO results: {len(result.mto_results)}")
+    except Exception as e:
+        print(f"Error: {str(e)}")

src/components/supabase_information_fetch.py

@@ -0,0 +1,409 @@
+import logging
+import os
+from dataclasses import dataclass
+from typing import List
+
+from supabase import create_client
+
+
+@dataclass
+class NecklaceData:
+    necklace_id: str
+    necklace_url: str
+    x_lean_offset: float
+    y_lean_offset: float
+    x_broad_offset: float
+    y_broad_offset: float
+    category: str
+
+
+supabase_url = os.getenv("SUPABASE_URL")
+supabase_key = os.getenv("SUPABASE_KEY")
+
+supabase = create_client(supabase_url, supabase_key)
+
+
+def fetch_necklace_offset_each_store(storename: str) -> List[NecklaceData]:
+    response = supabase.table("MagicMirror").select("*").eq("StoreName", storename).execute()
+
+    necklace_data_list = []
+
+    for item in response.data:
+        jewellery_url = (
+            f"https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/Stores/"
+            f"{storename}/{item['Category']}/image/{item['Id']}.png"
+        )
+
+        necklace_data = NecklaceData(
+            necklace_id=item['Id'],
+            necklace_url=jewellery_url,
+            x_lean_offset=item['x_lean'],
+            y_lean_offset=item['y_lean'],
+            x_broad_offset=item['x_chubby'],
+            y_broad_offset=item['y_chubby'],
+            category=item['Category']
+        )
+
+        necklace_data_list.append(necklace_data)
+
+    return necklace_data_list
+
+
+def fetch_model_body_type(image_url: str):
+    try:
+        response = supabase.table("JewelMirror_ModelImages").select("*").eq("image_url", image_url).execute()
+
+        for item in response.data:
+            print(item)
+            return item['image_url'], item['body_structure']
+
+
+
+
+    except Exception as e:
+        print(f"The image url is not found in the table", {e})
+        return None
+
+
+def upload_information_to_new_table(necklace_id, nto_images_urls, cto_images_urls, mto_urls, video_urls, model_name):
+    try:
+        response_check = supabase.table("JM_Productpage").select("*").eq("Id", necklace_id).execute()
+
+        if response_check.data:
+            existing_record = response_check.data[0]
+
+            updated_nto = f"{existing_record['nto_images_urls']},{nto_images_urls}" if existing_record[
+                'nto_images_urls'] else nto_images_urls
+            updated_cto = f"{existing_record['cto_images_urls']},{cto_images_urls}" if existing_record[
+                'cto_images_urls'] else cto_images_urls
+            updated_mto = f"{existing_record['mto_images_urls']},{mto_urls}" if existing_record[
+                'mto_images_urls'] else mto_urls
+            updated_video = f"{existing_record['video_urls']},{video_urls}" if existing_record[
+                'video_urls'] else video_urls
+            updated_model = f"{existing_record['model_name']},{model_name}" if existing_record[
+                'model_name'] else model_name
+
+            current_approve = existing_record.get('approve', {})
+            if not isinstance(current_approve, dict):
+                current_approve = {}
+
+            if model_name not in current_approve:
+                current_approve[model_name] = False
+
+            logging.info(f"Updated model name: {updated_model}")
+            logging.info(f"Updated approve status: {current_approve}")
+
+            response = supabase.table("JM_Productpage").update({
+                "nto_images_urls": updated_nto,
+                "cto_images_urls": updated_cto,
+                "mto_images_urls": updated_mto,
+                "video_urls": updated_video,
+                "model_name": updated_model,
+                "approve": current_approve
+            }).eq("Id", necklace_id).execute()
+
+            print("Updated existing record")
+
+        else:
+            response = supabase.table("JM_Productpage").insert({
+                "Id": necklace_id,
+                "nto_images_urls": nto_images_urls,
+                "cto_images_urls": cto_images_urls,
+                "mto_images_urls": mto_urls,
+                "video_urls": video_urls,
+                "model_name": model_name,
+                "approve": {
+                    model_name: False
+                }
+            }).execute()
+
+            print("Inserted new record")
+
+        return response
+
+    except Exception as e:
+        print(f"Error in uploading the data to the table: {str(e)}")
+        logging.error(f"Error in uploading the data to the table: {str(e)}")
+        return None
+
+
+def supabase_image_fetch_product_page(necklace_id, model_name):
+    try:
+        response = supabase.table("JM_Productpage").select("*").eq("Id", necklace_id).execute()
+
+        if not response.data:
+            print(f"No data found for necklace_id: {necklace_id}")
+            return None
+
+        # Get the first item from response
+        item = response.data[0]
+
+        def process_url_list(url_string):
+            if not url_string:
+                return []
+
+            url_groups = url_string.replace("[", "").replace("]", "").split("],[")
+
+            filtered_urls = []
+            for group in url_groups:
+                urls = [url.strip().strip('"\'') for url in group.split(",")]
+                matching_urls = [url for url in urls if model_name in url]
+                filtered_urls.extend(matching_urls)
+
+            return filtered_urls
+
+        try:
+            nto_images = process_url_list(item.get('nto_images_urls', ''))
+            cto_images = process_url_list(item.get('cto_images_urls', ''))
+            mto_images = process_url_list(item.get('mto_images_urls', ''))
+            print(f"nto_images: {nto_images}")
+            print(f"cto_images: {cto_images}")
+            print(f"mto_images: {mto_images}")
+
+            video_urls = []
+            if item.get('video_urls'):
+                video_list = item['video_urls'].split(',')
+                video_urls = [url.strip() for url in video_list if model_name in url]
+
+            print(f"video_urls: {video_urls}")
+
+        except AttributeError as e:
+            print(f"Error parsing URLs: {e}")
+            return None
+
+        response = {
+            'nto': nto_images,
+            'cto': cto_images,
+            'mto': mto_images,
+            'video': video_urls,
+            "status": "success"
+
+        }
+
+        return response
+    except Exception as e:
+        print(f"Error in fetching the data from the table: {e}")
+        return None
+
+
+def supabase_product_page_approval(necklace_id, nto_images_urls, cto_images_urls, mto_images_urls, video_urls,
+                                   model_name):
+    try:
+        existing_record = supabase.table("MagicMirror").select("*").eq("Id", necklace_id).execute()
+
+        def append_urls(existing_urls, new_urls):
+            existing = convert_to_string(existing_urls) if existing_urls else ""
+            new = convert_to_string(new_urls) if new_urls else ""
+
+            if existing and new:
+                return f"{existing},{new}"
+            return new or existing
+
+        if existing_record.data:
+            record = existing_record.data[0]
+
+            update_data = {
+                "nto_images_urls": append_urls(record.get("nto_images_urls"), nto_images_urls),
+                "cto_images_urls": append_urls(record.get("cto_images_urls"), cto_images_urls),
+                "mto_images_urls": append_urls(record.get("mto_images_urls"), mto_images_urls),
+                "video_urls": append_urls(record.get("video_urls"), video_urls)
+            }
+        else:
+            update_data = {
+                "nto_images_urls": convert_to_string(nto_images_urls),
+                "cto_images_urls": convert_to_string(cto_images_urls),
+                "mto_images_urls": convert_to_string(mto_images_urls),
+                "video_urls": convert_to_string(video_urls)
+            }
+
+        result = supabase.table("MagicMirror") \
+            .update(update_data) \
+            .eq("Id", necklace_id) \
+            .execute()
+
+        res = supabase_jmproductpage_approval_flag(necklace_id, model_name, True)
+
+        print(f"Successfully updated URLs for necklace ID: {necklace_id}")
+        return result.data
+
+    except Exception as e:
+        error_msg = f"Error updating URLs in MagicMirror table: {str(e)}"
+        print(error_msg)
+        raise Exception(error_msg)
+
+
+def convert_to_string(value):
+    if value is None:
+        return ""
+
+    if isinstance(value, list):
+        return ",".join(str(item) for item in value if item)
+
+    if isinstance(value, str):
+        # If it's already a string, clean it up
+        items = [item.strip() for item in value.split(",") if item.strip()]
+        return ",".join(items)
+
+    return str(value)
+
+
+def supabase_fetch_not_approved_necklaces():
+    try:
+        response = supabase.table("JM_Productpage").select("*").execute()
+
+        unapproved_necklaces = []
+        for record in response.data:
+            approve_dict = record.get('approve', {})
+            if not isinstance(approve_dict, dict):
+                continue
+
+            false_approvals = {
+                model: status
+                for model, status in approve_dict.items()
+                if status is False
+            }
+
+            if false_approvals:
+                filtered_record = {
+                    'Id': record['Id'],
+                    'approve': false_approvals
+                }
+                unapproved_necklaces.append(filtered_record)
+        print(f"Unapproved necklaces: {unapproved_necklaces}")
+        return unapproved_necklaces
+
+    except Exception as e:
+        print(f"Error fetching not approved necklaces: {str(e)}")
+        raise e
+
+
+def supabase_jmproductpage_approval_flag(necklace_id, model_name, flag_value):
+    try:
+        current_data = supabase.table("JM_Productpage").select("approve").eq("Id", necklace_id).execute()
+
+        current_approve = current_data.data[0].get("approve", {}) if current_data.data else {}
+
+        if model_name in current_approve and current_approve[model_name] is True:
+            print(f"Skipping update for {model_name} as it's already approved")
+            return current_data.data
+
+        if isinstance(current_approve, dict):
+            current_approve[model_name] = flag_value
+        else:
+            current_approve = {model_name: flag_value}
+
+        response = supabase.table("JM_Productpage").update(
+            {"approve": current_approve}
+        ).eq("Id", necklace_id).execute()
+
+        return response.data
+
+    except Exception as e:
+        print(f"Error updating approval flag for necklace ID {necklace_id}: {str(e)}")
+        raise e
+
+
+def upload_productpage_logs(necklace_id: str, status: bool, model_name: str) -> dict:
+    try:
+        existing_record = supabase.table("JM_productpagelog").select("*").eq("id", necklace_id).execute()
+
+        if not existing_record.data:
+            response = supabase.table("JM_productpagelog").insert([{
+                "id": necklace_id,
+                "status": bool(status),
+                "model_name": model_name
+            }]).execute()
+            print("Inserted new record")
+            return {
+                "status": "success",
+                "message": "Record inserted successfully"
+            }
+        else:
+            existing_models = existing_record.data[0].get('model_name', '').split(',')
+            existing_models = [model.strip() for model in existing_models]
+
+            if model_name in existing_models:
+                return {
+                    "status": "error",
+                    "message": f"Record already exists with model {model_name} and {necklace_id}"
+                }
+            else:
+                updated_models = ','.join(existing_models + [model_name])
+
+                response = supabase.table("JM_productpagelog").update({
+                    "model_name": updated_models
+                }).eq("id", necklace_id).execute()
+
+                print(f"Updated record with new model {model_name}")
+                return {
+                    "status": "success",
+                    "message": f"Added new model {model_name} to existing record"
+                }
+
+    except Exception as e:
+        error_msg = f"Error updating product page log for necklace {necklace_id}: {str(e)}"
+        logging.error(error_msg)
+        raise Exception(f"Failed to update product page log: {str(e)}")
+
+
+def fetch_productpage_logs(necklace_id):
+    try:
+        response = supabase.table("JM_productpagelog").select("*").eq("id", necklace_id).execute()
+        for item in response.data:
+            return item['status']
+    except Exception as e:
+        print(f"Error in fetching the data from the table", {e})
+        return None
+
+
+if __name__ == "__main__":
+    # store_name = "ChamundiJewelsMandir"
+    # necklaces = fetch_necklace_offset_each_store(store_name)
+    #
+    # count = 0
+    # for necklace in necklaces:
+    #     print(f"Necklace ID: {necklace.necklace_id}")
+    #     print(f"URL: {necklace.necklace_url}")
+    #     print(f"Lean offsets (x, y): ({necklace.x_lean_offset}, {necklace.y_lean_offset})")
+    #     print(f"Broad offsets (x, y): ({necklace.x_broad_offset}, {necklace.y_broad_offset})")
+    #     print(f"Category: {necklace.category}")
+    #     print("-" * 50)
+    #     count += 1
+    #
+    # print(f"Total necklaces fetched: {count}")
+    # image_url = "https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/JewelmirrorModelImages/M0X9e22b54f110493113feb79e3J.png"
+    # url,type=fetch_model_body_type(image_url)
+    # print(url,type)
+    # ---------------------------------
+    #
+    # upload_information_to_new_table(necklace_id="CJM0025",
+    #                                 nto_images_urls="https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/ProductPageOutputs/CJM0025-M0X0f16ad748dd979a48d3f79b4J-nto-Gold_Necklaces.png",
+    #                                 cto_images_urls="https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/ProductPageOutputs/CJM0025-cto-Blue Lehenga.png,https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/ProductPageOutputs/CJM0025-cto-Blue Kurti.png",
+    #                                 mto_urls="https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/ProductPageOutputs/CJM0025-M0X0f16ad748dd979a48d3f79b4J-mto-Blue_Kurti-lip_Carmine_Red_eye_Black_shadow_Maroon.png",
+    #                                 video_urls="https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/JewelmirrorVideoGeneration/video_bd48d59bc8f358c3.mp4")
+    # --------------------------------
+    # nto, cto, mto, video = supabase_image_fetch_product_page("CJM0025")
+    #
+    # print("nto", nto)
+    # print("cto", cto)
+    # print("mto", mto)
+    # print("video", video)
+    # # --------------------------------
+    # res = supabase_product_page_approval(necklace_id="CJM0025",
+    #                                      nto_images_urls="https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/ProductPageOutputs/CJM0025-M0X0f16ad748dd979a48d3f79b4J-nto-Gold_Necklaces.png",
+    #                                      cto_images_urls="https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/ProductPageOutputs/CJM0025-cto-Blue Lehenga.png,https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/ProductPageOutputs/CJM0025-cto-Blue Kurti.png",
+    #                                      mto_images_urls="https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/ProductPageOutputs/CJM0025-M0X0f16ad748dd979a48d3f79b4J-mto-Blue_Kurti-lip_Carmine_Red_eye_Black_shadow_Maroon.png",
+    #                                      video_urls="https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/JewelmirrorVideoGeneration/video_bd48d59bc8f358c3.mp4")
+    # print(res)
+
+    # res = supabase_fetch_not_approved_necklaces()
+    # print(res)
+    # --------------------------------
+    # res = supabase_jmproductpage_approval_flag("CJM001")
+    # print(res)
+
+    response = upload_productpage_logs("CJM0025", "True")
+
+    # -------------------------------
+    response = fetch_productpage_logs("CJM0025")
+    print(response)

src/components/video_generation.py

@@ -0,0 +1,153 @@
+"""
+project @ batch_generation_support_team
+created @ 2025-01-07
+author  @ github.com/ishworrsubedii
+"""
+from dataclasses import dataclass
+from typing import List
+
+import requests
+
+
+@dataclass
+class VideoTimings:
+    file_download_time: float
+    video_creation_time: float
+    supabase_upload_time: float
+
+
+@dataclass
+class VideoGenerationResult:
+    status: str
+    message: str
+    video_url: str
+    timings: VideoTimings
+
+
+def generate_combined_video(
+        intro_video_path: str,
+        font_path: str,
+        background_audio_path: str,
+        necklace_title: List[str],
+        necklace_images: List[str],
+        nto_image_title: List[List[str]],
+        nto_cto_image_title: List[List[str]],
+        makeup_image_title: List[List[str]],
+        necklace_try_on_output_images: List[List[str]],
+        clothing_output_images: List[List[str]],
+        makeup_output_images: List[List[str]],
+        background_colors: List[List[int]],
+        outro_title: str,
+        address: str,
+        phone_numbers: str,
+        logo_url: str,
+        image_display_duration: float = 2.5,
+        fps: int = 30,
+        transition_duration: float = 0.5,
+        transition_type: str = "None",
+        direction: str = "left",
+        outro_video_path: str = "default_outro.mp4"
+) -> VideoGenerationResult:
+    url = "https://techconspartners-video-gen.hf.space/createcombinedvideo/"
+
+    payload = {
+        "intro_video_path": intro_video_path,
+        "font_path": font_path,
+        "background_audio_path": background_audio_path,
+        "image_display_duration": image_display_duration,
+        "fps": fps,
+        "necklace_title": necklace_title,
+        "nto_image_title": nto_image_title,
+        "nto_cto_image_title": nto_cto_image_title,
+        "makeup_image_title": makeup_image_title,
+        "necklace_images": necklace_images,
+        "necklace_try_on_output_images": necklace_try_on_output_images,
+        "clothing_output_images": clothing_output_images,
+        "makeup_output_images": makeup_output_images,
+        "background_colors": background_colors,
+        "outro_title": outro_title,
+        "address": address,
+        "phone_numbers": phone_numbers,
+        "logo_url": logo_url,
+        "transition_duration": transition_duration,
+        "transition_type": transition_type,
+        "direction": direction,
+        "outro_video_path": outro_video_path
+    }
+    headers = {
+        "accept": "application/json",
+        "Content-Type": "application/json"
+    }
+
+    try:
+
+        response = requests.post(url, headers=headers, json=payload)
+        # response.raise_for_status()
+
+        data = response.json()
+
+        if data.get("status") != "success":
+            raise ValueError(f"API returned error: {data.get('message', 'Unknown error')}")
+
+        timings = VideoTimings(
+            file_download_time=data["timings"]["file_download_time"],
+            video_creation_time=data["timings"]["video_creation_time"],
+            supabase_upload_time=data["timings"]["supabase_upload_time"]
+        )
+
+        result = VideoGenerationResult(
+            status=data["status"],
+            message=data["message"],
+            video_url=data["video_url"],
+            timings=timings
+        )
+
+        return result
+
+    except requests.exceptions.RequestException as e:
+        raise RuntimeError(f"Failed to make API request: {str(e)}")
+    except (KeyError, ValueError) as e:
+        raise ValueError(f"Error processing API response: {str(e)}")
+
+
+if __name__ == "__main__":
+    # test_params = {
+    #     "intro_video_path": "Vellaimani_intro.mp4",
+    #     "font_path": "PlayfairDisplay-VariableFont.ttf",
+    #     "background_audio_path": "LoveIndianCinematicBGM.mp3",
+    #     "necklace_title": ["ARA01DI001"],
+    #     "necklace_images": [
+    #         "https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/Stores/VellaimaniJewellery/Diamond%20Necklaces/image/ARA01DI001.png"
+    #     ],
+    #     "nto_image_title": [["ARA01DI001"]],
+    #     "nto_cto_image_title": [["ARA01DI001", "ARA01DI001"]],
+    #     "makeup_image_title": [["ARA01DI001"]],
+    #     "necklace_try_on_output_images": [[
+    #         "https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/ProductPageOutputs/ARA01DI001-M0X75f993d5c22ab292b647e3c9J-nto-Diamond_Necklaces.png"
+    #     ]],
+    #     "clothing_output_images": [[
+    #         "https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/ProductPageOutputs/ARA01DI001-cto-Red Silk saree.png",
+    #         "https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/ProductPageOutputs/ARA01DI001-cto-Green south indian saree.png"
+    #     ]],
+    #     "makeup_output_images": [[
+    #         "https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/ProductPageOutputs/ARA01DI001-M0X75f993d5c22ab292b647e3c9J-mto-Purple_kurti-lip_Carmine_Red_eye_Black_shadow_Maroon.png"
+    #     ]],
+    #     "background_colors": [[245, 245, 245], [220, 245, 245]],
+    #     "outro_title": "Reach out to us for more information",
+    #     "address": "123, ABC Street, XYZ City",
+    #     "phone_numbers": "1234567890",
+    #     "logo_url": "https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/MagicMirror/FullImages/default.png"
+    # }
+
+    with open("/home/ishwor/Desktop/TCP/batch_generation_support_team/combined_video_params.json", "r") as f:
+        test_params = str(f.read())
+
+    try:
+        result = generate_combined_video(**test_params)
+        print(f"Status: {result.status}")
+        print(f"Message: {result.message}")
+        print(f"Video URL: {result.video_url}")
+        print(
+            f"Total processing time: {result.timings.file_download_time + result.timings.video_creation_time + result.timings.supabase_upload_time:.2f} seconds")
+    except Exception as e:
+        print(f"Error: {str(e)}")

src/pipeline/__init__.py

@@ -0,0 +1,5 @@
+"""
+project @ batch_generation_support_team
+created @ 2025-01-10
+author  @ github.com/ishworrsubedii
+"""

src/pipeline/main.py

@@ -0,0 +1,193 @@
+import logging
+import os
+import random
+from datetime import datetime
+from typing import List
+
+from src.components.image_generation import batch_image_generation
+from src.components.supabase_information_fetch import fetch_necklace_offset_each_store, fetch_model_body_type, \
+    upload_information_to_new_table, upload_productpage_logs
+from src.components.video_generation import generate_combined_video
+
+os.makedirs('logs', exist_ok=True)
+
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s',
+    handlers=[
+        logging.FileHandler('logs/running.log'),
+        logging.StreamHandler()
+    ]
+)
+
+
+def update_processed_necklaces(necklace_id: str, status: bool, model_name: str):
+    response = upload_productpage_logs(necklace_id, status, model_name=model_name)
+
+
+def get_random_clothing_combinations(clothing_list: List[str], colors: List[str], count: int = 5) -> List[str]:
+    """Generate random clothing combinations."""
+    combinations = []
+    for _ in range(count):
+        clothing = random.choice(clothing_list)
+        color = random.choice(colors)
+        combinations.append(f"{color} {clothing}")
+    return combinations
+
+
+def combined_image_and_video_generation(storename, image_url):
+    logging.info("Starting combined image and video generation process")
+
+    try:
+        necklace_data = fetch_necklace_offset_each_store(storename=storename)
+        url, typee = fetch_model_body_type(image_url=image_url)
+
+        model_name = url.split("/")[-1].split(".")[0]
+
+        for necklace in necklace_data:
+            necklace_id = necklace.necklace_id
+            response = upload_productpage_logs(necklace_id, True, model_name=model_name)
+            if response['status'] == "error":
+                print("Skipping", necklace_id)
+                logging.info(f"Skipping {necklace_id} - already processed")
+                continue
+
+            start_time = datetime.now()
+            logging.info(f"Processing necklace: {necklace_id}")
+
+            try:
+                if typee == "lean":
+                    x_offset = necklace.x_lean_offset
+                    y_offset = necklace.y_lean_offset
+                    logging.info("Body Type: lean")
+
+                elif typee == "medium":
+                    x_offset = necklace.x_broad_offset
+                    y_offset = necklace.y_broad_offset
+                    logging.info("Body Type: medium")
+
+                else:
+                    logging.info("Body Type: None")
+                    x_offset = None
+                    y_offset = None
+
+                clothing_combinations = get_random_clothing_combinations(
+                    clothing_list=["Salwar Kameez", "South Indian Saree", "Kurti", "Lehenga", "Silk Saree"],
+                    colors=["Red", "Blue", "Green", "Yellow", "Pink"]
+                )
+
+                makeup_data = {
+                    "lipstick": "Carmine Red",
+                    "eyeliner": "Black",
+                    "eyeshadow": "Maroon"
+                }
+
+                image_params = {
+                    "model_image": url,
+                    "necklace_id": necklace_id,
+                    "necklace_category": necklace.category,
+                    "storename": storename,
+                    "clothing_list": clothing_combinations,
+                    "makeup_colors": makeup_data,
+                    "x_offset": x_offset,
+                    "y_offset": y_offset
+
+                }
+                print("image: params", image_params)
+                logging.info("NTO-CTO-MTO images Generating for {}".format(necklace_id))
+
+                image_results = batch_image_generation(**image_params)
+                logging.info(f"Image generation result: {image_results}")
+
+                if image_results.status != 'success':
+                    raise Exception(f"Image generation failed: {image_results.message}")
+                cto_urls = [result['url'] for result in image_results.cto_results[:4]]  # First four CTO images
+                mto_urls = [image_results.mto_results[-1]['url']]
+                video_params = {
+                    "intro_video_path": f"{storename}_intro.mp4",
+                    "font_path": "PlayfairDisplay-VariableFont.ttf",
+                    "background_audio_path": "LoveIndianCinematicBGM.mp3",
+                    "necklace_title": [necklace_id],
+                    "necklace_images": [necklace.necklace_url],
+                    "nto_image_title": [[necklace_id]],
+                    "nto_cto_image_title": [[necklace_id, necklace_id, necklace_id, necklace_id]],
+                    "makeup_image_title": [[necklace_id]],
+                    "necklace_try_on_output_images": [[result['url'] for result in image_results.nto_results]],
+                    "clothing_output_images": [cto_urls
+                                               ],
+                    "makeup_output_images": [
+                        mto_urls
+                    ],
+
+                    "background_colors": [[245, 245, 245], [220, 245, 245]],
+                    "outro_title": "Reach out to us for more information",
+                    "address": "None",
+                    "phone_numbers": "None",
+                    "logo_url": "https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/MagicMirror/FullImages/default.png",
+                    "outro_video_path": f"{storename}_outro.mp4"
+                }
+
+                logging.info("Video Generating for {}".format(necklace_id))
+
+                video_result = generate_combined_video(**video_params)
+                logging.info(f"Video generation result: {video_result}")
+
+                if video_result.status != 'success':
+                    raise Exception(f"Video generation failed: {video_result.message}")
+
+                urls = {
+                    'video_url': video_result.video_url,
+                    'images_url': str([result['url'] for result in image_results.nto_results])
+                }
+
+                logging.info("Completed combined image and video generation process")
+                upload_information_to_new_table(necklace_id=necklace_id,
+                                                nto_images_urls=[result['url'] for result in image_results.nto_results],
+                                                cto_images_urls=cto_urls,
+                                                mto_urls=mto_urls,
+                                                video_urls=video_result.video_url,
+                                                model_name=model_name
+                                                )
+
+                update_processed_necklaces(
+                    necklace_id=necklace_id,
+                    status=True,
+                    model_name=model_name
+
+                )
+
+
+
+
+
+            except Exception as e:
+
+                raise e
+
+                logging.error(f"Error processing {necklace_id}: {str(e)}")
+
+                update_processed_necklaces(
+                    necklace_id=necklace_id,
+                    status=False,
+                    model_name=model_name
+
+                )
+        return {
+            "status": "success",
+        }
+
+
+    except Exception as e:
+        raise e
+        logging.error(f"Fatal error in combined generation process: {str(e)}")
+        return {
+            'status': 'error',
+            'message': str(e)
+        }
+
+
+if __name__ == "__main__":
+    image_url = "https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/JewelmirrorModelImages/p_01.png"
+    storename = "ChamundiJewelsMandir"
+    result = combined_image_and_video_generation(image_url=image_url, storename=storename)
+    print(f"Process completed with status: {result}")

static/index.html

@@ -0,0 +1,137 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Generation</title>
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+            margin: 0;
+            padding: 0;
+            display: flex;
+            flex-direction: column;
+            justify-content: center;
+            align-items: center;
+            background-color: #f0f4f8;
+        }
+
+        /* Navbar */
+        .navbar {
+            background-color: #007bff;
+            color: white;
+            width: 100%;
+            padding: 10px 0;
+            text-align: center;
+            font-size: 18px;
+            font-weight: bold;
+            position: fixed;
+            top: 0;
+            left: 0;
+            z-index: 1000;
+        }
+
+        .navbar a {
+            color: white;
+            text-decoration: none;
+            margin: 0 15px;
+            font-size: 16px;
+        }
+
+        .navbar a:hover {
+            text-decoration: underline;
+        }
+
+        /* Form Container */
+        .form-container {
+            background: white;
+            padding: 20px;
+            border-radius: 10px;
+            box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+            max-width: 400px;
+            width: 100%;
+            margin-top: 80px;
+        }
+
+        .form-container label {
+            display: block;
+            font-weight: bold;
+            margin-bottom: 5px;
+        }
+
+        .form-container input {
+            width: 100%;
+            padding: 8px;
+            margin-bottom: 15px;
+            border: 1px solid #ccc;
+            border-radius: 5px;
+        }
+
+        .form-container button {
+            width: 100%;
+            padding: 10px;
+            background-color: #007bff;
+            color: white;
+            border: none;
+            border-radius: 5px;
+            font-size: 16px;
+            cursor: pointer;
+        }
+
+        .form-container button:hover {
+            background-color: #0056b3;
+        }
+    </style>
+</head>
+<body>
+
+<!-- Navbar -->
+<div class="navbar">
+    <a href="index.html">Generation</a>
+    <a href="preview.html">Preview</a>
+</div>
+
+<!-- Form for Generation -->
+<div class="form-container">
+    <h2>Generate Content</h2>
+    <form id="generationForm">
+        <label for="storename">Store Name:</label>
+        <input type="text" id="storename" name="storename" placeholder="Enter store name" required>
+
+        <label for="imageurl">Image URL:</label>
+        <input type="url" id="imageurl" name="imageurl" placeholder="Enter image URL" required>
+
+        <button type="submit">Submit</button>
+    </form>
+</div>
+
+<script>
+    document.getElementById('generationForm').addEventListener('submit', async (event) => {
+        event.preventDefault();
+        const storename = document.getElementById('storename').value;
+        const imageurl = document.getElementById('imageurl').value;
+
+        try {
+            const url = `http://0.0.0.0:7860/generate?storename=${encodeURIComponent(storename)}&image_url=${encodeURIComponent(imageurl)}`;
+
+            const response = await fetch(url, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+            });
+
+            if (response.ok) {
+                alert('Generation request successful!');
+            } else {
+                alert('Generation failed.');
+            }
+        } catch (error) {
+            alert('Error submitting generation request.');
+        }
+    });
+</script>
+
+
+</body>
+</html>

static/preview.html

@@ -0,0 +1,415 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Preview</title>
+    <style>
+        body {
+            font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+            margin: 0;
+            padding: 0;
+            background-color: #f8f9fa;
+            min-height: 100vh;
+        }
+
+        .navbar {
+            background-color: #007bff;
+            color: white;
+            width: 100%;
+            padding: 15px 0;
+            text-align: center;
+            font-size: 20px;
+            font-weight: bold;
+            position: fixed;
+            top: 0;
+            left: 0;
+            z-index: 1000;
+            box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+        }
+
+        .navbar a {
+            color: white;
+            text-decoration: none;
+            margin: 0 15px;
+            font-size: 16px;
+        }
+
+        .navbar a:hover {
+            text-decoration: underline;
+        }
+
+        .form-container {
+            background: white;
+            padding: 30px;
+            border-radius: 12px;
+            box-shadow: 0 6px 12px rgba(0, 0, 0, 0.1);
+            max-width: 450px;
+            width: 90%;
+            margin: 100px auto 50px;
+            text-align: center;
+        }
+
+        .form-group {
+            margin-bottom: 20px;
+            display: flex;
+            gap: 15px;
+        }
+
+        .form-container select {
+            flex: 1;
+            padding: 12px;
+            border: 1px solid #ddd;
+            border-radius: 6px;
+            font-size: 16px;
+            background-color: white;
+        }
+
+        .button-group {
+            display: flex;
+            gap: 10px;
+            justify-content: center;
+            margin-top: 20px;
+        }
+
+        .btn {
+            padding: 12px 24px;
+            border: none;
+            border-radius: 6px;
+            font-size: 16px;
+            cursor: pointer;
+            transition: all 0.3s ease;
+        }
+
+        .btn-primary {
+            background-color: #007bff;
+            color: white;
+        }
+
+        .btn-success {
+            background-color: #28a745;
+            color: white;
+        }
+
+        .btn:hover {
+            transform: translateY(-2px);
+            box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);
+        }
+
+        /* Rest of the styles remain the same */
+        .content-grid {
+            max-width: 1200px;
+            margin: 0 auto;
+            padding: 20px;
+        }
+
+        .section {
+            margin-bottom: 40px;
+        }
+
+        .section-title {
+            font-size: 24px;
+            color: #333;
+            margin-bottom: 20px;
+            text-align: center;
+            font-weight: bold;
+            position: relative;
+            padding-bottom: 10px;
+        }
+
+        .section-title:after {
+            content: '';
+            position: absolute;
+            bottom: 0;
+            left: 50%;
+            transform: translateX(-50%);
+            width: 60px;
+            height: 3px;
+            background-color: #007bff;
+            border-radius: 2px;
+        }
+
+        .image-row {
+            display: flex;
+            justify-content: center;
+            gap: 20px;
+            flex-wrap: wrap;
+            margin-bottom: 30px;
+        }
+
+        .image-card {
+            background: white;
+            border-radius: 12px;
+            overflow: hidden;
+            box-shadow: 0 6px 12px rgba(0, 0, 0, 0.1);
+            transition: transform 0.3s ease;
+            width: 300px;
+            height: 300px;
+        }
+
+        .image-card img {
+            width: 100%;
+            height: 100%;
+            object-fit: cover;
+            transition: transform 0.3s ease;
+        }
+
+        .image-card:hover img {
+            transform: scale(1.05);
+        }
+
+        .video-container {
+            width: 100%;
+            max-width: 800px;
+            margin: 0 auto;
+            background: white;
+            border-radius: 12px;
+            overflow: hidden;
+            box-shadow: 0 6px 12px rgba(0, 0, 0, 0.1);
+        }
+
+        .video-container video {
+            width: 100%;
+            display: block;
+        }
+
+        .loading {
+            position: fixed;
+            top: 50%;
+            left: 50%;
+            transform: translate(-50%, -50%);
+            background: rgba(255, 255, 255, 0.9);
+            padding: 20px 40px;
+            border-radius: 8px;
+            box-shadow: 0 4px 12px rgba(0, 0, 0, 0.15);
+            z-index: 1000;
+        }
+
+        .success-message {
+            position: fixed;
+            top: 20px;
+            right: 20px;
+            background: #28a745;
+            color: white;
+            padding: 15px 25px;
+            border-radius: 6px;
+            animation: slideIn 0.3s ease-out;
+            z-index: 1001;
+        }
+
+        @keyframes slideIn {
+            from { transform: translateX(100%); }
+            to { transform: translateX(0); }
+        }
+    </style>
+</head>
+<body>
+    <div class="navbar">
+        <a href="index.html">Generation</a>
+        <a href="preview.html">Preview</a>
+    </div>
+
+    <div class="form-container">
+        <h2>Preview Content</h2>
+        <div class="form-group">
+            <select id="necklaceSelect">
+                <option value="">Select Necklace ID</option>
+            </select>
+            <select id="modelSelect" disabled>
+                <option value="">Select Model</option>
+            </select>
+        </div>
+        <div class="button-group">
+            <button class="btn btn-primary" id="previewBtn">Preview</button>
+            <button class="btn btn-success" id="approveBtn">Approve</button>
+        </div>
+    </div>
+
+    <div id="contentContainer" class="content-grid"></div>
+    <div id="loading" class="loading" style="display: none;">Loading...</div>
+
+    <script>
+        let necklaceData = {};
+
+        async function fetchNecklaceIds() {
+            try {
+                const response = await fetch('http://0.0.0.0:7860/list_necklace_id');
+                necklaceData = await response.json();
+                const select = document.getElementById('necklaceSelect');
+
+                Object.keys(necklaceData).forEach(id => {
+                    const option = document.createElement('option');
+                    option.value = id;
+                    option.textContent = id;
+                    select.appendChild(option);
+                });
+            } catch (error) {
+                console.error('Error fetching necklace IDs:', error);
+                alert('Failed to load necklace IDs');
+            }
+        }
+
+        document.getElementById('necklaceSelect').addEventListener('change', function() {
+            const modelSelect = document.getElementById('modelSelect');
+            modelSelect.innerHTML = '<option value="">Select Model</option>';
+
+            if (this.value) {
+                const models = necklaceData[this.value];
+                models.forEach(model => {
+                    const option = document.createElement('option');
+                    option.value = model;
+                    option.textContent = model;
+                    modelSelect.appendChild(option);
+                });
+                modelSelect.disabled = false;
+            } else {
+                modelSelect.disabled = true;
+            }
+        });
+
+        function showLoading(show) {
+            document.getElementById('loading').style.display = show ? 'block' : 'none';
+        }
+
+        function showSuccessMessage(message) {
+            const messageDiv = document.createElement('div');
+            messageDiv.className = 'success-message';
+            messageDiv.textContent = message;
+            document.body.appendChild(messageDiv);
+            setTimeout(() => messageDiv.remove(), 3000);
+        }
+
+        function createSection(title) {
+            const section = document.createElement('div');
+            section.className = 'section';
+            const titleElement = document.createElement('h2');
+            titleElement.className = 'section-title';
+            titleElement.textContent = title;
+            section.appendChild(titleElement);
+            return section;
+        }
+
+        function createImageCard(url) {
+            const card = document.createElement('div');
+            card.className = 'image-card';
+            const img = document.createElement('img');
+            img.src = url;
+            img.alt = 'Preview Image';
+            img.loading = 'lazy';
+            card.appendChild(img);
+            return card;
+        }
+
+        function displayContent(data) {
+            const container = document.getElementById('contentContainer');
+            container.innerHTML = '';
+
+            // NTO Section
+            if (data.nto && data.nto.length > 0) {
+                const ntoSection = createSection('Necklace Try-On (NTO)');
+                const ntoRow = document.createElement('div');
+                ntoRow.className = 'image-row';
+                data.nto.forEach(url => {
+                    ntoRow.appendChild(createImageCard(url));
+                });
+                ntoSection.appendChild(ntoRow);
+                container.appendChild(ntoSection);
+            }
+
+            // CTO Section
+            if (data.cto && data.cto.length > 0) {
+                const ctoSection = createSection('Clothing Try-On (CTO)');
+                for (let i = 0; i < data.cto.length; i += 2) {
+                    const row = document.createElement('div');
+                    row.className = 'image-row';
+                    row.appendChild(createImageCard(data.cto[i]));
+                    if (data.cto[i + 1]) {
+                        row.appendChild(createImageCard(data.cto[i + 1]));
+                    }
+                    ctoSection.appendChild(row);
+                }
+                container.appendChild(ctoSection);
+            }
+
+            // MTO Section
+            if (data.mto && data.mto.length > 0) {
+                const mtoSection = createSection('Makeup Try-On (MTO)');
+                for (let i = 0; i < data.mto.length; i += 2) {
+                    const row = document.createElement('div');
+                    row.className = 'image-row';
+                    row.appendChild(createImageCard(data.mto[i]));
+                    if (data.mto[i + 1]) {
+                        row.appendChild(createImageCard(data.mto[i + 1]));
+                    }
+                    mtoSection.appendChild(row);
+                }
+                container.appendChild(mtoSection);
+            }
+
+            // Video Section
+            if (data.video && data.video.length > 0) {
+                const videoSection = createSection('Video Preview');
+                const videoContainer = document.createElement('div');
+                videoContainer.className = 'video-container';
+                const video = document.createElement('video');
+                video.src = data.video[0];
+                video.controls = true;
+                videoContainer.appendChild(video);
+                videoSection.appendChild(videoContainer);
+                container.appendChild(videoSection);
+            }
+        }
+
+        document.getElementById('previewBtn').addEventListener('click', async () => {
+            const necklaceId = document.getElementById('necklaceSelect').value;
+            const modelName = document.getElementById('modelSelect').value;
+
+            if (!necklaceId || !modelName) {
+                alert('Please select both Necklace ID and Model');
+                return;
+            }
+
+            showLoading(true);
+            try {
+                const response = await fetch(
+                    `http://0.0.0.0:7860/image_fetch_product_page?necklace_id=${necklaceId}&model_name=${modelName}`,
+                    { method: 'POST' }
+                );
+                const data = await response.json();
+                displayContent(data);
+            } catch (error) {
+                console.error('Error:', error);
+                alert('Error fetching preview data');
+            } finally {
+                showLoading(false);
+            }
+        });
+
+        document.getElementById('approveBtn').addEventListener('click', async () => {
+            const necklaceId = document.getElementById('necklaceSelect').value;
+            if (!necklaceId) {
+                alert('Please select a necklace ID');
+                return;
+            }
+
+            showLoading(true);
+            try {
+                const response = await fetch(
+                    `http://0.0.0.0:7860/approve?necklace_id=${necklaceId}&model_name=${document.getElementById('modelSelect').value}`,
+                    { method: 'POST' }
+                );
+                const data = await response.json();
+                showSuccessMessage('Necklace approved successfully!');
+            } catch (error) {
+                console.error('Error:', error);
+                alert('Error approving necklace');
+            } finally {
+                showLoading(false);
+            }
+        });
+
+        // Initialize page
+        fetchNecklaceIds();
+    </script>
+</body>
+</html>