Upload 3 files

app.py

@@ -0,0 +1,147 @@
+import gradio as gr
+import json
+import boto3
+import os
+from typing import List, Dict
+import constants
+
+# Initialize S3 client
+s3 = boto3.client("s3")
+
+LORA_OPTIONS = ["Arjun", "Khushi", "Priya", "Ishaan"]
+
+
+def extract_unique_characters(json_file: str):
+    try:
+        with open(json_file, "r") as f:
+            data = json.load(f)
+            characters = set()
+            for item in data.get("result", []):
+                if item.get("character"):
+                    characters.add(item["character"])
+            return sorted(list(characters))  # Sort to maintain consistent order
+    except Exception as e:
+        print(f"Error reading JSON: {str(e)}")
+        return []
+
+
+def save_episode_json(
+    comic_id: str, epi_num: int, json_file: str, *character_lora_pairs
+):
+    try:
+        # Read the uploaded JSON file
+        with open(json_file, "r") as f:
+            json_content = json.load(f)
+
+        # Save the episode JSON with fixed name episode_old.json
+        save_path = f"{comic_id}/episodes/episode-{epi_num}/episode_old.json"
+        s3.put_object(
+            Bucket=constants.aws_bucket,
+            Key=save_path,
+            Body=json.dumps(json_content),
+            ContentType="application/json",
+        )
+
+        # Read existing metadata from S3
+        try:
+            response = s3.get_object(
+                Bucket=constants.aws_bucket, Key="comic_details/comic_details.json"
+            )
+            metadata = json.loads(response["Body"].read().decode("utf-8"))
+        except:
+            # If file doesn't exist, create new metadata with the given structure
+            metadata = {"comics": []}
+
+        # Create character-lora mapping in order
+        char_lora = {}
+        num_pairs = len(character_lora_pairs) // 2
+        for i in range(num_pairs):
+            character = character_lora_pairs[i]  # Get character from first half
+            lora = character_lora_pairs[i + num_pairs]  # Get lora from second half
+            if character:  # Only add if character name is not empty
+                char_lora[character] = lora
+
+        # Update or add comic entry
+        comic_found = False
+        for comic in metadata["comics"]:
+            if comic["comic_id"] == comic_id:
+                # Update existing comic
+                comic["char_lora"] = char_lora
+                if epi_num not in comic["episode_number"]:
+                    comic["episode_number"].append(epi_num)
+                comic["is_generate"] = False  # Always set to false
+                comic_found = True
+                break
+
+        if not comic_found:
+            # Add new comic entry
+            metadata["comics"].append(
+                {
+                    "comic_id": comic_id,
+                    "char_lora": char_lora,
+                    "episode_number": [epi_num],
+                    "is_generate": False,  # Always set to false
+                }
+            )
+
+        # Save updated metadata back to S3
+        s3.put_object(
+            Bucket=constants.aws_bucket,
+            Key="comic_details/comic_details.json",
+            Body=json.dumps(metadata, indent=4),
+            ContentType="application/json",
+        )
+
+        return "Successfully saved episode and updated metadata!"
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+
+def update_character_fields(json_file: str):
+    characters = extract_unique_characters(json_file)
+    if len(characters) > 6:
+        print(
+            f"Warning: Found {len(characters)} characters, but only showing first 6. Extra characters: {characters[5:]}"
+        )
+        characters = characters[:6]  # Take only first 5 characters
+    # Pad with empty strings if we have fewer than 5 characters
+    return characters + [""] * (6 - len(characters))
+
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Comic Episode Uploader")
+
+    with gr.Row():
+        comic_id = gr.Textbox(label="Comic ID")
+        episode_num = gr.Number(label="Episode Number", precision=0)
+
+    json_file = gr.File(label="Upload JSON File", file_types=[".json"])
+
+    # Create 5 fixed character-lora pairs
+    character_boxes = []
+    lora_dropdowns = []
+    for i in range(6):
+        with gr.Row():
+            character = gr.Textbox(label=f"Character Name {i+1}", interactive=True)
+            lora = gr.Dropdown(choices=LORA_OPTIONS, label=f"Lora Name {i+1}")
+            character_boxes.append(character)
+            lora_dropdowns.append(lora)
+
+    # Update character fields when JSON is uploaded
+    json_file.change(
+        fn=update_character_fields, inputs=[json_file], outputs=character_boxes
+    )
+
+    submit_btn = gr.Button("Submit")
+    output = gr.Textbox(label="Status")
+
+    submit_btn.click(
+        fn=save_episode_json,
+        inputs=[comic_id, episode_num, json_file] + character_boxes + lora_dropdowns,
+        outputs=output,
+    )
+
+if __name__ == "__main__":
+    demo.launch(
+        auth=("admin", "Qrt@12*34#immersfy"), share=True, ssr_mode=False, debug=True
+    )

constants.py

@@ -0,0 +1 @@
+aws_bucket = "blix-production"

requirements.txt

@@ -0,0 +1,99 @@
+aiofiles==24.1.0
+annotated-types==0.7.0
+anyio==4.6.0
+boto3==1.35.41
+botocore==1.35.41
+build==1.2.2.post1
+cachetools==5.5.0
+certifi==2024.8.30
+cfgv==3.4.0
+charset-normalizer==3.4.0
+click==8.1.7
+colorama==0.4.6
+colorlog==6.8.2
+dacite==1.9.2
+distlib==0.3.9
+distro==1.9.0
+elevenlabs==1.57.0
+exceptiongroup==1.2.2
+fastapi==0.115.12
+ffmpy==0.5.0
+filelock==3.16.1
+fsspec==2025.3.2
+google-api-core==2.23.0
+google-api-python-client==2.149.0
+google-auth==2.35.0
+google-auth-httplib2==0.2.0
+google-auth-oauthlib==1.2.1
+googleapis-common-protos==1.66.0
+gradio==5.28.0
+gradio_client==1.10.0
+groovy==0.1.2
+h11==0.14.0
+httpcore==1.0.6
+httplib2==0.22.0
+httpx==0.27.2
+huggingface-hub==0.30.2
+identify==2.6.1
+idna==3.10
+Jinja2==3.1.4
+jmespath==1.0.1
+markdown-it-py==3.0.0
+MarkupSafe==3.0.1
+mdurl==0.1.2
+nodeenv==1.9.1
+numpy==2.2.5
+oauth2client==4.1.3
+oauthlib==3.2.2
+openai==1.23.6
+orjson==3.10.18
+packaging==24.1
+pandas==2.2.3
+pillow==11.0.0
+pip==25.0
+pip-tools==7.4.1
+platformdirs==4.3.6
+pre_commit==4.0.1
+proto-plus==1.25.0
+protobuf==5.28.3
+pyasn1==0.6.1
+pyasn1_modules==0.4.1
+pydantic==2.9.2
+pydantic_core==2.23.4
+PyDrive==1.3.1
+pydub==0.25.1
+Pygments==2.19.1
+PyMySQL==1.1.1
+pyparsing==3.2.0
+pyproject_hooks==1.2.0
+python-dateutil==2.9.0.post0
+python-dotenv==1.0.1
+python-multipart==0.0.20
+pytz==2025.2
+PyYAML==6.0.2
+requests==2.32.3
+requests-oauthlib==2.0.0
+rich==14.0.0
+rsa==4.9
+ruff==0.11.7
+s3transfer==0.10.3
+safehttpx==0.1.6
+semantic-version==2.10.0
+setuptools==75.8.0
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.1
+starlette==0.46.2
+tomli==2.0.2
+tomlkit==0.13.2
+tqdm==4.66.5
+typer==0.15.3
+typing_extensions==4.12.2
+tzdata==2025.2
+uritemplate==4.1.1
+urllib3==2.2.3
+uvicorn==0.32.0
+virtualenv==20.26.6
+websocket-client==1.8.0
+websockets==15.0.1
+wheel==0.44.0