Diff3

Running

App Files Files Community

DazDin commited on Nov 18, 2024

Commit

4c967c0

verified ·

1 Parent(s): 8c58efd

app.py

Browse files

Files changed (1) hide show

app.py +182 -48

app.py CHANGED Viewed

@@ -31,65 +31,199 @@ def extend_choices(choices):
 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');
-"""
-css = """
-body, .gradio-container, .gradio-app, .gradio-interface, .gr-box, .gr-row, .gr-column, .gr-textbox, .gr-button, .gr-image, .gr-accordion, .gr-checkbox-group, .gr-html {
-    background-color: black !important;
-    color: white !important; /* Optional: Set text color to white for better visibility */
-}
-div.float.svelte-1mwvhlq {
-    position: absolute;
-    top: var(--block-label-margin);
-    left: var(--block-label-margin);
-    background: none;
-    border: none;
 }
 """
-with gr.Blocks(css=css) as demo:
-    gr.Markdown("<script>" + js_code + "</script>")
     make_me()
-demo.queue(concurrency_count=100)
 demo.launch()

 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", placeholder="Prompt")
+            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("")
+custom_css = """
+:root {
+    --body-background-fill: #2d3d4f;
+}
+body {
+    background-color: var(--body-background-fill) !important;
+    color: #2d3d4f;
+    margin: 0;
+    padding: 0;
+    font-family: Arial, sans-serif;
+    height: 100vh;
+    overflow-y: auto;
+}
+.gradio-container {
+    background-color: #2d3d4f;
+    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 auto; /* Center the container horizontally */
+    display: block; /* Ensure it's a block element */
+    min-height: 100vh; /* Adjust the height of the container */
+}
+.app_title {
+    background-color: #2d3d4f;
+    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 */
+}
+.custom_textbox {
+    background-color: #2d343f;
+    border: 1px solid #3b4252;
+    color: #7f8184;
+    padding: 10px;
+    border-radius: 4px;
+    margin-bottom: 10px;
+    width: 100%;
+    box-sizing: border-box;
+}
+.custom_gen_button {
+    background-color: #8b38ff;
+    border: 1px solid #ffffff;
+    color: blue;
+    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: 1px solid #ffffff;
+    color: blue;
+    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);
+}
+.custom_image {
+    border: 1px solid #3b4252;
+    background-color: #2d343f;
+    border-radius: 4px;
+    margin: 10px;
+    max-width: 100%;
+    box-sizing: border-box;
+}
+.custom_accordion {
+    background-color: #2d3d4f;
+    color: #7f8184;
+    border: 1px solid #3b4252;
+    border-radius: 4px;
+    margin-top: 20px;
+    width: 100%;
+    box-sizing: border-box;
+    transition: margin 0.2s ease; /* Smooth transition for margin */
+}
+.custom_accordion .gr-accordion-header {
+    background-color: #2d3d4f;
+    color: #7f8184;
+    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: #2d3d4f;
+}
+.custom_accordion .gr-accordion-content {
+    padding: 10px 20px;
+    background-color: #2d3d4f;
+    border-top: 1px solid #5b6270;
+    max-height: 0; /* Start with no height */
+    overflow: hidden;
+    transition: max-height 0.2s ease;
+}
+.custom_accordion .gr-accordion-content.open {
+    max-height: 500px;
+}
+.custom_checkbox_group {
+    background-color: #2d343f;
+    border: 1px solid #3b4252;
+    color: #7f8184;
+    border-radius: 4px;
+    padding: 10px;
+    width: 100%;
+    box-sizing: border-box;
+}
+@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()