Upload 3 files

app.py

@@ -1 +1,97 @@
-# Placeholder
+
+import gradio as gr
+import pandas as pd
+import random
+import json
+
+with open("presets.json") as f:
+    presets = json.load(f)
+
+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"]
+}
+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 Provia 100F", "Fujifilm Velvia 50", "Fujifilm Pro 400H"],
+    "Leica M11": ["Ilford HP5 Plus", "Kodak T-Max 100", "AgfaPhoto APX 100"]
+}
+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"
+]
+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"
+]
+
+def update_lens(camera):
+    return gr.update(choices=camera_lens_map.get(camera, []), value=camera_lens_map.get(camera, [None])[0])
+
+def update_film(camera):
+    return gr.update(choices=camera_film_map.get(camera, []), value=camera_film_map.get(camera, [None])[0])
+
+def generate_mid(img_camera, img_lens, img_film, img_light, mj_version, ar, q, s, seed, scene, num):
+    prompts=[]
+    varsamples = random.sample(scene_variations, num)
+    for i in range(num):
+        sc = f"{scene}, {varsamples[i]}"
+        args = f"--ar {ar} --q {q} --s {s}" + (f" --seed {seed}" if seed else "")
+        prompts.append(f"{sc}, shot with {img_camera} and {img_lens}, using {img_film} film, under {img_light} {args} --v {mj_version}")
+    return "\n".join(prompts), prompts
+
+def generate_sora(img_camera, img_lens, img_film, img_light, scene):
+    var = random.choice(scene_variations)
+    sc = f"{scene}, {var}"
+    prompt = f"A detailed scene of {sc}, captured using a {img_camera} and {img_lens}, simulated with {img_film} film look and {img_light} lighting."
+    return prompt, [prompt]
+
+def export_txt(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)
+    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")
+    with gr.Tab("Midjourney"):
+        preset = gr.Dropdown(list(presets.keys()), label="Select Preset")
+        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(["6","6.1","7"], 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,5,3,label="Number of Prompts")
+        out, state = gr.Textbox(lines=5), gr.State([])
+        btn = gr.Button("Generate")
+        btn.click(generate_mid, [camera, lens, film, light, mjv, ar, q, s, seed, scene, num], [out, state])
+        txtb = gr.Button("Export TXT"); csvb = gr.Button("Export CSV")
+        dtxt = gr.File(visible=False); dcsv = gr.File(visible=False)
+        txtb.click(export_txt, state, dtxt); csvb.click(export_csv, state, dcsv)
+        camera.change(update_lens, camera, lens); camera.change(update_film, camera, film)
+    with gr.Tab("ChatGPT Sora"):
+        scene_s = gr.Textbox(label="Scene Description")
+        out_s, state_s = gr.Textbox(lines=5), gr.State([])
+        btn_s = gr.Button("Generate")
+        btn_s.click(generate_sora, [camera, lens, film, light, scene_s], [out_s, state_s])
+        txt_s = gr.Button("Export TXT"); csv_s = gr.Button("Export CSV")
+        dtxt_s = gr.File(visible=False); dcsv_s = gr.File(visible=False)
+        txt_s.click(export_txt, state_s, dtxt_s); csv_s.click(export_csv, state_s, dcsv_s)
+        camera.change(update_lens, camera, lens); camera.change(update_film, camera, film)
+    gr.Markdown('<a href="https://tischeins.org">Pro Studio Mode by TischEins.org</a>')
+demo.launch()

requirements.txt

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