Upload 3 files

app.py

@@ -2,93 +2,371 @@
 import gradio as gr
 import pandas as pd
 import random
+import json
 
-# Sample mappings (should include all 20 entries)
 camera_lens_map = {
-    "Canon EOS R5": ["Canon RF 50mm f/1.2L USM", "Canon RF 35mm f/1.8 IS Macro STM", "Canon RF 85mm f/1.2L USM"],
-    "Nikon Z8": ["Nikon Z 50mm f/1.2 S", "Nikon Z 85mm f/1.8 S", "Nikon Z 24-70mm f/2.8 S"]
-    # ... add remaining cameras
+  "Canon EOS R5": [
+    "Canon RF 50mm f/1.2L USM",
+    "Canon RF 35mm f/1.8 IS Macro STM",
+    "Canon RF 85mm f/1.2L USM"
+  ],
+  "Canon EOS R3": [
+    "Canon RF 24-70mm f/2.8L IS USM",
+    "Canon RF 50mm f/1.2L USM",
+    "Canon RF 85mm f/1.2L USM"
+  ],
+  "Nikon Z8": [
+    "Nikon Z 50mm f/1.2 S",
+    "Nikon Z 85mm f/1.8 S",
+    "Nikon Z 24-70mm f/2.8 S"
+  ],
+  "Nikon Z9": [
+    "Nikon Z 70-200mm f/2.8 VR S",
+    "Nikon Z 24-120mm f/4 S",
+    "Nikon Z 50mm f/1.8 S"
+  ],
+  "Sony Alpha 7R V": [
+    "Sony FE 85mm f/1.4 GM",
+    "Sony FE 24-105mm f/4 G OSS",
+    "Sony FE 50mm f/1.2 GM"
+  ],
+  "Sony Alpha 1": [
+    "Sony FE 35mm f/1.4 GM",
+    "Sony FE 50mm f/1.2 GM",
+    "Sony FE 70-200mm f/2.8 GM OSS"
+  ],
+  "Fujifilm GFX 100 II": [
+    "Fujifilm GF 80mm f/1.7 R WR",
+    "Fujifilm GF 45mm f/2.8 R WR",
+    "Fujifilm GF 110mm f/2 R LM WR"
+  ],
+  "Fujifilm X-T5": [
+    "Fujinon XF 56mm f/1.2 R WR",
+    "Fujinon XF 23mm f/1.4 R LM WR",
+    "Fujinon XF 16-55mm f/2.8 R LM WR"
+  ],
+  "Leica M11": [
+    "Leica Summilux-M 50mm f/1.4 ASPH",
+    "Leica APO-Summicron-M 75mm f/2 ASPH",
+    "Leica Elmarit-M 28mm f/2.8 ASPH"
+  ],
+  "Leica SL2-S": [
+    "Leica APO-Summicron-SL 75mm f/2 ASPH",
+    "Leica Vario-Elmar-SL 24-90mm f/2.8-4 ASPH",
+    "Leica APO-Vario-Elmar-SL 90-280mm f/2.8-4"
+  ],
+  "Panasonic Lumix S5 II": [
+    "Panasonic Lumix S Pro 24-70mm f/2.8",
+    "Panasonic Lumix S Pro 50mm f/1.4",
+    "Panasonic Lumix S 85mm f/1.8"
+  ],
+  "Panasonic GH6": [
+    "Panasonic Leica DG Nocticron 42.5mm f/1.2",
+    "Panasonic Lumix G X Vario 12-35mm f/2.8",
+    "Panasonic Leica DG Summilux 25mm f/1.4"
+  ],
+  "Hasselblad X2D 100C": [
+    "Hasselblad XCD 80mm f/1.9",
+    "Hasselblad XCD 45mm f/2.8",
+    "Hasselblad XCD 120mm f/3.5 Macro"
+  ],
+  "Olympus OM-1": [
+    "Olympus M.Zuiko 25mm f/1.2 PRO",
+    "Olympus M.Zuiko 45mm f/1.2 PRO",
+    "Olympus M.Zuiko 12-100mm f/4 IS PRO"
+  ],
+  "Olympus PEN-F": [
+    "Olympus M.Zuiko 17mm f/1.8",
+    "Olympus M.Zuiko 45mm f/1.8",
+    "Olympus M.Zuiko 25mm f/1.8"
+  ],
+  "Pentax K-1 Mark II": [
+    "Pentax D FA* 50mm f/1.4 SDM AW",
+    "Pentax D FA 24-70mm f/2.8 SDM AW",
+    "Pentax D FA 100mm f/2.8 Macro"
+  ],
+  "Pentax 645Z": [
+    "Pentax FA 645 75mm f/2.8",
+    "Pentax FA 645 45mm f/2.8",
+    "Pentax FA 645 150mm f/2.8"
+  ],
+  "Sigma fp L": [
+    "Sigma 35mm f/1.4 DG DN Art",
+    "Sigma 45mm f/2.8 DG DN Contemporary",
+    "Sigma 85mm f/1.4 DG DN Art"
+  ],
+  "GoPro Hero 12 Black": [
+    "GoPro Standard Lens",
+    "GoPro Max Lens Mod",
+    "GoPro Dive Lens Mod"
+  ],
+  "Blackmagic Pocket Cinema Camera 6K Pro": [
+    "Sigma 18-35mm f/1.8 Art",
+    "Canon EF 50mm f/1.2L",
+    "Zeiss CP.3 85mm T2.1"
+  ]
 }
 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"]
-    # ... add remaining cameras
+  "Canon EOS R5": [
+    "Kodak Portra 400",
+    "Kodak Ektar 100",
+    "Kodak Gold 200"
+  ],
+  "Canon EOS R3": [
+    "Cinestill 800T",
+    "Kodak Vision3 500T",
+    "Fujifilm Pro 400H"
+  ],
+  "Nikon Z8": [
+    "Cinestill 800T",
+    "Kodak Tri-X 400",
+    "Fujifilm Superia X-TRA 400"
+  ],
+  "Nikon Z9": [
+    "Fujifilm Pro 400H",
+    "Kodak Ektar 100",
+    "Kodak Ultramax 400"
+  ],
+  "Sony Alpha 7R V": [
+    "Agfa Vista 200",
+    "Kodak Ultramax 400",
+    "Fujifilm C200"
+  ],
+  "Sony Alpha 1": [
+    "Kodak Portra 800",
+    "Kodak Gold 200",
+    "Fujifilm Velvia 50"
+  ],
+  "Fujifilm GFX 100 II": [
+    "Fujifilm Provia 100F",
+    "Fujifilm Velvia 50",
+    "Fujifilm Pro 400H"
+  ],
+  "Fujifilm X-T5": [
+    "Fujifilm Pro 160NS",
+    "Kodak Elite Chrome 100",
+    "Fujifilm Natura 1600"
+  ],
+  "Leica M11": [
+    "Ilford HP5 Plus",
+    "Kodak T-Max 100",
+    "AgfaPhoto APX 100"
+  ],
+  "Leica SL2-S": [
+    "Lomography Metropolis 100-400",
+    "Kodak Vision3 250D",
+    "Rollei Vario Chrome 320"
+  ],
+  "Panasonic Lumix S5 II": [
+    "Kodak Vision3 500T",
+    "CineStill 800T",
+    "Kodak Pro Image 100"
+  ],
+  "Panasonic GH6": [
+    "Kodak Aerocolor IV 125",
+    "Kodak Ektachrome E100",
+    "Fujifilm Neopan 100"
+  ],
+  "Hasselblad X2D 100C": [
+    "Kodak Elite Chrome 100",
+    "Kodak Vision3 500T",
+    "Kodak Vision3 250D"
+  ],
+  "Olympus OM-1": [
+    "Kodak Pro Image 100",
+    "Ilford Delta 3200",
+    "Fomapan 100 Classic"
+  ],
+  "Olympus PEN-F": [
+    "Fujifilm Sensia 200",
+    "Kodak Ektachrome 64T",
+    "Kodak Supra 400"
+  ],
+  "Pentax K-1 Mark II": [
+    "Adox Color Mission 200",
+    "Ferrania P30 Alpha",
+    "Rollei Retro 400S"
+  ],
+  "Pentax 645Z": [
+    "Kodak Panatomic-X",
+    "Agfa Portrait XPS 160",
+    "Kodak Vericolor III"
+  ],
+  "Sigma fp L": [
+    "Ferrania P30 Alpha",
+    "Cinestill BwXX",
+    "Kodak Portra 800"
+  ],
+  "GoPro Hero 12 Black": [
+    "Polaroid Originals Color i-Type",
+    "Fujifilm Instax Mini Film",
+    "Impossible Project B&W Film"
+  ],
+  "Blackmagic Pocket Cinema Camera 6K Pro": [
+    "Kodak Vision3 500T",
+    "Kodak Vision3 250D",
+    "Fujifilm Eterna 250D"
+  ]
 }
 light_conditions = [
-    # 20 conditions
-    "Golden Hour","Blue Hour","Backlighting","Overcast Light","High Noon Sunlight",
-    "Twilight Glow","Morning Fog","Studio Lighting","Soft Ambient Light","Window Light",
-    "Dappled Light","Hard Shadow","Spotlight","Stage Lighting","Silhouette Light",
-    "Fluorescent Light","Candlelight","Moonlight","Streetlight Glow","Reflected Light"
+  "Golden Hour",
+  "Blue Hour",
+  "Backlighting",
+  "Overcast Light",
+  "High Noon Sunlight",
+  "Twilight Glow",
+  "Morning Fog",
+  "Studio Lighting",
+  "Soft Ambient Light",
+  "Window Light",
+  "Dappled Light",
+  "Hard Shadow",
+  "Spotlight",
+  "Stage Lighting",
+  "Silhouette Light",
+  "Fluorescent Light",
+  "Candlelight",
+  "Moonlight",
+  "Streetlight Glow",
+  "Reflected Light"
 ]
-mj_versions = ["6","6.1","7"]
+mj_versions = ["6", "6.1", "7"]
+scene_variations = [
+  "with dramatic lens flare",
+  "with soft bokeh background",
+  "with rich cinematic contrast",
+  "with muted pastel tones",
+  "with vibrant color pop",
+  "with deep shadows",
+  "with high-key brightness",
+  "with low-key mood",
+  "with subtle grain texture",
+  "with glossy reflections",
+  "with backlit rim light",
+  "with fine detail sharpness",
+  "with wide-angle perspective",
+  "with telephoto compression",
+  "with macro-level close-up",
+  "with panoramic scope",
+  "with dreamy glow",
+  "with crisp clarity",
+  "with vintage film grain",
+  "with modern hyperrealism"
+]
+presets = {
+  "Sunset Portrait": {
+    "camera": "Canon EOS R5",
+    "lens": "Canon RF 50mm f/1.2L USM",
+    "film": "Kodak Portra 400",
+    "light": "Golden Hour",
+    "scene": "A woman standing in a field of tall grass during sunset, soft wind in her hair",
+    "version": "7"
+  },
+  "Urban Streetstyle": {
+    "camera": "Sony Alpha 7R V",
+    "lens": "Sony FE 85mm f/1.4 GM",
+    "film": "Kodak Ultramax 400",
+    "light": "Backlighting",
+    "scene": "Young man walking through a neon-lit city street at night with glowing billboards",
+    "version": "6.1"
+  },
+  "Forest Fantasy": {
+    "camera": "Fujifilm GFX 100 II",
+    "lens": "Fujifilm GF 80mm f/1.7 R WR",
+    "film": "Fujifilm Provia 100F",
+    "light": "Diffused Light",
+    "scene": "A mysterious elf walking through a foggy forest with glowing mushrooms",
+    "version": "6"
+  },
+  "Classic Black and White": {
+    "camera": "Leica M11",
+    "lens": "Leica Summilux-M 50mm f/1.4 ASPH",
+    "film": "Ilford HP5 Plus",
+    "light": "Soft Ambient Light",
+    "scene": "Man reading newspaper in a Parisian caf\u00e9, 1950s",
+    "version": "7"
+  }
+}
 
-def update_lens(camera): return gr.update(choices=camera_lens_map.get(camera, []))
-def update_film(camera): return gr.update(choices=camera_film_map.get(camera, []))
+def apply_preset(p):
+    pre = presets[p]
+    return pre["camera"], pre["lens"], pre["film"], pre["light"], pre["scene"], pre["version"]
 
-def generate_midjourney(camera,lens,film,light,mj_version,aspect_ratio,quality,stylize,seed,scene,num):
-    prompts=[]
-    for _ in range(num):
-        prompt=f"{scene}, shot with {camera} and {lens}, using {film} film, under {light} --ar {aspect_ratio} --q {quality} --s {stylize}" + (f" --seed {seed}" if seed else "") + f" --v {mj_version}"
-        prompts.append(prompt)
-    return "\n".join(prompts), prompts
+def update_lens(camera):
+    return gr.update(choices=camera_lens_map.get(camera, []), value=camera_lens_map[camera][0])
+
+def update_film(camera):
+    return gr.update(choices=camera_film_map.get(camera, []), value=camera_film_map[camera][0])
 
-def generate_sora(camera,lens,film,light,scene,num):
-    prompts=[]
-    for _ in range(num):
-        prompts.append(f"A detailed scene of {scene}, captured using a {camera} and {lens}, simulated with {film} film look and {light} lighting.")
+def generate_midjourney(camera, lens, film, light, mj_version, aspect_ratio, quality, stylize, seed, base_scene, num):
+    prompts = []
+    vars = random.sample(scene_variations, min(len(scene_variations), num))
+    for i in range(num):
+        scene = f"{base_scene}, {vars[i]}"
+        mj_args = f"--ar {aspect_ratio} --q {quality} --s {stylize}"
+        if seed: mj_args += f" --seed {seed}"
+        prompts.append(f"{scene}, shot with {camera} and {lens}, using {film} film, under {light} {mj_args} --v {mj_version}")
     return "\n".join(prompts), prompts
 
+def generate_sora(camera, lens, film, light, base_scene):
+    var = random.choice(scene_variations)
+    scene = f"{base_scene}, {var}"
+    prompt = f"A detailed scene of {scene}, captured using a {camera} and {lens}, simulated with {film} film look and {light} lighting."
+    return prompt, [prompt]
+
 def export_txt(prompts):
-    with open("prompts.txt","w") as f: f.write("\n".join(prompts))
+    with open("prompts.txt", "w") as f:
+        f.write("\n".join(prompts))
     return gr.File.update(value="prompts.txt")
 
 def export_csv(prompts):
-    df=pd.DataFrame({"Prompt":prompts})
-    df.to_csv("prompts.csv",index=False)
+    df = pd.DataFrame({"Prompt": prompts})
+    df.to_csv("prompts.csv", index=False)
     return gr.File.update(value="prompts.csv")
 
 with gr.Blocks() as demo:
     gr.Markdown("## TischEins.org | Midjourney Prompt Engine\nPro Studio Prompt Generator for Midjourney and Sora")
 
+    preset = gr.Dropdown(choices=list(presets.keys()), value=list(presets.keys())[0], label="Select Preset")
+    camera = gr.Dropdown(choices=list(camera_lens_map.keys()), value=presets[list(presets.keys())[0]]["camera"], label="Camera Model")
+    lens = gr.Dropdown(choices=camera_lens_map[camera.value], value=presets[list(presets.keys())[0]]["lens"], label="Lens Model")
+    film = gr.Dropdown(choices=camera_film_map[camera.value], value=presets[list(presets.keys())[0]]["film"], label="Color Film Type")
+    light = gr.Dropdown(choices=light_conditions, value=presets[list(presets.keys())[0]]["light"], label="Light Condition")
+
     with gr.Tab("Midjourney"):
-        camera=gr.Dropdown(list(camera_lens_map.keys()),label="Camera Model")
-        lens=gr.Dropdown([],label="Lens Model")
-        film=gr.Dropdown([],label="Color Film Type")
-        light=gr.Dropdown(light_conditions,label="Light Condition")
-        mjv=gr.Dropdown(mj_versions,label="Midjourney Version",value="7")
-        ar=gr.Textbox(label="Aspect Ratio",value="5:4")
-        q=gr.Textbox(label="Quality",value="2")
-        s=gr.Textbox(label="Stylize",value="0")
-        seed=gr.Textbox(label="Seed (optional)")
-        scene=gr.Textbox(label="Scene Description")
-        num=gr.Slider(1,50,5,label="Number of Prompts")
-        out_mid, state_mid = gr.Textbox(lines=10), gr.State([])
-        btn_mid=gr.Button("Generate")
-        btn_mid.click(generate_midjourney, inputs=[camera,lens,film,light,mjv,ar,q,s,seed,scene,num], outputs=[out_mid,state_mid])
-        txt_mid=gr.Button("Export TXT"); csv_mid=gr.Button("Export CSV")
-        dl_txt_mid=gr.File(visible=False); dl_csv_mid=gr.File(visible=False)
-        txt_mid.click(export_txt, inputs=state_mid, outputs=dl_txt_mid)
-        csv_mid.click(export_csv, inputs=state_mid, outputs=dl_csv_mid)
-        camera.change(update_lens, inputs=camera, outputs=lens)
-        camera.change(update_film, inputs=camera, outputs=film)
+        mj_version = gr.Dropdown(choices=mj_versions, value=presets[list(presets.keys())[0]]["version"], label="Midjourney Version")
+        aspect_ratio = gr.Textbox(label="Aspect Ratio", value="5:4")
+        quality = gr.Textbox(label="Quality", value="2")
+        stylize = gr.Textbox(label="Stylize", value="0")
+        seed = gr.Textbox(label="Seed (optional)")
+        num = gr.Slider(1,50,value=3,label="Number of Prompts")
+        out_mid = gr.Textbox(lines=10, label="Generated Prompts")
+        state_mid = gr.State([])
+
+        btn_mid = gr.Button("Generate")
+        btn_mid.click(fn=generate_midjourney,
+                      inputs=[camera, lens, film, light, mj_version, aspect_ratio, quality, stylize, seed, presets[preset.value]["scene"], num],
+                      outputs=[out_mid, state_mid])
+        txt_mid = gr.Button("Export TXT"); csv_mid = gr.Button("Export CSV")
+        dl_txt_mid = gr.File(visible=False); dl_csv_mid = gr.File(visible=False)
+        txt_mid.click(fn=export_txt, inputs=state_mid, outputs=dl_txt_mid)
+        csv_mid.click(fn=export_csv, inputs=state_mid, outputs=dl_csv_mid)
 
     with gr.Tab("ChatGPT Sora"):
-        camera_s=gr.Dropdown(list(camera_lens_map.keys()),label="Camera Model")
-        lens_s=gr.Dropdown([],label="Lens Model")
-        film_s=gr.Dropdown([],label="Color Film Type")
-        light_s=gr.Dropdown(light_conditions,label="Light Condition")
-        scene_s=gr.Textbox(label="Scene Description")
-        num_s=gr.Slider(1,50,5,label="Number of Prompts")
-        out_s, state_s=gr.Textbox(lines=10), gr.State([])
-        btn_s=gr.Button("Generate")
-        btn_s.click(generate_sora, inputs=[camera_s,lens_s,film_s,light_s,scene_s,num_s], outputs=[out_s,state_s])
-        txt_s=gr.Button("Export TXT"); csv_s=gr.Button("Export CSV")
-        dl_txt_s=gr.File(visible=False); dl_csv_s=gr.File(visible=False)
-        txt_s.click(export_txt, inputs=state_s, outputs=dl_txt_s)
-        csv_s.click(export_csv, inputs=state_s, outputs=dl_csv_s)
-        camera_s.change(update_lens, inputs=camera_s, outputs=lens_s)
-        camera_s.change(update_film, inputs=camera_s, outputs=film_s)
+        out_s = gr.Textbox(lines=10, label="Generated Prompt")
+        state_s = gr.State([])
+        btn_s = gr.Button("Generate")
+        btn_s.click(fn=generate_sora,
+                    inputs=[camera, lens, film, light, presets[preset.value]["scene"]],
+                    outputs=[out_s, state_s])
+        txt_s = gr.Button("Export TXT"); csv_s = gr.Button("Export CSV")
+        dl_txt_s = gr.File(visible=False); dl_csv_s = gr.File(visible=False)
+        txt_s.click(fn=export_txt, inputs=state_s, outputs=dl_txt_s)
+        csv_s.click(fn=export_csv, inputs=state_s, outputs=dl_csv_s)
+
+    camera.change(fn=update_lens, inputs=camera, outputs=lens)
+    camera.change(fn=update_film, inputs=camera, outputs=film)
+    preset.change(fn=apply_preset, inputs=preset, outputs=[camera, lens, film, light, None, mj_version])
 
     gr.Markdown('<a href="https://tischeins.org">Pro Studio Mode by TischEins.org</a>')
 

presets.json

@@ -1,34 +1,34 @@
 {
-    "Sunset Portrait": {
-        "camera": "Canon EOS R5",
-        "lens": "Canon RF 50mm f/1.2L USM",
-        "film": "Kodak Portra 400",
-        "light": "Golden Hour",
-        "scene": "A woman standing in a field of tall grass during sunset, soft wind in her hair",
-        "version": "7"
-    },
-    "Urban Streetstyle": {
-        "camera": "Sony Alpha 7R V",
-        "lens": "Sony FE 85mm f/1.4 GM",
-        "film": "Kodak Ultramax 400",
-        "light": "Backlighting",
-        "scene": "Young man walking through a neon-lit city street at night with glowing billboards",
-        "version": "6.1"
-    },
-    "Forest Fantasy": {
-        "camera": "Fujifilm GFX 100 II",
-        "lens": "Fujifilm GF 80mm f/1.7 R WR",
-        "film": "Fujifilm Provia 100F",
-        "light": "Diffused Light",
-        "scene": "A mysterious elf walking through a foggy forest with glowing mushrooms",
-        "version": "6"
-    },
-    "Classic Black and White": {
-        "camera": "Leica M11",
-        "lens": "Leica Summilux-M 50mm f/1.4 ASPH",
-        "film": "Ilford HP5 Plus",
-        "light": "Soft Ambient Light",
-        "scene": "Man reading newspaper in a Parisian caf\u00e9, 1950s",
-        "version": "7"
-    }
+  "Sunset Portrait": {
+    "camera": "Canon EOS R5",
+    "lens": "Canon RF 50mm f/1.2L USM",
+    "film": "Kodak Portra 400",
+    "light": "Golden Hour",
+    "scene": "A woman standing in a field of tall grass during sunset, soft wind in her hair",
+    "version": "7"
+  },
+  "Urban Streetstyle": {
+    "camera": "Sony Alpha 7R V",
+    "lens": "Sony FE 85mm f/1.4 GM",
+    "film": "Kodak Ultramax 400",
+    "light": "Backlighting",
+    "scene": "Young man walking through a neon-lit city street at night with glowing billboards",
+    "version": "6.1"
+  },
+  "Forest Fantasy": {
+    "camera": "Fujifilm GFX 100 II",
+    "lens": "Fujifilm GF 80mm f/1.7 R WR",
+    "film": "Fujifilm Provia 100F",
+    "light": "Diffused Light",
+    "scene": "A mysterious elf walking through a foggy forest with glowing mushrooms",
+    "version": "6"
+  },
+  "Classic Black and White": {
+    "camera": "Leica M11",
+    "lens": "Leica Summilux-M 50mm f/1.4 ASPH",
+    "film": "Ilford HP5 Plus",
+    "light": "Soft Ambient Light",
+    "scene": "Man reading newspaper in a Parisian caf\u00e9, 1950s",
+    "version": "7"
+  }
 }