PromptSuperHeroImageGenerator

Runtime error

App Files Files Community

awacke1 commited on Aug 26, 2023

Commit

e4f2a60

1 Parent(s): edc20ce

Update app.py

Browse files

Files changed (1) hide show

app.py +40 -125

app.py CHANGED Viewed

@@ -1,143 +1,58 @@
 import gradio as gr
-import os
-import sys
 from pathlib import Path
 def load_models_from_file(filename):
     with open(filename, 'r') as f:
         return [line.strip() for line in f]
-if __name__ == "__main__":
-    models = load_models_from_file('models.txt')
-    print(models)
-    #removed to removed.txt
 current_model = models[0]
-text_gen1=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion_link")
 models2 = [gr.Interface.load(f"models/{model}", live=True, preprocess=False) for model in models]
-def text_it1(inputs,text_gen1=text_gen1):
-        go_t1=text_gen1(inputs)
-        return(go_t1)
-def set_model(current_model):
-    current_model = models[current_model]
-    return gr.update(label=(f"{current_model}"))
 def send_it1(inputs, model_choice):
-        proc1=models2[model_choice]
-        output1=proc1(inputs)
-        return(output1)
-css=""""""
-with gr.Blocks(css=css) as myface:
-    gr.HTML("""<!DOCTYPE html>
-<html lang="en">
-  <head>
-    <meta charset="utf-8" />
-    <meta name="twitter:card" content="player"/>
-    <meta name="twitter:site" content=""/>
-    <meta name="twitter:player" content="https://omnibus-maximum-multiplier-places.hf.space"/>
-    <meta name="twitter:player:stream" content="https://omnibus-maximum-multiplier-places.hf.space"/>
-    <meta name="twitter:player:width" content="100%"/>
-    <meta name="twitter:player:height" content="600"/>
-    <meta property="og:title" content="Embedded Live Viewer"/>
-    <meta property="og:description" content="Tweet Genie - A Huggingface Space"/>
-    <meta property="og:image" content="https://cdn.glitch.global/80dbe92e-ce75-44af-84d5-74a2e21e9e55/omnicard.png?v=1676772531627"/>
-    <!--<meta http-equiv="refresh" content="0; url=https://huggingface.co/spaces/corbt/tweet-genie">-->
-  </head>
-</html>
-""")
-    with gr.Row():
-        with gr.Tab("Title"):
-                gr.HTML("""    <title>Minimum Multiplier</title><div style="text-align: center; max-width: 1500px; margin: 0 auto;">
-                <h1>Fill the Textbox at the top and click Generate Image</h1>
-                <br><h4>The first time you load a model it takes 200 seconds</h4>
-                <br><h4>But after it loads each image takes 20 seconds to generate!</h4>
-                                            """)
-        with gr.Tab("Description"):
-            gr.HTML("""<div style="text-align:center;">
-            <h4>As many Text-to-Image Models as I can fit here</h4><br>
-            <h4>Suggest more up in the "Community" button</h4>
-                       </div>""")
-        with gr.Tab("Tools"):
-                    with gr.Tab("View"):
-                      with gr.Row():
-                        with gr.Column(style="width=50%, height=70%"):
-                                gr.Pil(label="Crop")
-                        with gr.Column(style="width=50%, height=70%"):
-                                gr.Pil(label="Crop")
-                    with gr.Tab("Draw"):
-                        with gr.Column(style="width=50%, height=70%"):
-                                gr.Pil(label="Crop")
-                        with gr.Column(style="width=50%, height=70%"):
-                                gr.Pil(label="Draw")
-                                gr.ImagePaint(label="Draw")
-                    with gr.Tab("Text"):
-                        with gr.Row():
-                            with gr.Column(scale=50):
-                                gr.Textbox(label="", lines=8, interactive=True)
-                            with gr.Column(scale=50):
-                                gr.Textbox(label="", lines=8, interactive=True)
-                    with gr.Tab("Color Picker"):
-                        with gr.Row():
-                            with gr.Column(scale=50):
-                                gr.ColorPicker(label="Color", interactive=True)
-                            with gr.Column(scale=50):
-                                gr.ImagePaint(label="Draw", interactive=True)
-    with gr.Row():
-        with gr.Column(scale=100):
-            magic1=gr.Textbox(lines=4)
-            run=gr.Button("Generate Image")
-    with gr.Row():
-        with gr.Column(scale=100):
-            #Model selection dropdown
-            model_name1 = gr.Dropdown(label="Select Model", choices=[m for m in models], type="index", value=current_model, interactive=True)
-    with gr.Row():
-        with gr.Column(style="width=800px"):
-            output1=gr.Image(label=(f"{current_model}"))
-    with gr.Row():
-        with gr.Column(scale=50):
-            input_text=gr.Textbox(label="Prompt Idea",lines=2)
-            use_short=gr.Button("Use Short Prompt")
-            see_prompts=gr.Button("Extend Idea")
-    def short_prompt(inputs):
-        return(inputs)
-    model_name1.change(set_model,inputs=model_name1,outputs=[output1])
-    run.click(send_it1, inputs=[magic1, model_name1], outputs=[output1])
-    use_short.click(short_prompt,inputs=[input_text],outputs=magic1)
-    see_prompts.click(text_it1,inputs=[input_text],outputs=magic1)
-myface.queue(concurrency_count=200)
-myface.launch(inline=True, show_api=False, max_threads=400)

 import gradio as gr
 from pathlib import Path
+import os
+# Load models from file
 def load_models_from_file(filename):
     with open(filename, 'r') as f:
         return [line.strip() for line in f]
+# Initialize model
+models = load_models_from_file('models.txt')
 current_model = models[0]
+# Load Interfaces
+text_gen1 = gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion_link")
 models2 = [gr.Interface.load(f"models/{model}", live=True, preprocess=False) for model in models]
+# Function to generate text
+def text_it1(inputs):
+    go_t1 = text_gen1(inputs)
+    return go_t1
+# Function to set model
+def set_model(model_choice):
+    current_model = models[model_choice]
+    return f"{current_model}"
+# Function to send input
 def send_it1(inputs, model_choice):
+    proc1 = models2[model_choice]
+    output1 = proc1(inputs)
+    # Save image and prompt to file
+    prompt_safe_name = "".join(e for e in inputs if e.isalnum())
+    image_path = f"images/{prompt_safe_name}.png"
+    output1.save(image_path)
+    with open('prompts.txt', 'a') as f:
+        f.write(f"{inputs}\n{image_path}\n")
+    return output1
+# Initialize Gradio Interface
+with gr.Blocks() as myface:
+    with gr.Row():
+        gr.Textbox(lines=4, label="Prompt").store("magic1")
+        gr.Dropdown(label="Select Model", choices=models, type="index", value=0, interactive=True).store("model_name1")
+        gr.Button("Generate Image").click(send_it1, inputs=["magic1", "model_name1"]).outputs("output1")
+        gr.Image(label="Image Output").store("output1")
+        gr.Textbox(label="Prompt Idea", lines=2).store("input_text")
+        gr.Button("Use Short Prompt").click(text_it1, inputs=["input_text"]).outputs("magic1")
+    # Sidebar to show saved prompts
+    with open('prompts.txt', 'r') as f:
+        saved_prompts = f.readlines()
+    gr.Textbox(value=saved_prompts, lines=20, label="Saved Prompts", interactive=False, sidetab=True)
+myface.launch()