Upload 2 files

app.py

@@ -1,68 +1,84 @@
 
 import gradio as gr
 import json
-import os
 
-PRESET_FILE = "presets.json"
+camera_film_map = {
+    "Canon EOS R5": ["Kodak Portra 400", "Kodak Ektar 100", "Kodak Gold 200"],
+    "Nikon Z8": ["Cinestill 800T", "Kodak Tri-X 400", "Fujifilm Superia X-TRA 400"],
+    "Sony Alpha 7R V": ["Agfa Vista 200", "Kodak Ultramax 400", "Fujifilm C200"],
+    "Fujifilm GFX 100 II": ["Fujifilm Pro 400H", "Fujifilm Provia 100F", "Fujifilm Velvia 50"],
+    "Leica M11": ["Ilford HP5 Plus", "Kodak T-Max 100", "AgfaPhoto APX 100"],
+    "Panasonic Lumix S5II": ["Kodak Vision3 500T", "Cinestill 50D", "Kodak Gold 200"],
+    "Olympus OM-D E-M1 Mark III": ["Kodak Ektachrome E100", "Fujifilm Neopan 100", "Kodak ColorPlus 200"],
+    "Pentax K-3 Mark III": ["Ilford Delta 3200", "Fomapan 100 Classic", "Rollei Retro 400S"],
+    "Canon EOS 5D Mark IV": ["Kodak Pro Image 100", "Kodak Vision3 250D", "Fujicolor 100"],
+    "Nikon D850": ["AgfaPhoto Vista plus 400", "Fujicolor Industrial 100", "Kodak T-Max 400"],
+    "Sony A7 III": ["Cinestill BwXX", "Kodak Gold 100", "Fujifilm Natura 1600"],
+    "Fujifilm X-T5": ["Fujifilm Pro 160NS", "Kodak Elite Chrome 100", "Fujifilm Fortia SP"],
+    "Leica Q2": ["Ilford XP2 Super 400", "Kodak ColorPlus 100", "Fujicolor Press 800"],
+    "Panasonic GH6": ["Agfa Scala 200x", "Kodak High-Speed Infrared", "Rollei Vario Chrome"],
+    "Olympus PEN-F": ["Fujifilm Sensia 200", "Kodak Ektachrome 64T", "Kodak Supra 400"],
+    "Pentax 645Z": ["Kodak Panatomic-X", "Agfa Portrait XPS 160", "Kodak Vericolor III"],
+    "Canon EOS R6 Mark II": ["Kodak Vision3 200T", "Kodak Gold 400", "Ilford Pan F Plus 50"],
+    "Nikon Z9": ["Fujifilm Velvia 100", "Kodak Ektachrome 160", "Ilford Delta 100"],
+    "Sony FX3": ["Kodak Vision3 500D", "Cinestill 400D", "Fujifilm Provia 400X"],
+    "Fujifilm X-H2": ["Kodak Technical Pan", "AgfaPhoto CT Precisa 100", "Fujifilm Velvia 100F"]
+}
 
-if not os.path.exists(PRESET_FILE):
-    with open(PRESET_FILE, "w") as f:
-        json.dump({}, f)
-
-def load_presets():
-    with open(PRESET_FILE, "r") as f:
-        return json.load(f)
-
-cameras = ['Canon EOS R5', 'Nikon Z8', 'Sony Alpha 7R V', 'Fujifilm GFX 100 II', 'Leica M11']
-lens_map = {'Canon EOS R5': ['Canon RF 50mm f/1.2L USM'], 'Nikon Z8': ['Nikon Z 50mm f/1.2 S'], 'Sony Alpha 7R V': ['Sony FE 85mm f/1.4 GM'], 'Fujifilm GFX 100 II': ['Fujifilm GF 80mm f/1.7 R WR'], 'Leica M11': ['Leica Summilux-M 50mm f/1.4 ASPH']}
-films = [
-    "Kodak Portra 400", "Kodak Ektar 100", "Cinestill 800T", "Fujifilm Provia 100F", "Kodak Gold 200"
-]
+mj_versions = ["6", "6.1", "7"]
 light_conditions = [
     "Golden Hour", "Blue Hour", "Overcast Light", "Diffused Light", "Backlighting",
     "Soft Ambient Light", "Low-Key Lighting", "High-Key Lighting", "Window Light", "Dappled Light"
 ]
 
-def update_lens(camera):
-    return gr.update(choices=lens_map.get(camera, []), value=lens_map.get(camera, [])[0])
+def update_film_dropdown(camera):
+    return gr.update(choices=camera_film_map[camera], value=camera_film_map[camera][0])
 
-def generate_prompt(preset, custom_scene, camera, lens, film, light, aspect_ratio, quality, stylize, seed, num_prompts):
-    presets = load_presets()
-    scene_description = custom_scene if preset == "Custom Input" else presets.get(preset, "")
-    mj_parameters = f"--ar {aspect_ratio} --q {quality} --s {stylize}"
-    if seed != "":
-        mj_parameters += f" --seed {seed}"
+def toggle_visibility(engine):
+    visible = engine == "Midjourney"
+    return gr.update(visible=visible), gr.update(visible=visible), gr.update(visible=visible), gr.update(visible=visible)
+
+def generate_prompt(engine, mj_version, camera, lens, film, light, aspect_ratio, quality, stylize, seed, scene, num):
+    mj_args = f"--ar {aspect_ratio} --q {quality} --s {stylize}"
+    if seed:
+        mj_args += f" --seed {seed}"
     prompts = []
-    for _ in range(num_prompts):
-        prompt = f"{scene_description}, shot with {camera} and {lens}, using {film} film, under {light} {mj_parameters}"
+    for _ in range(num):
+        if engine == "Midjourney":
+            prompt = "{}, shot with {} and {}, using {} film, under {} {} --v {}".format(scene, camera, lens, film, light, mj_args, mj_version)
+        else:
+            prompt = "A detailed scene of {}, captured using a {} and {}, simulated with {} film look and {} lighting.".format(scene, camera, lens, film, light)
         prompts.append(prompt)
     return "\n".join(prompts)
 
 with gr.Blocks() as demo:
     gr.Markdown("# TischEins.org | Midjourney Prompt Engine")
 
-    with gr.Tab("Prompt Generator"):
-        preset_choice = gr.Dropdown(choices=list(load_presets().keys()) + ["Custom Input"], label="Select Preset")
-        custom_scene = gr.Textbox(label="Or enter your custom scene")
+    engine_choice = gr.Radio(choices=["Midjourney", "ChatGPT Sora"], label="Select Target Engine", value="Midjourney")
+    mj_version_dropdown = gr.Dropdown(choices=mj_versions, label="Midjourney Version", value="7", visible=True)
+
+    camera_dropdown = gr.Dropdown(choices=list(camera_film_map.keys()), label="Camera Model")
+    lens_input = gr.Textbox(label="Lens Model")
+    film_dropdown = gr.Dropdown(choices=[], label="Color Film Type")
+    light_dropdown = gr.Dropdown(choices=light_conditions, label="Light Condition")
 
-        camera_dropdown = gr.Dropdown(choices=cameras, label="Camera Model")
-        lens_dropdown = gr.Dropdown(choices=[], label="Lens Model")
+    aspect_ratio = gr.Textbox(label="Aspect Ratio", value="5:4", visible=True)
+    quality = gr.Textbox(label="Quality", value="2", visible=True)
+    stylize = gr.Textbox(label="Stylize", value="0", visible=True)
+    seed = gr.Textbox(label="Seed (optional)")
 
-        film_dropdown = gr.Dropdown(choices=films, label="Color Film Type")
-        light_dropdown = gr.Dropdown(choices=light_conditions, label="Light Condition")
-        aspect_ratio = gr.Textbox(label="Aspect Ratio", value="5:4")
-        quality = gr.Textbox(label="Quality", value="2")
-        stylize = gr.Textbox(label="Stylize", value="750")
-        seed = gr.Textbox(label="Seed (optional)", placeholder="Leave empty for random")
-        num = gr.Slider(1, 100, 10, step=1, label="Number of Prompts")
+    scene = gr.Textbox(label="Scene Description", placeholder="Describe your scene here...")
+    num = gr.Slider(1, 50, value=5, label="Number of Prompts")
 
-        camera_dropdown.change(fn=update_lens, inputs=camera_dropdown, outputs=lens_dropdown)
+    output = gr.Textbox(label="Generated Prompts", lines=20)
+    generate_btn = gr.Button("Generate Prompts")
 
-        btn = gr.Button("Generate Prompts")
-        output = gr.Textbox(label="Generated Prompts", lines=20)
-        btn.click(fn=generate_prompt,
-                  inputs=[preset_choice, custom_scene, camera_dropdown, lens_dropdown, film_dropdown, light_dropdown, aspect_ratio, quality, stylize, seed, num],
-                  outputs=output)
+    camera_dropdown.change(fn=update_film_dropdown, inputs=camera_dropdown, outputs=film_dropdown)
+    engine_choice.change(fn=toggle_visibility, inputs=engine_choice, outputs=[mj_version_dropdown, aspect_ratio, quality, stylize])
+    generate_btn.click(fn=generate_prompt, inputs=[
+        engine_choice, mj_version_dropdown, camera_dropdown, lens_input, film_dropdown,
+        light_dropdown, aspect_ratio, quality, stylize, seed, scene, num
+    ], outputs=output)
 
     gr.Markdown('<a href="https://tischeins.org" target="_blank">Pro Studio Mode by TischEins.org</a>')
 

requirements.txt

@@ -1,2 +1 @@
 gradio
-pandas