Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 import uuid
 from datetime import datetime
 import gradio as gr
+import gradio.themes as themes  # <-- Import the theme module
 import numpy as np
 import spaces
 import torch
@@ -16,7 +17,7 @@ if not os.path.exists(SAVE_DIR):
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 repo_id = "black-forest-labs/FLUX.1-dev"
-adapter_id = "openfree/korea-president-yoon"
+adapter_id = "openfree/president-k-dj"
 
 pipeline = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch.bfloat16)
 pipeline.load_lora_weights(adapter_id)
@@ -47,8 +48,11 @@ def load_generated_images():
         return []
     
     # Load all images from the directory
-    image_files = [os.path.join(SAVE_DIR, f) for f in os.listdir(SAVE_DIR) 
-                  if f.endswith(('.png', '.jpg', '.jpeg', '.webp'))]
+    image_files = [
+        os.path.join(SAVE_DIR, f)
+        for f in os.listdir(SAVE_DIR)
+        if f.endswith(('.png', '.jpg', '.jpeg', '.webp'))
+    ]
     # Sort by creation time (newest first)
     image_files.sort(key=lambda x: os.path.getctime(x), reverse=True)
     return image_files
@@ -90,30 +94,38 @@ def inference(
     return image, seed, load_generated_images()
 
 examples = [
-    "A man playing fetch with a golden retriever in a sunny park. He wears casual weekend clothes and throws a red frisbee with joy. The dog leaps gracefully through the air, tail wagging with excitement. Warm afternoon sunlight filters through the trees, creating a peaceful scene of companionship. [president yoon]",
-
-    "A soldier standing at attention in full military gear, holding a standard-issue rifle. His uniform is crisp and properly adorned with medals. Behind him, other soldiers march in formation during a military parade. The scene conveys discipline and duty. [president yoon]",
-
-    "A medieval knight in gleaming armor, holding an ornate sword and shield. He stands proudly in front of a majestic castle, his cape flowing in the wind. The shield bears intricate heraldic designs, and sunlight glints off his polished armor. [president yoon]",
-
-    "A charismatic political leader addressing a crowd from a podium. He wears a well-fitted suit and gestures confidently while speaking. The audience fills a large plaza, holding supportive banners and signs. News cameras capture the moment as he delivers his speech. [president yoon]",
-
-    "A man enjoying a peaceful morning at home, reading a newspaper at his breakfast table. He wears comfortable home clothes and sips coffee from a favorite mug. Sunlight streams through the kitchen window, and a house plant adds a touch of nature to the cozy domestic scene. [president yoon]",
-
-    "A businessman walking confidently through a modern office building. He carries a leather briefcase and wears a tailored navy suit. Floor-to-ceiling windows reveal a cityscape behind him, and his expression shows determination and purpose. [president yoon]"
+    "A man playing fetch with a golden retriever in a sunny park. He wears casual weekend clothes and throws a red frisbee with joy. The dog leaps gracefully through the air, tail wagging with excitement. Warm afternoon sunlight filters through the trees, creating a peaceful scene of companionship. [president_K_DJ ]",
+    "A soldier standing at attention in full military gear, holding a standard-issue rifle. His uniform is crisp and properly adorned with medals. Behind him, other soldiers march in formation during a military parade. The scene conveys discipline and duty. [president_K_DJ ]",
+    "A medieval knight in gleaming armor, holding an ornate sword and shield. He stands proudly in front of a majestic castle, his cape flowing in the wind. The shield bears intricate heraldic designs, and sunlight glints off his polished armor. [president_K_DJ ]",
+    "A charismatic political leader addressing a crowd from a podium. He wears a well-fitted suit and gestures confidently while speaking. The audience fills a large plaza, holding supportive banners and signs. News cameras capture the moment as he delivers his speech. [president_K_DJ ]",
+    "A man enjoying a peaceful morning at home, reading a newspaper at his breakfast table. He wears comfortable home clothes and sips coffee from a favorite mug. Sunlight streams through the kitchen window, and a house plant adds a touch of nature to the cozy domestic scene. [president_K_DJ]",
+    "A businessman walking confidently through a modern office building. He carries a leather briefcase and wears a tailored navy suit. Floor-to-ceiling windows reveal a cityscape behind him, and his expression shows determination and purpose. [president_K_DJ ]"
 ]
 
 css = """
 footer {
     visibility: hidden;
 }
+#col-container {
+    margin-top: 10px;
+}
 """
 
-with gr.Blocks(theme=gr.themes.Soft, css=css, analytics_enabled=False) as demo:
-    gr.HTML('<div class="title"> President DJ(김대중) in KOREA </div>')
+# Create a customized Soft theme for better visuals
+custom_theme = themes.Soft(
+    primary_hue="blue",       # Color for primary actions
+    secondary_hue="gray",     # Color for secondary actions
+    neutral_hue="cyan",       # Neutral color scheme
+    text_size="md",           # Medium text size
+    spacing_size="md",        # Medium spacing for elements
+    radius_size="lg",         # Larger corner radius
+    font=["Open Sans", "sans-serif"]  # Modern, clean font
+)
+
+with gr.Blocks(theme=custom_theme, css=css, analytics_enabled=False) as demo:
+    gr.HTML('<div class="title" style="text-align:center; font-size:2em; margin-bottom:20px;">\
+             <strong> President DJ(김대중) in KOREA </strong></div>')
 
-
-    
     with gr.Tabs() as tabs:
         with gr.Tab("Generation"):
             with gr.Column(elem_id="col-container"):
@@ -207,6 +219,7 @@ with gr.Blocks(theme=gr.themes.Soft, css=css, analytics_enabled=False) as demo:
         outputs=generated_gallery,
     )
 
+    # tie the run_button and prompt.submit to the same function
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=inference,
@@ -224,4 +237,4 @@ with gr.Blocks(theme=gr.themes.Soft, css=css, analytics_enabled=False) as demo:
     )
 
 demo.queue()
-demo.launch()
+demo.launch()