Upload 2 files

app.py

@@ -1,209 +1,95 @@
 
 import gradio as gr
-import random
 import pandas as pd
-import json
+import random
 
+# 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"
-  ],
-  "Sony Alpha 7R V": [
-    "Sony FE 85mm f/1.4 GM",
-    "Sony FE 24-70mm f/2.8 GM II",
-    "Sony FE 50mm f/1.2 GM"
-  ],
-  "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"
-  ],
-  "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"
-  ]
+    "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
 }
 camera_film_map = {
-  "Canon EOS R5": [
-    "Kodak Portra 400",
-    "Kodak Ektar 100",
-    "Kodak Gold 200"
-  ],
-  "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"
-  ]
+    "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
 }
 light_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"
+    # 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"
 ]
-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"
-  }
-}
-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(camera):
-    return gr.update(choices=camera_lens_map[camera], value=camera_lens_map[camera][0])
+mj_versions = ["6","6.1","7"]
 
-def update_film(camera):
-    return gr.update(choices=camera_film_map[camera], value=camera_film_map[camera][0])
+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"],
-        camera_lens_map[pre["camera"]][0],
-        camera_film_map[pre["camera"]][0],
-        pre["light"],
-        pre["scene"],
-        pre["version"]
-    )
-
-def generate_prompts(engine, mj_version, camera, lens, film, light, aspect_ratio, quality, stylize, seed, base_scene, num):
-    prompts = []
-    # 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:
-            prompt = 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,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 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.")
+    return "\n".join(prompts), prompts
+
 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)
+    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)
-    return gr.File.update(value="prompts.csv", visible=True)
+    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 <br> Pro Studio Prompt Generator for Midjourney and Sora ")
-
-    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])
+    gr.Markdown("## TischEins.org | Midjourney Prompt Engine\nPro Studio Prompt Generator for Midjourney and Sora")
 
-    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])
+    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)
 
-    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)
+    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)
 
-    gr.Markdown('<a href="https://tischeins.org">Pro Studio Prompt Generator by TischEins.org</a>')
+    gr.Markdown('<a href="https://tischeins.org">Pro Studio Mode by TischEins.org</a>')
 
 demo.launch()