Update app.py

app.py

@@ -21,30 +21,27 @@ model_path = hf_hub_download(
     filename="recocoamixXL3_coamixXL3.safetensors"
 )
 
-# 単一ファイルからパイプラインを構築
+# パイプライン構築
 pipe = StableDiffusionXLPipeline.from_single_file(
     model_path,
     torch_dtype=torch_dtype,
     use_safetensors=True
 ).to(device)
 
-# スケジューラのプリセット
+# スケジューラ設定
 euler_scheduler = EulerAncestralDiscreteScheduler.from_config(
     pipe.scheduler.config,
     use_karras_sigmas=True
 )
-dpm_scheduler = DPMSolverMultistepScheduler.from_config(
-    pipe.scheduler.config
-)
+dpm_scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.scheduler = euler_scheduler
 
-# 生成履歴を (PIL.Image, キャプションテキスト) タプルで保持
+# 履歴保存
 history = []
 
-# HTMLテーブル生成ヘルパー
 def make_html_table(caption):
     rows = caption.split("\n")
-    html = '<table style="width:100%;border-collapse:collapse;">'
+    html = '<table style="width:100%;border-collapse:collapse;background:#fffaf1;color:#000">'
     for row in rows:
         if ": " in row:
             key, val = row.split(": ", 1)
@@ -55,19 +52,12 @@ def make_html_table(caption):
     html += '</table>'
     return html
 
-# 推論関数
-def infer(
-    prompt, neg, seed, rand, w, h, cfg, steps, scheduler_type,
-    progress=gr.Progress(track_tqdm=True)
-):
+def infer(prompt, neg, seed, rand, w, h, cfg, steps, scheduler_type, progress=gr.Progress(track_tqdm=True)):
     if rand:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device=device).manual_seed(seed)
 
-    # スケジューラ切替
-    pipe.scheduler = (
-        euler_scheduler if scheduler_type == "Euler Ancestral" else dpm_scheduler
-    )
+    pipe.scheduler = euler_scheduler if scheduler_type == "Euler Ancestral" else dpm_scheduler
     pipe.scheduler.set_timesteps(steps)
 
     def _callback(pipeline, step_idx, timestep, callback_kwargs):
@@ -86,7 +76,6 @@ def infer(
     )
     img = output.images[0]
 
-    # キャプション構築
     caption_text = (
         f"Prompt: {prompt}\n"
         f"Negative: {neg or 'None'}\n"
@@ -96,27 +85,22 @@ def infer(
         f"Steps: {steps}\n"
         f"Scheduler: {scheduler_type}"
     )
-
-    # 履歴に追加（最新を先頭）
     history.insert(0, (img, caption_text))
     progress(1.0, desc="Done!")
 
-    # Gallery には (PIL.Image, HTML caption) のリストを渡す
-    gallery_items = [
-        (item[0], make_html_table(item[1])) for item in history
-    ]
+    gallery_items = [(item[0], make_html_table(item[1])) for item in history]
     return img, gallery_items
 
-# CSS設定（温かい喫茶店風＋ライトモード強制）
+# CSS 設定（ダークモード強制防止＋カフェ風テーマ）
 css = """
 body {
-  background-color: #f4e1c1;
-  color: #3b2f2f;
+  background-color: #f4e1c1 !important;
   font-family: 'Georgia', serif;
+  color: #000 !important;
 }
-body.dark {
-  background-color: #f4e1c1 !important;
-  color: #3b2f2f !important;
+html, .gradio-container, .dark, .dark * {
+  background: #fffaf1 !important;
+  color: #000 !important;
 }
 #col-container {
   background: #fffaf1;
@@ -126,11 +110,6 @@ body.dark {
   margin: auto;
   max-width: 780px;
 }
-#col-container a {
-  color: #8b5e3c;
-  text-decoration: none;
-  border-bottom: 1px dotted #8b5e3c;
-}
 .gr-button {
   background-color: #d4a373 !important;
   color: white !important;
@@ -183,19 +162,16 @@ with gr.Blocks(css=css) as demo:
 
         with gr.Row():
             prompt = gr.Textbox(lines=1, placeholder="Prompt…", value="1girl, cocoart, masterpiece, anime,")
-            neg = gr.Textbox(lines=1, placeholder="Negative prompt", value=(
-                "low quality, worst quality, bad shadow, lowres, error, miss stroke, "
-                "sketch art, smoke, ugly, extra digits, creepy, imprecise, glowing blur,"
-            ))
+            neg = gr.Textbox(lines=1, placeholder="Negative prompt", value="low quality, worst quality, bad shadow, lowres, error, miss stroke, sketch art, smoke, ugly, extra digits, creepy, imprecise, glowing blur,")
         with gr.Row():
             seed_sl = gr.Slider(0, MAX_SEED, step=1, value=0, label="Seed")
-            rand = gr.Checkbox(True, label="Randomize seed")
+            rand    = gr.Checkbox(True, label="Randomize seed")
         with gr.Row():
-            width = gr.Slider(256, MAX_SIZE, step=32, value=512, label="Width")
+            width  = gr.Slider(256, MAX_SIZE, step=32, value=512, label="Width")
             height = gr.Slider(256, MAX_SIZE, step=32, value=512, label="Height")
         with gr.Row():
-            cfg = gr.Slider(1.0, 30.0, step=0.1, value=7.5, label="CFG Scale")
-            steps = gr.Slider(1, 50, step=1, value=12, label="Steps")
+            cfg    = gr.Slider(1.0, 30.0, step=0.1, value=7.5, label="CFG Scale")
+            steps  = gr.Slider(1, 50, step=1, value=12,   label="Steps")
         with gr.Row():
             scheduler_type = gr.Radio(choices=["Euler Ancestral", "DPM++ 2M SDE"], value="Euler Ancestral", label="Scheduler")
             run = gr.Button("Generate")
@@ -220,7 +196,7 @@ with gr.Blocks(css=css) as demo:
         )
 
         img_out = gr.Image()
-        state = gr.State([])
+        state   = gr.State([])
         history_gallery = gr.Gallery(label="生成履歴", columns=4, height=280, show_label=False, interactive=True, type="pil")
 
     run.click(