Update app.py

app.py

@@ -38,7 +38,6 @@ model = SF3D.from_pretrained(
 )
 model.eval().cuda()
 
-# Load example files
 example_files = [
     os.path.join("demo_files/examples", f) for f in os.listdir("demo_files/examples")
 ]
@@ -52,7 +51,6 @@ def run_model(input_image):
             trimesh_mesh, _glob_dict = model.generate_mesh(model_batch, 1024)
             trimesh_mesh = trimesh_mesh[0]
 
-    # Create new temporary file
     tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".glb")
     trimesh_mesh.export(tmp_file.name, file_type="glb", include_normals=True)
 
@@ -80,7 +78,6 @@ def create_batch(input_image: Image) -> dict[str, Any]:
         "intrinsic_cond": intrinsic.unsqueeze(0),
         "intrinsic_normed_cond": intrinsic_normed_cond.unsqueeze(0),
     }
-    # Add batch dimension
     batched = {k: v.unsqueeze(0) for k, v in batch_elem.items()}
     return batched
 
@@ -89,7 +86,6 @@ def checkerboard(squares: int, size: int, min_value: float = 0.5):
     base = np.zeros((squares, squares)) + min_value
     base[1::2, ::2] = 1
     base[::2, 1::2] = 1
-
     repeat_mult = size // squares
     return (
         base.repeat(repeat_mult, axis=0)
@@ -100,22 +96,12 @@ def checkerboard(squares: int, size: int, min_value: float = 0.5):
 def remove_background(input_image: Image) -> Image:
     return rembg.remove(input_image, session=rembg_session)
 
-def resize_foreground(
-    image: Image,
-    ratio: float,
-) -> Image:
+def resize_foreground(image: Image, ratio: float) -> Image:
     image = np.array(image)
     assert image.shape[-1] == 4
     alpha = np.where(image[..., 3] > 0)
-    y1, y2, x1, x2 = (
-        alpha[0].min(),
-        alpha[0].max(),
-        alpha[1].min(),
-        alpha[1].max(),
-    )
-    # Crop the foreground
+    y1, y2, x1, x2 = alpha[0].min(), alpha[0].max(), alpha[1].min(), alpha[1].max()
     fg = image[y1:y2, x1:x2]
-    # Pad to square
     size = max(fg.shape[0], fg.shape[1])
     ph0, pw0 = (size - fg.shape[0]) // 2, (size - fg.shape[1]) // 2
     ph1, pw1 = size - fg.shape[0] - ph0, size - fg.shape[1] - pw0
@@ -126,7 +112,6 @@ def resize_foreground(
         constant_values=0,
     )
 
-    # Compute padding according to the ratio
     new_size = int(new_image.shape[0] / ratio)
     ph0, pw0 = (new_size - size) // 2, (new_size - size) // 2
     ph1, pw1 = new_size - size - ph0, new_size - size - pw0
@@ -136,9 +121,7 @@ def resize_foreground(
         mode="constant",
         constant_values=0,
     )
-    new_image = Image.fromarray(new_image, mode="RGBA").resize(
-        (COND_WIDTH, COND_HEIGHT)
-    )
+    new_image = Image.fromarray(new_image, mode="RGBA").resize((COND_WIDTH, COND_HEIGHT))
     return new_image
 
 def square_crop(input_image: Image) -> Image:
@@ -147,9 +130,7 @@ def square_crop(input_image: Image) -> Image:
     top = (input_image.size[1] - min_size) // 2
     right = (input_image.size[0] + min_size) // 2
     bottom = (input_image.size[1] + min_size) // 2
-    return input_image.crop((left, top, right, bottom)).resize(
-        (COND_WIDTH, COND_HEIGHT)
-    )
+    return input_image.crop((left, top, right, bottom)).resize((COND_WIDTH, COND_HEIGHT))
 
 def show_mask_img(input_image: Image) -> Image:
     img_numpy = np.array(input_image)
@@ -196,7 +177,6 @@ def requires_bg_remove(image, fr):
     min_alpha = alpha_channel.min()
 
     if min_alpha == 0:
-        print("Already has alpha")
         sqr_crop = square_crop(image)
         fr_res = resize_foreground(sqr_crop, fr)
         return (
@@ -223,7 +203,6 @@ def update_foreground_ratio(img_proc, fr):
         gr.update(value=show_mask_img(foreground_res)),
     )
 
-# Define custom theme
 class CustomTheme(gr.themes.Base):
     def __init__(self):
         super().__init__()
@@ -241,12 +220,13 @@ class CustomTheme(gr.themes.Base):
             "https://fonts.googleapis.com/css2?family=Poppins:wght@400;500;700&display=swap",
         )
 
-# Custom CSS
 css = """
-body {
-    background-color: #191a1e !important;
+html, body {
     margin: 0;
     padding: 0;
+    background-color: #191a1e !important;
+    height: 100vh;
+    width: 100vw;
 }
 
 :root {
@@ -255,12 +235,14 @@ body {
     --background-fill-tertiary: #191a1e !important;
 }
 
-/* Apply background to gradio container */
-.gradio-container {
+.gradio-container, .gr-app {
     background-color: #191a1e !important;
+    color: #FFFFFF !important;
+    height: 100%;
+    width: 100%;
 }
 
-/* Blocks background */
+/* Set blocks */
 .gr-block {
     background-color: #191a1e !important;
     border: none !important;
@@ -275,7 +257,6 @@ footer {
     overflow: hidden;
 }
 
-/* Apply fonts */
 body, input, button, textarea, select, .gr-button {
     font-family: 'Poppins', sans-serif;
     color: #FFFFFF;
@@ -288,12 +269,15 @@ h1, h2, h3, h4, h5, h6 {
     color: #FFFFFF;
 }
 
-/* Button styles */
+/* Increase button size */
 .generate-button {
     background-color: #5271FF !important;
     color: #FFFFFF !important;
     border: none;
     font-weight: bold;
+    font-size: 1.2em; /* Увеличиваем размер шрифта */
+    padding: 0.75em 2em; /* Увеличиваем отступы внутри кнопки */
+    border-radius: 0.3em; /* Небольшое скругление */
 }
 
 /* Examples container */
@@ -302,18 +286,17 @@ h1, h2, h3, h4, h5, h6 {
     border: none !important;
 }
 
-/* 3D Model container override if needed */
+/* 3D Model container if needed */
 .litmodel3d { 
     background-color: #191a1e !important;
 }
 """
 
-# Build the Gradio interface
 with gr.Blocks(theme=CustomTheme(), css=css) as demo:
     img_proc_state = gr.State()
     background_remove_state = gr.State()
 
-    with gr.Row():  # removed variant="panel"
+    with gr.Row():
         with gr.Column():
             with gr.Row():
                 input_img = gr.Image(