Spaces:

charliebaby2023
/

sd_pony_variants_grid

Running

App Files Files Community

charliebaby2023 commited on Jan 28

Commit

f1c6c08

verified ·

1 Parent(s): 2c694c9

Update app.py

Browse files

Files changed (1) hide show

app.py +27 -99

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from concurrent.futures import ThreadPoolExecutor
 import requests
 now2 = 0
-kii=" mohawk femboy racecar driver ";
 def get_current_time():
@@ -16,6 +16,7 @@ def get_current_time():
     ki = f'{kii} {current_time}'
     return ki
 def load_fn(models):
     global models_load
     models_load = {}
@@ -35,16 +36,10 @@ num_models = len(models)
 default_models = models[:num_models]
 def extend_choices(choices):
     return choices + (num_models - len(choices)) * ['NA']
 def update_imgbox(choices):
     choices_plus = extend_choices(choices)
     return [gr.Image(None, label=m, visible=(m != 'NA')) for m in choices_plus]
@@ -52,6 +47,7 @@ def update_imgbox(choices):
 executor = ThreadPoolExecutor(max_workers=num_models)
 def gen_fn(model_str, prompt, negative_prompt):
     if model_str == 'NA':
         return None
@@ -60,23 +56,13 @@ def gen_fn(model_str, prompt, negative_prompt):
     combined_prompt = f'{prompt} {model_str} {negative_prompt} {noise}'
     print(f"Generating with prompt: {combined_prompt}")
-#    try:
-#        future = executor.submit(models_load[model_str], f'{prompt} {negative_prompt} {noise}')
-#        result = future.result()
-#        return result
-#    except requests.exceptions.Timeout:
-#        print(f"Timeout occurred for model {model_str}. Please try again later.")
-#        return None
-#    except Exception as e:
-#        print(f"Error occurred: {e}")
-#        return None
     try:
         # Attempt to generate the image
         image_response = models_load[model_str](f'{prompt} {negative_prompt} {noise}')
         # Check if the image_response is a tuple, handle accordingly
         if isinstance(image_response, tuple):
-            # If the response is a tuple, return the first item assuming it's the image
             image_response = image_response[0]
         # Ensure the response is an image or image-like object
@@ -91,71 +77,31 @@ def gen_fn(model_str, prompt, negative_prompt):
         print(f"Error occurred: {e}")
         return None
-   # noise = str(randint(0, 9999))
-   # combined_prompt = f'{prompt} {model_str} {negative_prompt} {noise}'
-   # print(f"Generating with prompt: {combined_prompt}")  # Debug line
-   # result = models_load[model_str](f'{prompt} {negative_prompt} {noise}')
-   # end_time = time.time()  # End timing
-   # runtime = end_time - start_time
-   # model_timings[model_str] = runtime  # Log the model's execution time
-   # queue_size -= 1  # Decrement queue size after processing
-   # return f"Model {model_str} ran for {runtime:.2f} seconds", result
-   # return models_load[model_str](f'{prompt} {negative_prompt} {noise}')
 def make_me():
-   # with gr.Tab('The Dream'):
-        with gr.Row():
-            txt_input = gr.Textbox(lines=2, value=kii )
-            #txt_input = gr.Textbox(label='Your prompt:', lines=2, value=kii)
-            negative_prompt_input = gr.Textbox(lines=2, value="", label="Negative Prompt" )
-            gen_button = gr.Button('Generate images')
-            stop_button = gr.Button('Stop', variant='secondary', interactive=False)
-            gen_button.click(lambda _: gr.update(interactive=True), None, stop_button)
-            gr.HTML("""
-            <div style="text-align: center; max-width: 100%; margin: 0 auto;">
-                <body>
-                </body>
-            </div>
-            """)
-        with gr.Row():
-            output = [gr.Image(label=m ) for m in default_models]
-            current_models = [gr.Textbox(m, visible=False) for m in default_models]
-            for m, o in zip(current_models, output):
-                gen_event = gen_button.click(gen_fn, [m, txt_input, negative_prompt_input], o, queue = False)
-          #      stop_button.click(lambda _: gr.update(interactive=False), None, stop_button, cancels=[gen_event])
-        with gr.Accordion('Model selection'):
-     #       model_choice = gr.CheckboxGroup(models, label=f' {num_models} different models selected', value=default_models, multiselect=True, max_choices=num_models, interactive=True, filterable=False)
-     #       model_choice.change(update_imgbox, model_choice, output)
-     #       model_choice.change(extend_choices, model_choice, current_models)
-            model_choice = gr.CheckboxGroup(models, label=f' {num_models} different models selected', value=default_models,  interactive=True )
-            model_choice.change(update_imgbox, model_choice, output)
-            model_choice.change(extend_choices, model_choice, current_models)
-            #   with gr.Row():
-     #       gr.HTML("""
-     #           <div class="footer">
-     #           <p> Based on the <a href="https://huggingface.co/spaces/derwahnsinn/TestGen">TestGen</a> Space by derwahnsinn, the <a href="https://huggingface.co/spaces/RdnUser77/SpacIO_v1">SpacIO</a> Space by RdnUser77 and Omnibus's Maximum Multiplier!
-     #           </p>
-     #       """)
 js_code = """
@@ -171,24 +117,6 @@ js_code = """
 </script>
 """
 with gr.Blocks(css="""
      label.float.svelte-i3tvor {  top:auto!important;  bottom: 0;    position: absolute;    background: rgba(0,0,0,0.0);    left: var(--block-label-margin);    color: rgba(200,200,200,.7);}
     .genbut { max-width: 50px; max-height: 30px; width:150px; height:30px}

 import requests
 now2 = 0
+kii=" mohawk femboy racecar driver "
 def get_current_time():
     ki = f'{kii} {current_time}'
     return ki
 def load_fn(models):
     global models_load
     models_load = {}
 default_models = models[:num_models]
 def extend_choices(choices):
     return choices + (num_models - len(choices)) * ['NA']
 def update_imgbox(choices):
     choices_plus = extend_choices(choices)
     return [gr.Image(None, label=m, visible=(m != 'NA')) for m in choices_plus]
 executor = ThreadPoolExecutor(max_workers=num_models)
 def gen_fn(model_str, prompt, negative_prompt):
     if model_str == 'NA':
         return None
     combined_prompt = f'{prompt} {model_str} {negative_prompt} {noise}'
     print(f"Generating with prompt: {combined_prompt}")
     try:
         # Attempt to generate the image
         image_response = models_load[model_str](f'{prompt} {negative_prompt} {noise}')
         # Check if the image_response is a tuple, handle accordingly
         if isinstance(image_response, tuple):
+            # If the response is a tuple, assume the first item is the image
             image_response = image_response[0]
         # Ensure the response is an image or image-like object
         print(f"Error occurred: {e}")
         return None
 def make_me():
+    with gr.Row():
+        txt_input = gr.Textbox(lines=2, value=kii)
+        negative_prompt_input = gr.Textbox(lines=2, value="", label="Negative Prompt")
+        gen_button = gr.Button('Generate images')
+        stop_button = gr.Button('Stop', variant='secondary', interactive=False)
+        gen_button.click(lambda _: gr.update(interactive=True), None, stop_button)
+        gr.HTML("""
+        <div style="text-align: center; max-width: 100%; margin: 0 auto;">
+            <body>
+            </body>
+        </div>
+        """)
+    with gr.Row():
+        output = [gr.Image(label=m) for m in default_models]
+        current_models = [gr.Textbox(m, visible=False) for m in default_models]
+        for m, o in zip(current_models, output):
+            gen_event = gen_button.click(gen_fn, [m, txt_input, negative_prompt_input], o, queue=False)
+    with gr.Accordion('Model selection'):
+        model_choice = gr.CheckboxGroup(models, label=f' {num_models} different models selected', value=default_models, interactive=True)
+        model_choice.change(update_imgbox, model_choice, output)
+        model_choice.change(extend_choices, model_choice, current_models)
 js_code = """
 </script>
 """
 with gr.Blocks(css="""
      label.float.svelte-i3tvor {  top:auto!important;  bottom: 0;    position: absolute;    background: rgba(0,0,0,0.0);    left: var(--block-label-margin);    color: rgba(200,200,200,.7);}
     .genbut { max-width: 50px; max-height: 30px; width:150px; height:30px}