Diff2

Running

App Files Files Community

DazDin commited on Nov 16, 2024

Commit

20cbff0

verified ·

1 Parent(s): 5a044c1

app.py

Browse files

Files changed (1) hide show

app.py +43 -241

app.py CHANGED Viewed

@@ -1,264 +1,66 @@
 import gradio as gr
 from random import randint
 from all_models import models
-def load_fn(models):
-    global models_load
-    models_load = {}
-    for model in models:
-        if model not in models_load.keys():
-            try:
-                m = gr.load(f'models/{model}')
-            except Exception as error:
-                m = gr.Interface(lambda txt: None, ['text'], ['image'])
-            models_load.update({model: m})
-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'), elem_id="custom_image") for m in choices_plus]
 def gen_fn(model_str, prompt):
     if model_str == 'NA':
         return None
-    noise = str(randint(0, 9999999))
     return models_load[model_str](f'{prompt} {noise}')
 def make_me():
-    with gr.Row():
-        with gr.Column(scale=1):
-            txt_input = gr.Textbox(label='Your prompt:', lines=3, container=False, elem_id="custom_textbox")
-            with gr.Row():
-                gen_button = gr.Button('Generate images', elem_id="custom_gen_button")
-                stop_button = gr.Button('Stop', variant='secondary', interactive=False, elem_id="custom_stop_button")
-                def on_generate_click():
-                    return gr.Button('Generate images', elem_id="custom_gen_button"), gr.Button('Stop', variant='secondary', interactive=True, elem_id="custom_stop_button")
-                def on_stop_click():
-                    return gr.Button('Generate images', elem_id="custom_gen_button"), gr.Button('Stop', variant='secondary', interactive=False, elem_id="custom_stop_button")
-                gen_button.click(on_generate_click, inputs=None, outputs=[gen_button, stop_button])
-                stop_button.click(on_stop_click, inputs=None, outputs=[gen_button, stop_button])
-    with gr.Row():
-        output = [gr.Image(label=m, min_width=250, height=250, elem_id="custom_image") 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], o)
-            stop_button.click(on_stop_click, inputs=None, outputs=[gen_button, stop_button], cancels=[gen_event])
-    with gr.Accordion('Model selection', elem_id="custom_accordion"):
-        model_choice = gr.CheckboxGroup(models, label=f'{num_models} different models selected', value=default_models, interactive=True, elem_id="custom_checkbox_group")
-        model_choice.change(update_imgbox, model_choice, output)
-        model_choice.change(extend_choices, model_choice, current_models)
-    with gr.Row():
-        gr.HTML("")
-    # Inject JavaScript to handle accordion toggle
-    js_code = """
-    <script>
-    document.addEventListener('DOMContentLoaded', function() {
-        const accordionHeaders = document.querySelectorAll('.gr-accordion-header');
-        accordionHeaders.forEach(header => {
-            header.addEventListener('click', function() {
-                const content = this.nextElementSibling;
-                if (content.classList.contains('open')) {
-                    content.classList.remove('open');
-                } else {
-                    content.classList.add('open');
-                }
-            });
-        });
-    });
-    </script>
-    """
-    gr.HTML(js_code)
-custom_css = """
-/* General Styles */
-body {
-    background-color: #1a242e;
-    color: #c5c6c7;
-    margin: 0;
-    padding: 0;
-    font-family: Arial, sans-serif;
-    display: flex;
-    justify-content: center;
-    align-items: center; /* Align items from the center */
-    height: 100vh;
-    overflow-y: auto;
-}
-/* Container Styles */
-.gradio-container {
-    background-color: #1a242e;
-    color: #c5c6c7;
-    padding: 20px;
-    border-radius: 8px;
-    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.2);
-    width: 100%;
-    max-width: 1200px;
-    margin: 20px;
-    display: flex;
-    flex-direction: column;
-    align-items: center; /* Center items horizontally */
-    min-height: 100vh; /* Adjust the height of the container */
-}
-/* Header Styles */
-.app_title {
-    background-color: #1a242e;
-    color: #c5c6c7;
-    padding: 10px 20px;
-    border-bottom: 1px solid #3b4252;
-    text-align: center;
-    font-size: 24px;
-    font-weight: bold;
-    width: 100%;
-    box-sizing: border-box;
-    margin-bottom: 20px; /* Add margin to separate the header from content */
-}
-/* Textbox Styles */
-.custom_textbox {
-    background-color: #2d343f;
-    border: 1px solid #3b4252;
-    color: #c5c6c7;
-    padding: 10px;
-    border-radius: 4px;
-    margin-bottom: 10px;
-    width: 100%;
-    box-sizing: border-box;
-}
-/* Button Styles */
-.custom_gen_button {
-    background-color: #4a90e2;
-    border: none;
-    color: white;
-    padding: 15px 32px;
-    text-align: center;
-    text-decoration: none;
-    display: inline-block;
-    font-size: 16px;
-    margin: 4px 2px;
-    cursor: pointer;
-    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.2);
-    transition: transform 0.2s, box-shadow 0.2s;
-    border-radius: 4px;
-}
-.custom_gen_button:hover {
-    transform: translateY(-2px);
-    box-shadow: 0 6px 10px rgba(0, 0, 0, 0.3);
-}
-.custom_stop_button {
-    background-color: #6200ea;
-    border: none;
-    color: white;
-    padding: 15px 32px;
-    text-align: center;
-    text-decoration: none;
-    display: inline-block;
-    font-size: 16px;
-    margin: 4px 2px;
-    cursor: pointer;
-    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.2);
-    transition: transform 0.2s, box-shadow 0.2s;
-    border-radius: 4px;
-}
-.custom_stop_button:hover {
-    transform: translateY(-2px);
-    box-shadow: 0 6px 10px rgba(0, 0, 0, 0.3);
-}
-/* Image Styles */
-.custom_image {
-    border: 1px solid #3b4252;
-    background-color: #2d343f;
-    border-radius: 4px;
-    margin: 10px;
-    max-width: 100%;
-    box-sizing: border-box;
-}
-/* Accordion Styles */
-.custom_accordion {
-    background-color: #1a242e;
-    color: #c5c6c7;
-    border: 1px solid #3b4252;
-    border-radius: 4px;
-    margin-top: 20px;
-    width: 100%;
-    box-sizing: border-box;
-    transition: margin 0.3s ease; /* Smooth transition for margin */
-}
-.custom_accordion .gr-accordion-header {
-    background-color: #3b4252;
-    color: #c5c6c7;
-    padding: 10px 20px;
-    border-bottom: 1px solid #5b6270;
-    cursor: pointer;
-    font-size: 18px;
-    font-weight: bold;
-    height: 40px; /* Fixed height for the header */
-    display: flex;
-    align-items: center;
-}
-.custom_accordion .gr-accordion-header:hover {
-    background-color: #4b5260;
-}
-.custom_accordion .gr-accordion-content {
-    padding: 10px 20px;
-    background-color: #2d343f;
-    border-top: 1px solid #5b6270;
-    max-height: 0; /* Start with no height */
-    overflow: hidden;
-    transition: max-height 0.3s ease; /* Smooth transition for height */
-}
-.custom_accordion .gr-accordion-content.open {
-    max-height: 500px; /* Set a reasonable max height */
-}
-/* Checkbox Group Styles */
-.custom_checkbox_group {
-    background-color: #2d343f;
-    border: 1px solid #3b4252;
-    color: #c5c6c7;
-    border-radius: 4px;
-    padding: 10px;
-    width: 100%;
-    box-sizing: border-box;
-}
-/* Responsive Design */
-@media (max-width: 768px) {
-   .gradio-container {
-        width: 100%;
-        margin: 0;
-        padding: 10px;
-    }
-   .custom_textbox,.custom_image,.custom_checkbox_group {
-        width: 100%;
-        box-sizing: border-box;
-    }
-}
 """
-with gr.Blocks(css=custom_css) as demo:
     make_me()
-demo.queue(concurrency_count=200)
 demo.launch()

 import gradio as gr
 from random import randint
 from all_models import models
+from datetime import datetime
+now2 = 0
+def get_current_time():
+    now = datetime.now()
+    now2 = now
+    current_time = now2.strftime("%Y-%m-%d %H:%M:%S")
+    ki = f'{kii} {current_time}'
+    return ki
+def load_fn(models):
+    global models_load
 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))
     return models_load[model_str](f'{prompt} {noise}')
 def make_me():
+        with gr.Row():
+            txt_input = gr.Textbox(label='Your prompt:', lines=3, width=800, max_height=100)
+            gen_button = gr.Button('Generate images', width=30, height=30)
+            stop_button = gr.Button('Stop', variant='secondary', interactive=False, width=30, height=30)
+            gen_button.click(lambda s: 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, min_width=250, height=250) 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], o)
+                stop_button.click(lambda s: 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)
+        with gr.Row():
+            gr.HTML()
+js_code = """
+    console.log('ghgh');
 """
+with gr.Blocks(css="div.float.svelte-1mwvhlq {    position: absolute;    top: var(--block-label-margin);    left: var(--block-label-margin);    background: none;    border: none;}") as demo:
+    gr.Markdown("<script>" + js_code + "</script>")
     make_me()
+demo.queue(concurrency_count=100)
 demo.launch()