Update app.py

app.py

@@ -14,56 +14,108 @@ def app():
     }}
     .gr-row {{
         display: flex;
-        justify-content: left;
+        justify-content: center;
         align-items: center;
     }}
     .gr-column {{
+        display: flex;
+        flex-direction: column;
+        align-items: center;  /* Centramos todo en la columna */
         padding: 0 !important;
         margin: 0 !important;
-        flex: 0 0 auto !important;
-        width: {image_width}px !important;
+    }}
+    #centered-title {{
+        text-align: center;
+    }}
+    #centered-text {{
+        text-align: center;
+        margin-bottom: 10px;
+    }}
+    .custom-button {{
+        display: inline-block;
+        padding: 10px 20px;
+        font-size: 16px;
+        font-weight: bold;
+        color: white;
+        border-radius: 5px;
+        margin: 10px;
+        text-decoration: none;
+        text-align: center;
+    }}
+    .btn-grey {{
+        background-color: #4b4b4b;
+    }}
+    .btn-red {{
+        background-color: #e94e42;
     }}
     """) as demo:
-        gr.Markdown("## High Dynamic Range Modulo Imaging for Robust Object Detection in Autonomous Driving")
-        
-        gr.Markdown("### Authors:")
+        # Título completamente centrado
+        gr.Markdown("## Modulo Imaging for Computer Vision", elem_id="centered-title")
 
+        # Organizar los artículos en dos columnas con botones centrados bajo el título
         with gr.Row():
-            with gr.Column(elem_classes="gr-column"):
-                gr.Image("autores.png", show_download_button=False, elem_classes="fixed-size-image")
-                gr.Markdown("Kebin Contreras *, Brayan Monrroy ~ and Jorge Bacca ~")
-                gr.Markdown("*Department of Physics, Universidad del Cauca, ~Department of Computer Engineer, Universidad Industrial de Santander  ")
-                
-        
-            #with gr.Column():
-                #gr.Markdown("*Department of Physics, Universidad del Cauca ")
-                #gr.Markdown("~Department of Computer Engineer, Universidad Industrial de Santander ")
-        
-        # Agregar la funcionalidad original de imágenes y controles
-        with gr.Row():
+            # Primer artículo con su botón centrado en su columna
+            with gr.Column():
+                gr.Markdown("### High Dynamic Range Modulo Imaging for Robust Object Detection in Autonomous Driving", elem_id="centered-text")
+                gr.HTML('<a href="https://openreview.net/pdf?id=2GqZFx2I7s" target="_blank" class="custom-button btn-grey">Article</a>')
+            
+            # Segundo artículo con su botón centrado en su columna
             with gr.Column():
-                image = gr.Image(type="pil", label="Upload Image")
-                model_id = gr.Dropdown(label="Model", choices=["yolov10n", "yolov10s", "yolov10m", "yolov10b", "yolov10l", "yolov10x"], value="yolov10x")
-                image_size = gr.Slider(label="Image Size", minimum=320, maximum=1280, step=32, value=640)
-                conf_threshold = gr.Slider(label="Confidence Threshold", minimum=0.0, maximum=1.0, step=0.01, value=0.85)
-                correction = gr.Slider(label="Correction Factor", minimum=0, maximum=1.0, step=0.1, value=1.0)
-                sat_factor = gr.Slider(label="Saturation Factor", minimum=1.0, maximum=5.0, step=0.1, value=2.0)
-                kernel_size = gr.Slider(label="Blur Kernel Size", minimum=1, maximum=7, step=1, value=7)
-                DO = gr.Radio(label="DO", choices=["1", "2"], value="1")
-                t = gr.Slider(label="t", minimum=0.0, maximum=1.0, step=0.1, value=0.5)
-                vertical = gr.Radio(label="Vertical", choices=["True", "False"], value="True")
-                process_button = gr.Button("Process Image")
+                gr.Markdown("### Autoregressive High-Order Finite Difference Modulo Imaging: High-Dynamic Range for Computer Vision Applications", elem_id="centered-text")
+                gr.HTML('<a href="https://cvlai.net/aim/2024/" target="_blank" class="custom-button btn-red">Article</a>')
 
+        # Ajustes de sliders y controles eliminados, valores fijos
+        image = gr.Image(type="pil", label="Upload Image")
+        model_id = gr.Dropdown(label="Model", choices=["yolov10n", "yolov10s", "yolov10m", "yolov10b", "yolov10l", "yolov10x"], value="yolov10x")
+        
+        # Valores fijos
+        conf_threshold = 0.85
+        correction = 1.0
+        kernel_size = 7
+        DO = "1"
+        t = 0.7
+        vertical = "True"
+
+        # Mantener el control del nivel de saturación
+        sat_factor = gr.Slider(label="Saturation Factor", minimum=1.0, maximum=5.0, step=0.1, value=2.0)
+
+        process_button = gr.Button("Process Image")
+
+        # Mostrar las imágenes en dos columnas, dos imágenes por columna
+        with gr.Row():
             with gr.Column():
                 output_original = gr.Image(label="Original + blur")
-                output_clip = gr.Image(label="Clipped ")
+                output_clip = gr.Image(label="Clipped")
+            with gr.Column():
                 output_wrap = gr.Image(label="Wrapped")
                 output_recons = gr.Image(label="Reconstructed")
-                metrics_clip = gr.Textbox(label="Metrics for Clipped Image")
-                metrics_wrap = gr.Textbox(label="Metrics for Wrapped Image")
-                metrics_recons = gr.Textbox(label="Metrics for Reconstructed Image")
+
+        # Añadir botón que despliega cómo citar el artículo en formato BibTeX
+        def show_bibtex_citation():
+            bibtex_citation = '''@inproceedings{contreras2024modulo,
+  title={High Dynamic Range Modulo Imaging for Robust Object Detection in Autonomous Driving},
+  author={Contreras, Kebin and Monroy, Brayan and Bacca, Jorge},
+  booktitle={Autonomous Driving Conference},
+  year={2024},
+  address={Bucaramanga, Colombia},
+  publisher={Universidad Industrial de Santander}
+}'''
+            return bibtex_citation
+
+        # Añadir botón para copiar el texto
+        citation_button = gr.Button("How to Cite this Article (BibTeX)")
+        citation_text = gr.Textbox(visible=False, label="BibTeX Citation", interactive=False)
+        copy_button = gr.Button("Copy Citation", visible=False)
+
+        # Mostrar la cita en formato BibTeX y habilitar botón de copiar
+        def copy_citation(citation):
+            return gr.update(visible=True), citation
+
+        citation_button.click(show_bibtex_citation, outputs=[citation_text])
+        copy_button.click(lambda x: x, inputs=citation_text, outputs=citation_text)
 
     return demo
 
+
 if __name__ == "__main__":
     app().launch()