Spaces:

charliebaby2023
/

Diffusion333380XX2

Paused

App Files Files Community

charliebaby2023 commited on Jan 22

Commit

edc0a4e

verified ·

1 Parent(s): f5d5867

Update app.py

Browse files

Files changed (1) hide show

app.py +53 -86

app.py CHANGED Viewed

@@ -1,108 +1,74 @@
 import gradio as gr
 from random import randint
 from datetime import datetime
-# Define manual grouping for models
-model_groups = {
-    "Group A": [
-        "Bakanayatsu/ponyDiffusion-V6-XL-Turbo-DPO",
-        "John6666/photo-realistic-pony-v5-sdxl",
-        "John6666/photo-realistic-pony-v5-sdxl",
-    ],
-    "Group B": [
-        "John6666/jib-mix-pony-realistic-v2-sdxl",
-        "John6666/3x3x3mixxl-v2-sdxl-spo",
-        "John6666/3x3x3mixxl-v2-sdxl",
-    ],
-    "Group C": [
-        "John6666/3x3x3mixxl-v2-sdxl-spo",
-        "John6666/3x3x3mixxl-v2-sdxl",
-        "John6666/titania-mix-realistic-pony-gbv10-sdxl",
-    ],
-    "Group D": [
-        "John6666/titania-mix-realistic-pony-gbv20-sdxl",
-        "John6666/titania-mix-realistic-pony-gbv30-sdxl",
-        "John6666/mala-anime-mix-nsfw-pony-xl-v3-sdxl",
-    ],
-}
-# Placeholder for models
-models_load = {model: None for group in model_groups.values() for model in group}
-# Function to simulate model loading
 def load_fn(models):
-    loaded_models = {}
     for model in models:
-        try:
-            # Simulate model loading
-            loaded_models[model] = lambda txt: f"Generated image for {txt}"
-        except Exception as error:
-            loaded_models[model] = lambda txt: None
-    return loaded_models
-models_load = load_fn(models_load.keys())
-# Function to simulate generation with tally tracking
-def gen_fn(model_str, prompt, tallies):
     if model_str == 'NA':
-        return None, tallies
     noise = str(randint(0, 9999))
     combined_prompt = f'{prompt} {model_str} {noise}'
-    print(f"Generating with prompt: {combined_prompt}")
-    try:
-        result = models_load[model_str](combined_prompt)
-        tallies[model_str] += 1
-        return result, tallies
-    except Exception as e:
-        print(f"Error generating for {model_str}: {e}")
-        return None, tallies
-# Build the Gradio interface
 def make_me():
     with gr.Row():
-        model_group_selector = gr.Dropdown(
-            choices=list(model_groups.keys()),
-            label="Select Model Group",
-            value="Group A",  # Default group
-        )
-        txt_input = gr.Textbox(label='Your prompt:', lines=2, value="mohawk femboy racecar driver")
         gen_button = gr.Button('Generate images', elem_id="generate-btn")
     with gr.Row():
-        # Define dynamic states
-        output = gr.State([])  # Placeholder for output image elements
-        tallies = gr.State({})  # Tally counter for each model
-        def update_outputs(group_name):
-            selected_models = model_groups[group_name]
-            # Create image and tally elements for the selected group
-            outputs = [gr.Image(label=model, visible=True, width=170, height=170) for model in selected_models]
-            tallies_dict = {model: 0 for model in selected_models}
-            tally_boxes = [gr.Textbox(value="0", label=f"Tally for {model}", interactive=False) for model in selected_models]
-            return outputs, tally_boxes, tallies_dict
-        model_group_selector.change(
-            update_outputs,
-            inputs=[model_group_selector],
-            outputs=[output, tallies],
-        )
-        def generate_images(prompt, outputs, tallies):
-            for idx, model_output in enumerate(outputs):
-                model_str = list(tallies.keys())[idx]
-                result, tallies = gen_fn(model_str, prompt, tallies)
-                model_output.update(value=result)
-            for idx, tally_box in enumerate(tallies.values()):
-                tally_box.update(value=str(tally_box))
-            return tallies
-        gen_button.click(
-            generate_images,
-            inputs=[txt_input, output, tallies],
-            outputs=[tallies],
-        )
 js_code = """
 <script>
@@ -119,6 +85,7 @@ js_code = """
 """
 with gr.Blocks() as demo:
     make_me()
     gr.Markdown(js_code)

 import gradio as gr
 from random import randint
+from all_models import models
 from datetime import datetime
+kii = "mohawk femboy racecar driver"
+def get_current_time():
+    now = datetime.now()
+    current_time = now.strftime("%Y-%m-%d %H:%M:%S")
+    return f'{kii} {current_time}'
 def load_fn(models):
+    models_load = {}
     for model in models:
+        if model not in models_load:
+            try:
+                m = gr.load(f'models/{model}')  # Adjust `gr.load` as needed
+            except Exception as error:
+                m = gr.Interface(lambda txt: None, ['text'], ['image'])
+            models_load[model] = m
+    return models_load
+models_load = load_fn(models)
+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]
+def gen_fn(model_str, prompt):
     if model_str == 'NA':
+        return None
     noise = str(randint(0, 9999))
     combined_prompt = f'{prompt} {model_str} {noise}'
+    print(f"Generating with prompt: {combined_prompt}")  # Debug line
+    return models_load.get(model_str, lambda txt: None)(f'{prompt} {noise}')
 def make_me():
     with gr.Row():
+        txt_input = gr.Textbox(label='Your prompt:', lines=2, value=kii)
         gen_button = gr.Button('Generate images', elem_id="generate-btn")
+        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, width=170, height=170) for m in default_models]
+        current_models = [gr.Textbox(value=m, visible=False) for m in default_models]
+        for m, o in zip(current_models, output):
+            gen_button.click(gen_fn, inputs=[m, txt_input], outputs=o)
 js_code = """
 <script>
 """
 with gr.Blocks() as demo:
+    gr.Markdown("<div></div>")
     make_me()
     gr.Markdown(js_code)