Update app.py

app.py

@@ -17,8 +17,7 @@ else:
 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)
 
-MAX_IMAGE_SIZE = 512  
-
+MAX_IMAGE_SIZE = 512
 
 examples = [
     "A border collie lying in some Fall leaves as the forest trees change colors",
@@ -27,13 +26,13 @@ examples = [
 ]
 
 
-@spaces.GPU(duration=30) 
+@spaces.GPU(duration=30)
 def infer(
     prompt,
     width=512,
     height=512,
-    num_inference_steps=20, 
-    guidance_scale=7.5,  
+    num_inference_steps=20,
+    guidance_scale=7.5,
 ):
     # Генерация изображения
     image = pipe(
@@ -49,15 +48,16 @@ def infer(
 class CustomTheme(gr.themes.Base):
     def __init__(self):
         super().__init__()
-        self.primary_hue = "#5271FF"
-        self.background_fill_primary = "#AEB3B8"
-        self.background_fill_secondary = "#AEB3B8"
-        self.background_fill_tertiary = "#AEB3B8"
-        self.text_color_primary = "#AEB3B8"
-        self.text_color_secondary = "#AEB3B8"
-        self.text_color_tertiary = "#AEB3B8"
-        self.input_background_fill = "#AEB3B8"  
-        self.input_text_color = "#AEB3B8"       
+        self.primary_hue = "#17181B"
+        self.background_fill_primary = "#17181B"
+        self.background_fill_secondary = "#17181B"
+        self.background_fill_tertiary = "#17181B"
+        self.text_color_primary = "#FFFFFF"
+        self.text_color_secondary = "#FFFFFF"
+        self.text_color_tertiary = "#FFFFFF"
+        self.input_background_fill = "#17181B"
+        self.input_text_color = "#FFFFFF"
+
 
 css = """
 /* Скрываем нижний колонтитул */
@@ -68,30 +68,41 @@ footer {
     padding: 0;
     overflow: hidden;
 }
-
-
 @import url('https://fonts.googleapis.com/css2?family=Poppins:wght@400;500;700&display=swap');
+
 /* Применяем шрифты */
 body, input, button, textarea, select, .gr-button {
     font-family: 'Poppins', sans-serif;
+    background-color: #17181B !important;
+    color: #FFFFFF;
 }
+
 /* Настройки заголовков */
 h1, h2, h3, h4, h5, h6 {
     font-family: 'Poppins', sans-serif;
     font-weight: 700;
+    color: #FFFFFF;
 }
 
+/* Стиль для текстовых полей и кнопок */
+input[type="text"], textarea {
+    background-color: #17181B !important;
+    color: #FFFFFF;
+    border: 1px solid #FFFFFF;
+}
 
-/* Дополнительный стиль для слайдеров и чекбоксов */
 input[type="range"]::-webkit-slider-thumb {
     background: #5271FF;
 }
+
 input[type="range"]::-moz-range-thumb {
     background: #5271FF;
 }
+
 input[type="range"]::-ms-thumb {
     background: #5271FF;
 }
+
 input[type="checkbox"]:checked {
     background-color: #5271FF;
 }
@@ -99,13 +110,12 @@ input[type="checkbox"]:checked {
 /* Выделяем текст для Prompt */
 .prompt-text {
     font-weight: bold;
+    color: #FFFFFF;
 }
 """
 
-
 with gr.Blocks(theme=CustomTheme(), css=css) as demo:
     with gr.Column(elem_id="col-container"):
-
         gr.Markdown("**Prompt**", elem_classes="prompt-text")
 
         with gr.Row():
@@ -121,22 +131,20 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
                 "Generate",
                 scale=0,
                 variant="primary",
-                elem_classes="generate-button"
+                elem_classes="generate-button",
             )
 
         result = gr.Image(label="Result", show_label=False)
 
-
         gr.Examples(
             examples=examples,
             inputs=[prompt],
             outputs=[result],
             fn=infer,
             cache_examples=True,
-            cache_mode="lazy"
+            cache_mode="lazy",
         )
 
-
         run_button.click(
             fn=infer,
             inputs=[prompt],
@@ -148,5 +156,5 @@ if __name__ == "__main__":
         server_name="0.0.0.0",
         server_port=7860,
         share=True,
-        show_api=False
+        show_api=False,
     )