Upload 2 files

app.py

@@ -1,6 +1,8 @@
 
 import gradio as gr
+import random
 import pandas as pd
+import json
 
 camera_lens_map = {
   "Canon EOS R5": [
@@ -8,11 +10,6 @@ camera_lens_map = {
     "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"
-  ],
   "Sony Alpha 7R V": [
     "Sony FE 85mm f/1.4 GM",
     "Sony FE 24-70mm f/2.8 GM II",
@@ -27,81 +24,6 @@ camera_lens_map = {
     "Leica Summilux-M 50mm f/1.4 ASPH",
     "Leica APO-Summicron-M 75mm f/2 ASPH",
     "Leica Elmarit-M 28mm f/2.8 ASPH"
-  ],
-  "Panasonic Lumix S5II": [
-    "Panasonic Lumix S Pro 50mm f/1.4",
-    "Sigma 35mm f/1.4 DG DN",
-    "Panasonic Lumix S 85mm f/1.8"
-  ],
-  "Olympus OM-D E-M1 Mark III": [
-    "Olympus M.Zuiko Digital ED 45mm f/1.2 PRO",
-    "Olympus M.Zuiko Digital ED 25mm f/1.2 PRO",
-    "Olympus M.Zuiko Digital ED 12-40mm f/2.8 PRO"
-  ],
-  "Pentax K-3 Mark III": [
-    "HD Pentax-D FA* 50mm f/1.4 SDM AW",
-    "HD Pentax-DA 16-85mm f/3.5-5.6 ED DC WR",
-    "HD Pentax-D FA 70-210mm f/4 ED SDM WR"
-  ],
-  "Canon EOS 5D Mark IV": [
-    "Canon EF 24-70mm f/2.8L II USM",
-    "Canon EF 70-200mm f/2.8L IS III USM",
-    "Canon EF 50mm f/1.2L USM"
-  ],
-  "Nikon D850": [
-    "Nikon AF-S NIKKOR 24-70mm f/2.8E ED VR",
-    "Nikon AF-S NIKKOR 105mm f/1.4E ED",
-    "Nikon AF-S 50mm f/1.8G"
-  ],
-  "Sony A7 III": [
-    "Sony FE 35mm f/1.4 GM",
-    "Sony FE 70-200mm f/2.8 GM OSS",
-    "Tamron 28-75mm f/2.8 Di III RXD"
-  ],
-  "Fujifilm X-T5": [
-    "Fujinon XF 23mm f/1.4 R LM WR",
-    "Fujinon XF 56mm f/1.2 R WR",
-    "Fujinon XF 16-55mm f/2.8 R LM WR"
-  ],
-  "Leica Q2": [
-    "Leica Summilux 28mm f/1.7 ASPH (fixed)",
-    "Leica APO-Summicron 90mm f/2 ASPH",
-    "Leica Elmarit-M 21mm f/2.8 ASPH"
-  ],
-  "Panasonic GH6": [
-    "Panasonic Leica DG 25mm f/1.4 II ASPH",
-    "Olympus M.Zuiko ED 75mm f/1.8",
-    "Panasonic Lumix 42.5mm f/1.7"
-  ],
-  "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 645Z": [
-    "Pentax D FA 645 55mm f/2.8 AL",
-    "Pentax FA 645 150mm f/2.8",
-    "Pentax FA 645 75mm f/2.8"
-  ],
-  "Canon EOS R6 Mark II": [
-    "Canon RF 24-105mm f/4L IS USM",
-    "Canon RF 85mm f/2 Macro IS STM",
-    "Canon RF 70-200mm f/2.8L IS USM"
-  ],
-  "Nikon Z9": [
-    "Nikon Z 24-120mm f/4 S",
-    "Nikon Z 400mm f/2.8 TC VR S",
-    "Nikon Z 70-200mm f/2.8 VR S"
-  ],
-  "Sony FX3": [
-    "Sony FE 24mm f/1.4 GM",
-    "Sony FE PZ 28-135mm f/4 G OSS",
-    "Sigma 85mm f/1.4 DG DN Art"
-  ],
-  "Fujifilm X-H2": [
-    "Fujinon XF 18mm f/1.4 R LM WR",
-    "Fujinon XF 33mm f/1.4 R LM WR",
-    "Fujinon XF 56mm f/1.2 R APD"
   ]
 }
 camera_film_map = {
@@ -110,11 +32,6 @@ camera_film_map = {
     "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",
@@ -129,81 +46,6 @@ camera_film_map = {
     "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"
   ]
 }
 light_conditions = [
@@ -263,34 +105,54 @@ presets = {
     "version": "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"
+]
 
-def update_lens_dropdown(camera):
+def update_lens(camera):
     return gr.update(choices=camera_lens_map[camera], value=camera_lens_map[camera][0])
 
-def update_film_dropdown(camera):
+def update_film(camera):
     return gr.update(choices=camera_film_map[camera], value=camera_film_map[camera][0])
 
-def toggle_visibility(engine):
-    visible = engine == "Midjourney"
-    return [gr.update(visible=visible)] * 4
-
-def apply_preset(preset_name):
-    preset = presets[preset_name]
+def apply_preset(p):
+    pre = presets[p]
     return (
-        preset["camera"],
-        camera_lens_map[preset["camera"]][0],
-        camera_film_map[preset["camera"]][0],
-        preset["light"],
-        preset["scene"],
-        preset["version"]
+        pre["camera"],
+        camera_lens_map[pre["camera"]][0],
+        camera_film_map[pre["camera"]][0],
+        pre["light"],
+        pre["scene"],
+        pre["version"]
     )
 
-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}"
+def generate_prompts(engine, mj_version, camera, lens, film, light, aspect_ratio, quality, stylize, seed, base_scene, num):
     prompts = []
-    for _ in range(num):
+    # choose variations
+    vars = random.sample(scene_variations, min(num, len(scene_variations)))
+    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}"
         if engine == "Midjourney":
             prompt = f"{scene}, shot with {camera} and {lens}, using {film} film, under {light} {mj_args} --v {mj_version}"
         else:
@@ -298,63 +160,50 @@ def generate_prompt(engine, mj_version, camera, lens, film, light, aspect_ratio,
         prompts.append(prompt)
     return "\n".join(prompts), prompts
 
-def export_to_txt(prompts):
-    with open("prompts.txt", "w") as f:
+def export_txt(prompts):
+    with open("prompts.txt","w") as f:
         f.write("\n".join(prompts))
     return gr.File.update(value="prompts.txt", visible=True)
 
-def export_to_csv(prompts):
-    df = pd.DataFrame({"Prompt": prompts})
+def export_csv(prompts):
+    df = pd.DataFrame({"Prompt":prompts})
     df.to_csv("prompts.csv", index=False)
     return gr.File.update(value="prompts.csv", visible=True)
 
 with gr.Blocks() as demo:
     gr.Markdown("# TischEins.org | Midjourney Prompt Engine")
 
-    preset_selector = gr.Dropdown(choices=['Sunset Portrait', 'Urban Streetstyle', 'Forest Fantasy', 'Classic Black and White'], label="Select Preset")
-
-    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_lens_map.keys()), label="Camera Model", value=list(camera_lens_map.keys())[0])
-    lens_dropdown = gr.Dropdown(choices=camera_lens_map[list(camera_lens_map.keys())[0]], label="Lens Model")
-    film_dropdown = gr.Dropdown(choices=camera_film_map[list(camera_film_map.keys())[0]], label="Color Film Type")
-    light_dropdown = gr.Dropdown(choices=light_conditions, label="Light Condition")
-
-    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)")
-
-    scene = gr.Textbox(label="Scene Description", placeholder="Describe your scene here...")
-    num = gr.Slider(1, 50, value=5, label="Number of Prompts")
-
-    output = gr.Textbox(label="Generated Prompts", lines=20)
-    prompts_state = gr.State([])
-
-    generate_btn = gr.Button("Generate Prompts")
-    export_txt_btn = gr.Button("Export as TXT")
-    export_csv_btn = gr.Button("Export as CSV")
-    download_txt = gr.File(visible=False)
-    download_csv = gr.File(visible=False)
-
-    preset_selector.change(fn=apply_preset, inputs=preset_selector, outputs=[
-        camera_dropdown, lens_dropdown, film_dropdown, light_dropdown, scene, mj_version_dropdown
-    ])
-    camera_dropdown.change(fn=update_lens_dropdown, inputs=camera_dropdown, outputs=lens_dropdown)
-    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_dropdown, film_dropdown,
-        light_dropdown, aspect_ratio, quality, stylize, seed, scene, num
-    ], outputs=[output, prompts_state])
-
-    export_txt_btn.click(fn=export_to_txt, inputs=prompts_state, outputs=download_txt)
-    export_csv_btn.click(fn=export_to_csv, inputs=prompts_state, outputs=download_csv)
-
-    gr.Markdown('<a href="https://tischeins.org" target="_blank">Pro Studio Mode by TischEins.org</a>')
+    psel = gr.Dropdown(choices=list(presets.keys()), label="Select Preset")
+    eng = gr.Radio(["Midjourney","ChatGPT Sora"], label="Engine", value="Midjourney")
+    mjv = gr.Dropdown(choices=mj_versions, label="MJ Version", value="7", visible=True)
+    cam = gr.Dropdown(choices=list(camera_lens_map.keys()), label="Camera Model")
+    len_ = gr.Dropdown(choices=[], label="Lens Model")
+    film = gr.Dropdown(choices=[], label="Color Film Type")
+    light = gr.Dropdown(choices=light_conditions, label="Light Condition")
+    ar = gr.Textbox(label="Aspect Ratio", value="5:4", visible=True)
+    q = gr.Textbox(label="Quality", value="2", visible=True)
+    s = gr.Textbox(label="Stylize", value="0", visible=True)
+    seed = gr.Textbox(label="Seed", placeholder="optional")
+    base_scene = gr.Textbox(label="Scene Description")
+    num = gr.Slider(1,50,value=3,label="Number of Prompts")
+    out = gr.Textbox(label="Prompts", lines=20)
+    state = gr.State([])
+
+    psel.change(fn=apply_preset, inputs=psel, outputs=[cam,len_,film,light,base_scene,mjv])
+    cam.change(fn=update_lens, inputs=cam, outputs=len_)
+    cam.change(fn=update_film, inputs=cam, outputs=film)
+    eng.change(lambda e: [gr.update(visible=e=="Midjourney")]*3, inputs=eng, outputs=[mjv,ar,q,s])
+
+    btn = gr.Button("Generate")
+    btn.click(fn=generate_prompts, inputs=[eng,mjv,cam,len_,film,light,ar,q,s,seed,base_scene,num], outputs=[out,state])
+
+    txt_btn = gr.Button("Export TXT")
+    csv_btn = gr.Button("Export CSV")
+    dl_txt = gr.File(visible=False)
+    dl_csv = gr.File(visible=False)
+    txt_btn.click(fn=export_txt, inputs=state, outputs=dl_txt)
+    csv_btn.click(fn=export_csv, inputs=state, outputs=dl_csv)
+
+    gr.Markdown('<a href="https://tischeins.org">Pro Studio Mode by TischEins.org</a>')
 
 demo.launch()

requirements.txt

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