Upload 3 files

app.py

@@ -1,5 +1,6 @@
 
 import gradio as gr
+import pandas as pd
 
 camera_lens_map = {
   "Canon EOS R5": [
@@ -103,6 +104,108 @@ camera_lens_map = {
     "Fujinon XF 56mm f/1.2 R APD"
   ]
 }
+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"
+  ],
+  "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"
+  ],
+  "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 = [
   "Golden Hour",
   "Blue Hour",
@@ -130,6 +233,9 @@ mj_versions = ["6", "6.1", "7"]
 def update_lens_dropdown(camera):
     return gr.update(choices=camera_lens_map[camera], value=camera_lens_map[camera][0])
 
+def update_film_dropdown(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
@@ -145,7 +251,17 @@ def generate_prompt(engine, mj_version, camera, lens, film, light, aspect_ratio,
         else:
             prompt = f"A detailed scene of {scene}, captured using a {camera} and {lens}, simulated with {film} film look and {light} lighting."
         prompts.append(prompt)
-    return "\n".join(prompts)
+    return "\n".join(prompts), prompts
+
+def export_to_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})
+    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")
@@ -167,14 +283,25 @@ with gr.Blocks() as demo:
     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)
 
     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)
+    ], 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>')