Update app.py

app.py

@@ -146,20 +146,19 @@ def run():
     """
 
     with gr.Blocks(css=custom_css) as demo:
-        with gr.Box(elem_classes=["main-container"]):
-            gr.Markdown(
-                """
-                <div class='title-container'>
-                    <h1>Talkie</h1>
-                    <div class='subtitle'>Upload an image, add some audio or text, and watch the magic happen! ✨</div>
-                </div>
-                """
-            )
-
-            # Imagen
-            input_image = gr.Image(label="📸 Your image")
-
-            # Audio + texto
+        with gr.Group(elem_id="main-container"):
+
+            # Encabezado
+            with gr.Row(elem_classes="title-container"):
+                gr.Markdown("""
+                <h1>Talkie</h1>
+                <div class='subtitle'>Upload an image, add some audio or text, and watch the magic happen! ✨</div>
+                """)
+
+            # Entradas
+            with gr.Row():
+                input_image = gr.Image(label="📸 Your image")
+
             with gr.Row():
                 input_audio = gr.Audio(label="🎵 Your audio (Optional)")
                 input_text = gr.Textbox(label="💭 Your text", placeholder="Type your text here...")
@@ -167,15 +166,14 @@ def run():
             # Botón
             btn = gr.Button("GENERATE", elem_classes=["generate-btn"])
 
-            # Salida de video
+            # Salida
             video_out = gr.Video(label="🎥 Your video", show_label=True, elem_classes=["output-video"])
 
-        # Conexión del botón (fuera del bloque visual)
+        # Evento del botón
         btn.click(run_inference, inputs=[input_image, input_audio, input_text], outputs=video_out)
 
         demo.queue()
         demo.launch()
 
-
 if __name__ == "__main__":
     run()