Update app.py

app.py

@@ -13,20 +13,18 @@ from functools import partial
 from tsr.system import TSR
 from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orientation
 
-#HF_TOKEN = os.getenv("HF_TOKEN")
+# HF_TOKEN = os.getenv("HF_TOKEN")
 
 HEADER = """
-
 """
 
-
 if torch.cuda.is_available():
     device = "cuda:0"
 else:
     device = "cpu"
 
 d = os.environ.get("DEVICE", None)
-if d != None:
+if d!= None:
     device = d 
 
 model = TSR.from_pretrained(
@@ -80,56 +78,109 @@ def run_example(image_pil):
     mesh_name, mesn_name2 = generate(preprocessed)
     return preprocessed, mesh_name, mesh_name2
 
-with gr.Blocks() as demo:
-    gr.Markdown(HEADER)
-    with gr.Row(variant="panel"):
-        with gr.Column():
-            with gr.Row():
-                input_image = gr.Image(
-                    label="Input Image",
-                    image_mode="RGBA",
-                    sources="upload",
-                    type="pil",
-                    elem_id="content_image",
-                )
-                processed_image = gr.Image(label="Processed Image", interactive=False)
-            with gr.Row():
-                with gr.Group():
-                    do_remove_background = gr.Checkbox(
-                        label="Remove Background", value=True
-                    )
-                    foreground_ratio = gr.Slider(
-                        label="Foreground Ratio",
-                        minimum=0.5,
-                        maximum=1.0,
-                        value=0.85,
-                        step=0.05,
-                    )
-            with gr.Row():
-                submit = gr.Button("Generate", elem_id="generate", variant="primary")
-        with gr.Column():
-            with gr.Tab("obj"):
-                output_model = gr.Model3D(
-                    label="Output Model",
-                    interactive=False,
+class CustomTheme(gr.themes.Base):
+    def __init__(self):
+        super().__init__()
+        self.primary_hue = "#191a1e"
+        self.background_fill_primary = "#191a1e"
+        self.background_fill_secondary = "#191a1e"
+        self.background_fill_tertiary = "#191a1e"
+        self.text_color_primary = "#FFFFFF"
+        self.text_color_secondary = "#FFFFFF"
+        self.text_color_tertiary = "#FFFFFF"
+        self.input_background_fill = "#191a1e"
+        self.input_text_color = "#FFFFFF"
+
+css = """
+/* Скрываем нижний колонтитул */
+footer {
+    visibility: hidden;
+    height: 0;
+    margin: 0;
+    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: #191a1e!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: #191a1e!important;
+    color: #FFFFFF;
+    border: 1px solid #FFFFFF;
+}
+/* Цвет кнопки Generate */
+.generate-button {
+    background-color: #5271FF!important;
+    color: #FFFFFF!important;
+    border: none;
+    font-weight: bold;
+}
+.generate-button:hover {
+    background-color: #405BBF!important; /* Цвет при наведении */
+}
+/* Выделяем текст для 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("**Input Image**", elem_classes="prompt-text")
+
+        with gr.Row():
+            input_image = gr.Image(
+                label="",
+                image_mode="RGBA",
+                sources="upload",
+                type="pil",
+                container=False,
+            )
+            with gr.Group():
+                do_remove_background = gr.Checkbox(
+                    label="Remove Background", value=True
                 )
-            with gr.Tab("glb"):
-                output_model2 = gr.Model3D(
-                    label="Output Model",
-                    interactive=False,
+                foreground_ratio = gr.Slider(
+                    label="Foreground Ratio",
+                    minimum=0.5,
+                    maximum=1.0,
+                    value=0.85,
+                    step=0.05,
                 )
-    # with gr.Row(variant="panel"):
-    #     gr.Examples(
-    #         examples=[
-    #             os.path.join("examples", img_name) for img_name in sorted(os.listdir("examples"))
-    #         ],
-    #         inputs=[input_image],
-    #         outputs=[processed_image, output_model, output_model2],
-    #         #cache_examples=True,
-    #         fn=partial(run_example),
-    #         label="Examples",
-    #         examples_per_page=20
-    #     )
+
+        processed_image = gr.Image(label="Processed Image", interactive=False)
+
+        with gr.Row():
+            submit = gr.Button(
+                "Generate",
+                scale=0,
+                variant="primary",
+                elem_classes="generate-button",
+            )
+
+        with gr.Tab("obj"):
+            output_model = gr.Model3D(
+                label="Output Model",
+                interactive=False,
+            )
+        with gr.Tab("glb"):
+            output_model2 = gr.Model3D(
+                label="Output Model",
+                interactive=False,
+            )
+
     submit.click(fn=check_input_image, inputs=[input_image]).success(
         fn=preprocess,
         inputs=[input_image, do_remove_background, foreground_ratio],
@@ -141,4 +192,4 @@ with gr.Blocks() as demo:
     )
 
 demo.queue(max_size=10)
-demo.launch()
+demo.launch()