Upload 3 files

app.py

@@ -229,6 +229,40 @@ light_conditions = [
   "Reflected Light"
 ]
 mj_versions = ["6", "6.1", "7"]
+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_dropdown(camera):
     return gr.update(choices=camera_lens_map[camera], value=camera_lens_map[camera][0])
@@ -240,6 +274,17 @@ def toggle_visibility(engine):
     visible = engine == "Midjourney"
     return [gr.update(visible=visible)] * 4
 
+def apply_preset(preset_name):
+    preset = presets[preset_name]
+    return (
+        preset["camera"],
+        camera_lens_map[preset["camera"]][0],
+        camera_film_map[preset["camera"]][0],
+        preset["light"],
+        preset["scene"],
+        preset["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:
@@ -266,12 +311,14 @@ def export_to_csv(prompts):
 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")
-    lens_dropdown = gr.Dropdown(choices=[], label="Lens Model")
-    film_dropdown = gr.Dropdown(choices=[], label="Color Film Type")
+    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)
@@ -291,9 +338,14 @@ with gr.Blocks() as demo:
     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])
+    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,