Update app.py

app.py

@@ -8,143 +8,22 @@ from PIL import Image
 from fastapi import FastAPI, Request
 from deep_translator import GoogleTranslator
 
-# CSS yang lebih lengkap dengan berbagai elemen styling
 css_code = """
-/* General Styling */
-.gradio-container {
-    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
-    background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
-    min-height: 100vh;
-    padding: 20px;
-}
-
-/* Textbox Styling */
 #custom_textbox {
+    width: 100%;
     min-height: 150px;
-    padding: 15px;
-    border-radius: 12px;
-    border: 2px solid #e0e0e0;
-    font-size: 16px;
-    background: rgba(255, 255, 255, 0.9);
-    box-shadow: 0 2px 10px rgba(0, 0, 0, 0.05);
-    transition: all 0.3s ease;
 }
-
-#custom_textbox:focus {
-    border-color: #4facfe;
-    box-shadow: 0 0 0 3px rgba(79, 172, 254, 0.2);
-    outline: none;
-}
-
-/* Button Styling */
 #custom_gen_button {
-    background: linear-gradient(to right, #4facfe 0%, #00f2fe 100%);
-    color: white;
-    border: none;
-    border-radius: 12px;
-    padding: 12px 24px;
-    font-weight: 600;
-    font-size: 16px;
-    cursor: pointer;
-    transition: all 0.3s ease;
-    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-    width: 100%;
-    margin-bottom: 10px;
+    background: #4CAF50 !important;
+    color: white !important;
 }
-
-#custom_gen_button:hover {
-    transform: translateY(-2px);
-    box-shadow: 0 6px 12px rgba(0, 0, 0, 0.15);
-}
-
 #custom_stop_button {
-    background: linear-gradient(to right, #ff4d4d 0%, #f97878 100%);
-    color: white;
-    border: none;
-    border-radius: 12px;
-    padding: 12px 24px;
-    font-weight: 600;
-    font-size: 16px;
-    cursor: pointer;
-    transition: all 0.3s ease;
-    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-    width: 100%;
-}
-
-#custom_stop_button:hover {
-    transform: translateY(-2px);
-    box-shadow: 0 6px 12px rgba(0, 0, 0, 0.15);
+    background: #F44336 !important;
+    color: white !important;
 }
-
-/* Image Styling */
 #custom_image {
-    border-radius: 16px;
-    box-shadow: 0 8px 24px rgba(0, 0, 0, 0.1);
-    transition: all 0.3s ease;
-    background: white;
-    padding: 10px;
-    max-width: 100%;
-    height: auto;
-}
-
-#custom_image:hover {
-    transform: scale(1.01);
-    box-shadow: 0 12px 28px rgba(0, 0, 0, 0.15);
-}
-
-/* Dropdown Styling */
-.dark .dropdown-option {
-    background: #2d3748 !important;
-}
-
-.dark .dropdown-option:hover {
-    background: #4a5568 !important;
-}
-
-/* Panel Styling */
-.gr-box {
-    border-radius: 16px !important;
-    background: rgba(255, 255, 255, 0.8) !important;
-    backdrop-filter: blur(10px);
-    border: 1px solid rgba(255, 255, 255, 0.2) !important;
-    box-shadow: 0 8px 32px rgba(0, 0, 0, 0.1) !important;
-    padding: 20px !important;
-}
-
-/* Label Styling */
-.gr-label {
-    font-weight: 600 !important;
-    color: #2d3748 !important;
-    margin-bottom: 8px !important;
-    font-size: 16px !important;
-}
-
-/* Responsive Adjustments */
-@media (max-width: 768px) {
-    .gradio-container {
-        padding: 10px;
-    }
-    
-    #custom_textbox {
-        min-height: 120px;
-        font-size: 14px;
-    }
-    
-    #custom_gen_button, #custom_stop_button {
-        padding: 10px 15px;
-        font-size: 14px;
-    }
-}
-
-/* Animation for loading state */
-@keyframes pulse {
-    0% { opacity: 0.6; }
-    50% { opacity: 1; }
-    100% { opacity: 0.6; }
-}
-
-.loading {
-    animation: pulse 1.5s infinite;
+    width: 100%;
+    max-height: 768px;
 }
 """
 
@@ -209,75 +88,99 @@ async def api_generate(request: Request):
 # Gradio Interface
 def make_me():
     with gr.Row():
-        with gr.Column(scale=4):
+        # Left Column (50% width)
+        with gr.Column(scale=1, min_width=400):
             txt_input = gr.Textbox(
                 label='Your prompt:', 
                 lines=4, 
                 container=False, 
                 elem_id="custom_textbox", 
-                placeholder="Describe the image you want to generate...",
-                interactive=True
+                placeholder="Enter your prompt here..."
             )
-        
-        with gr.Column(scale=1):
-            gen_button = gr.Button('Generate image', elem_id="custom_gen_button", variant="primary")
-            stop_button = gr.Button('Stop', variant='secondary', interactive=False, 
-                                elem_id="custom_stop_button")
             
-            def on_generate_click():
-                return gr.Button('Generating...', interactive=False, elem_id="custom_gen_button", variant="stop"), gr.Button('Stop', variant='secondary', interactive=True, elem_id="custom_stop_button")
-            
-            def on_stop_click():
-                return gr.Button('Generate image', elem_id="custom_gen_button", variant="primary"), 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
+            )
             
-            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,
-                                      interactive=True,
-                                      elem_classes=["model-selector"])
+            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):
             output_image = gr.Image(
                 label="Generated Image", 
-                width=512,
-                height=768,
                 elem_id="custom_image", 
                 show_label=True, 
                 interactive=False
             )
             
-            # JSON output
-            json_output = gr.JSON(label="API Response", elem_id="api-response")
-            
-            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
+            json_output = gr.JSON(
+                label="API Response",
+                container=False
+            )
             
-            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])
+    # 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]
+    )
 
 # Create Gradio app
-with gr.Blocks(css=css_code, theme=gr.themes.Soft()) as demo: 
-    gr.Markdown("# AI Image Generator", elem_id="title")
-    gr.Markdown("Generate stunning images from text prompts using advanced AI models", elem_id="subtitle")
+with gr.Blocks(css=css_code, title="Image Generation App") as demo: 
+    gr.Markdown("# Image Generation Tool")
     make_me()
 
 # Enable queue before mounting