Update app.py

app.py

@@ -8,27 +8,10 @@ from PIL import Image
 from fastapi import FastAPI, Request
 from deep_translator import GoogleTranslator
 
-css_code = """
-#custom_textbox {
-    width: 100%;
-    min-height: 150px;
-}
-#custom_gen_button {
-    background: #4CAF50 !important;
-    color: white !important;
-}
-#custom_stop_button {
-    background: #F44336 !important;
-    color: white !important;
-}
-#custom_image {
-    width: 100%;
-    max-height: 768px;
-}
-"""
+css_code = os.getenv("DazDinGo_CSS")
 
 # Initialize translator
-translator = GoogleTranslator(source='auto', target='en')
+translator = DeepTranslator(source="auto", target="en")
 
 # Load models
 models_load = {}
@@ -88,99 +71,70 @@ async def api_generate(request: Request):
 # Gradio Interface
 def make_me():
     with gr.Row():
-        # Left Column (50% width)
-        with gr.Column(scale=1, min_width=400):
+        with gr.Column(scale=4):
             txt_input = gr.Textbox(
                 label='Your prompt:', 
                 lines=4, 
                 container=False, 
                 elem_id="custom_textbox", 
-                placeholder="Enter your prompt here..."
+                placeholder="Prompt"
             )
+        
+        with gr.Column(scale=1):
+            gen_button = gr.Button('Generate image', elem_id="custom_gen_button")
+            stop_button = gr.Button('Stop', variant='secondary', interactive=False, 
+                                elem_id="custom_stop_button")
             
-            model_dropdown = gr.Dropdown(
-                models, 
-                label="Select LoRA Model", 
-                value=models[0] if models else None,
-                container=False
-            )
+            def on_generate_click():
+                return gr.Button('Generate image', elem_id="custom_gen_button"), gr.Button('Stop', variant='secondary', interactive=True, elem_id="custom_stop_button")
             
-            with gr.Row():
-                gen_button = gr.Button(
-                    'Generate Image', 
-                    elem_id="custom_gen_button",
-                    variant='primary'
-                )
-                stop_button = gr.Button(
-                    'Stop', 
-                    variant='stop',
-                    elem_id="custom_stop_button",
-                    interactive=False
-                )
-        
-        # Right Column (50% width)
-        with gr.Column(scale=1, min_width=400):
+            def on_stop_click():
+                return gr.Button('Generate image', 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():
+        with gr.Column():
+            model_dropdown = gr.Dropdown(models, label="Select Model", 
+                                      value=models[0] if models else None)
             output_image = gr.Image(
                 label="Generated Image", 
+                width=512,
+                height=768,
                 elem_id="custom_image", 
                 show_label=True, 
                 interactive=False
             )
             
-            json_output = gr.JSON(
-                label="API Response",
-                container=False
-            )
+            # JSON output
+            json_output = gr.JSON(label="API Response")
             
-    # Functionality remains the same
-    def generate_wrapper(model_str, prompt):
-        # Translate prompt to English
-        translated_prompt = translator.translate(prompt)
-        image = gen_image(model_str, translated_prompt)
-        if image is None:
-            return None, {"error": "Generation failed"}
-        
-        base64_str = image_to_base64(image)
-        response = {
-            "status": "success",
-            "model": model_str,
-            "original_prompt": prompt,
-            "translated_prompt": translated_prompt,
-            "image_base64": base64_str,
-            "image_format": "jpeg"
-        }
-        return image, response
-    
-    def on_generate_click():
-        return gr.Button(interactive=False), gr.Button(interactive=True)
-    
-    def on_stop_click():
-        return gr.Button(interactive=True), gr.Button(interactive=False)
-    
-    gen_event = gen_button.click(
-        on_generate_click,
-        inputs=None,
-        outputs=[gen_button, stop_button]
-    ).then(
-        generate_wrapper,
-        [model_dropdown, txt_input],
-        [output_image, json_output]
-    ).then(
-        on_stop_click,
-        inputs=None,
-        outputs=[gen_button, stop_button]
-    )
-    
-    stop_button.click(
-        on_stop_click,
-        inputs=None,
-        outputs=[gen_button, stop_button],
-        cancels=[gen_event]
-    )
+            def generate_wrapper(model_str, prompt):
+                # Translate prompt to English
+                translated_prompt = translator.translate(prompt)
+                image = gen_image(model_str, translated_prompt)
+                if image is None:
+                    return None, {"error": "Generation failed"}
+                
+                base64_str = image_to_base64(image)
+                response = {
+                    "status": "success",
+                    "model": model_str,
+                    "original_prompt": prompt,
+                    "translated_prompt": translated_prompt,
+                    "image_base64": base64_str,
+                    "image_format": "jpeg"
+                }
+                return image, response
+            
+            gen_event = gen_button.click(generate_wrapper, [model_dropdown, txt_input], 
+                                     [output_image, json_output])
+            stop_button.click(on_stop_click, inputs=None, 
+                            outputs=[gen_button, stop_button], cancels=[gen_event])
 
 # Create Gradio app
-with gr.Blocks(css=css_code, title="Image Generation App") as demo: 
-    gr.Markdown("# Image Generation Tool")
+with gr.Blocks(css=css_code) as demo: 
     make_me()
 
 # Enable queue before mounting