elismasilva commited on
Commit
fa3cf34
·
1 Parent(s): e8cf629

fixed theme dark

Browse files
Files changed (2) hide show
  1. app.py +13 -22
  2. pipeline/util.py +0 -60
app.py CHANGED
@@ -5,8 +5,7 @@ import gradio as gr
5
 
6
  from pipeline.mod_controlnet_tile_sr_sdxl import StableDiffusionXLControlNetTileSRPipeline, calculate_overlap
7
  from pipeline.util import (
8
- SAMPLERS,
9
- Platinum,
10
  create_hdr_effect,
11
  progressive_upscale,
12
  select_scheduler,
@@ -64,8 +63,8 @@ def predict(
64
  image = create_hdr_effect(image, hdr)
65
 
66
  # Update target height and width
67
- target_height = control_image.height
68
- target_width = control_image.width
69
  print(f"Target resolution: H:{target_height} x W:{target_width}")
70
  print(f"Applied HDR effect: {True if hdr > 0 else False}")
71
 
@@ -114,23 +113,18 @@ def select_tile_weighting_method(tile_weighting_method):
114
  # endregion
115
 
116
  css = """
117
- body {
118
- background: linear-gradient(135deg, #F0F0F0, #FFFFFF);
119
- font-family: 'Helvetica Neue', Helvetica, Arial, sans-serif;
120
- color: #333;
121
  margin: 0;
122
  padding: 0;
123
  }
124
- .gradio-container {
125
- background: rgba(255, 255, 255, 0.95);
126
  border-radius: 15px;
127
  padding: 30px 40px;
128
  box-shadow: 0 8px 30px rgba(0, 0, 0, 0.3);
129
- margin: 40px 340px;
130
- /*max-width: 1200px;*/
131
  }
132
- .gradio-container h1 {
133
- color: #333;
134
  text-shadow: 1px 1px 2px rgba(0, 0, 0, 0.2);
135
  }
136
  .fillable {
@@ -147,16 +141,13 @@ body {
147
  #tips_row{
148
  padding-left: 20px;
149
  }
150
- .sidebar {
151
- background: rgba(255, 255, 255, 0.98);
152
  border-radius: 10px;
153
  padding: 10px;
154
  box-shadow: 0 4px 15px rgba(0, 0, 0, 0.2);
155
  }
156
- .sidebar .toggle-button {
157
- background: linear-gradient(90deg, #7367f0, #9c93f4);
158
- border: none;
159
- color: #fff;
160
  padding: 12px 24px;
161
  text-transform: uppercase;
162
  font-weight: bold;
@@ -204,7 +195,7 @@ about = """
204
  If you have any questions or suggestions, feel free to send your question to <b>[email protected]</b>.
205
  """
206
 
207
- with gr.Blocks(css=css, theme=Platinum(), title="MoD ControlNet Tile Upscaler") as app:
208
  gr.Markdown(title)
209
  with gr.Row():
210
  with gr.Column(scale=3):
@@ -351,7 +342,7 @@ with gr.Blocks(css=css, theme=Platinum(), title="MoD ControlNet Tile Upscaler")
351
  ],
352
  fn=predict,
353
  outputs=result,
354
- cache_examples=True,
355
  )
356
 
357
  max_tile_size.select(fn=set_maximum_resolution, inputs=[max_tile_size, resolution], outputs=resolution)
 
5
 
6
  from pipeline.mod_controlnet_tile_sr_sdxl import StableDiffusionXLControlNetTileSRPipeline, calculate_overlap
7
  from pipeline.util import (
8
+ SAMPLERS,
 
9
  create_hdr_effect,
10
  progressive_upscale,
11
  select_scheduler,
 
63
  image = create_hdr_effect(image, hdr)
64
 
65
  # Update target height and width
66
+ target_height = image.height
67
+ target_width = image.width
68
  print(f"Target resolution: H:{target_height} x W:{target_width}")
69
  print(f"Applied HDR effect: {True if hdr > 0 else False}")
70
 
 
113
  # endregion
114
 
115
  css = """
116
+ body {
117
+ font-family: 'Helvetica Neue', Helvetica, Arial, sans-serif;
 
 
118
  margin: 0;
119
  padding: 0;
120
  }
121
+ .gradio-container {
 
122
  border-radius: 15px;
123
  padding: 30px 40px;
124
  box-shadow: 0 8px 30px rgba(0, 0, 0, 0.3);
125
+ margin: 40px 340px;
 
126
  }
127
+ .gradio-container h1 {
 
128
  text-shadow: 1px 1px 2px rgba(0, 0, 0, 0.2);
129
  }
130
  .fillable {
 
141
  #tips_row{
142
  padding-left: 20px;
143
  }
144
+ .sidebar {
 
145
  border-radius: 10px;
146
  padding: 10px;
147
  box-shadow: 0 4px 15px rgba(0, 0, 0, 0.2);
148
  }
149
+ .sidebar .toggle-button {
150
+ border: none;
 
 
151
  padding: 12px 24px;
152
  text-transform: uppercase;
153
  font-weight: bold;
 
195
  If you have any questions or suggestions, feel free to send your question to <b>[email protected]</b>.
196
  """
197
 
198
+ with gr.Blocks(css=css, theme=gr.themes.Ocean(), title="MoD ControlNet Tile Upscaler") as app:
199
  gr.Markdown(title)
200
  with gr.Row():
201
  with gr.Column(scale=3):
 
342
  ],
343
  fn=predict,
344
  outputs=result,
345
+ cache_examples=False,
346
  )
347
 
348
  max_tile_size.select(fn=set_maximum_resolution, inputs=[max_tile_size, resolution], outputs=resolution)
pipeline/util.py CHANGED
@@ -18,9 +18,6 @@ import cv2
18
  import numpy as np
19
  import torch
20
  from PIL import Image
21
- from gradio.themes import Default
22
- import gradio as gr
23
-
24
 
25
  MAX_SEED = np.iinfo(np.int32).max
26
  SAMPLERS = {
@@ -84,63 +81,6 @@ SAMPLERS = {
84
  "TCD trailing": ("TCDScheduler", {"timestep_spacing": "trailing"}),
85
  }
86
 
87
- class Platinum(Default):
88
- def __init__(
89
- self,
90
- ):
91
- super().__init__(
92
- font = (
93
- gr.themes.GoogleFont("Karla"), 'Segoe UI Emoji', 'Public Sans', 'system-ui', 'sans-serif'
94
- )
95
- )
96
- self.name = "Diffusers"
97
- super().set(
98
- block_border_width='1px',
99
- block_border_width_dark='1px',
100
- block_info_text_size='13px',
101
- block_info_text_weight='450',
102
- block_info_text_color='#474a50',
103
- block_label_background_fill='*background_fill_secondary',
104
- block_label_text_color='*neutral_700',
105
- block_title_text_color='black',
106
- block_title_text_weight='600',
107
- block_background_fill='#fcfcfc',
108
- body_background_fill='*background_fill_secondary',
109
- body_text_color='black',
110
- background_fill_secondary='#f8f8f8',
111
- border_color_accent='*primary_50',
112
- border_color_primary='#ededed',
113
- color_accent='#7367f0',
114
- color_accent_soft='#fcfcfc',
115
- panel_background_fill='#fcfcfc',
116
- section_header_text_weight='600',
117
- checkbox_background_color='*background_fill_secondary',
118
- input_background_fill='white',
119
- input_placeholder_color='*neutral_300',
120
- loader_color = '#7367f0',
121
- slider_color='#7367f0',
122
- table_odd_background_fill='*neutral_100',
123
- button_small_radius='*radius_sm',
124
- button_primary_background_fill='linear-gradient(to bottom right, #7367f0, #9c93f4)',
125
- button_primary_background_fill_hover='linear-gradient(to bottom right, #9c93f4, #9c93f4)',
126
- button_primary_background_fill_hover_dark='linear-gradient(to bottom right, #5e50ee, #5e50ee)',
127
- button_cancel_background_fill='linear-gradient(to bottom right, #fc0379, #ff88ac)',
128
- button_cancel_background_fill_dark='linear-gradient(to bottom right, #dc2626, #b91c1c)',
129
- button_cancel_background_fill_hover='linear-gradient(to bottom right, #f592c9, #f592c9)',
130
- button_cancel_background_fill_hover_dark='linear-gradient(to bottom right, #dc2626, #dc2626)',
131
- button_primary_border_color='#5949ed',
132
- button_primary_text_color='white',
133
- button_cancel_text_color='white',
134
- button_cancel_text_color_dark='#dc2626',
135
- button_cancel_border_color='#f04668',
136
- button_cancel_border_color_dark='#dc2626',
137
- button_cancel_border_color_hover='#fe6565',
138
- button_cancel_border_color_hover_dark='#dc2626',
139
- form_gap_width='1px',
140
- layout_gap='5px'
141
- )
142
-
143
-
144
  def select_scheduler(pipe, selected_sampler):
145
  import diffusers
146
 
 
18
  import numpy as np
19
  import torch
20
  from PIL import Image
 
 
 
21
 
22
  MAX_SEED = np.iinfo(np.int32).max
23
  SAMPLERS = {
 
81
  "TCD trailing": ("TCDScheduler", {"timestep_spacing": "trailing"}),
82
  }
83
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
84
  def select_scheduler(pipe, selected_sampler):
85
  import diffusers
86