Update apps/gradio_app.py

apps/gradio_app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import os
+import logging
 from gradio_app.config import setup_logging, setup_sys_path
 from gradio_app.processor import gradio_process, update_preview, update_visibility
 
@@ -7,8 +8,21 @@ from gradio_app.processor import gradio_process, update_preview, update_visibili
 setup_logging()
 setup_sys_path()
 
+# Configure logging
+logger = logging.getLogger(__name__)
+
 # Load custom CSS
-custom_css = open(os.path.join(os.path.dirname(__file__), "gradio_app", "static", "styles.css"), "r").read()
+custom_css = """
+.custom-table tr:hover { background-color: #f0f0f0; cursor: pointer; }
+.custom-table .selected { background-color: #d0e8ff; }
+.custom-file-input { margin-bottom: 10px; }
+.custom-radio { margin-bottom: 10px; }
+.custom-button { margin: 5px; }
+.custom-image, .custom-video { width: 100%; }
+.custom-textbox { width: 100%; }
+"""
+# Optionally, you can still load from styles.css if preferred
+# custom_css = open(os.path.join(os.path.dirname(__file__), "gradio_app", "static", "styles.css"), "r").read()
 
 # Path to examples directory
 examples_dir = os.path.join(os.path.dirname(__file__), "gradio_app", "assets", "examples", "license_plate_detector_ocr")
@@ -27,13 +41,27 @@ examples = [
     }
 ]
 
+# Validate example file paths
+for ex in examples:
+    if not os.path.exists(ex["Input File"]):
+        logger.error(f"Input file not found: {ex['Input File']}")
+    if not os.path.exists(ex["Output File"]):
+        logger.error(f"Output file not found: {ex['Output File']}")
+
 # Function to handle example selection
 def select_example(selected_row, examples_data):
-    if selected_row is not None and isinstance(selected_row, list) and len(selected_row) > 0:
-        row_index = selected_row[0]  # Get the index of the selected row
-        selected_example = examples[row_index]
-        return selected_example["Input File"], selected_example["Input Type"], None, None, None
-    return None, "Image", None, None, None  # Default return if no row is selected
+    logger.info(f"Selected row: {selected_row}")
+    if selected_row is not None:
+        try:
+            row_index = int(selected_row[0]) if isinstance(selected_row, list) else int(selected_row)
+            if 0 <= row_index < len(examples):
+                selected_example = examples[row_index]
+                input_file_path = selected_example["Input File"]
+                logger.info(f"Loading file: {input_file_path}")
+                return input_file_path, selected_example["Input Type"], None, None, None
+        except (IndexError, ValueError, TypeError) as e:
+            logger.error(f"Error selecting example: {e}")
+    return None, "Image", None, None, None  # Default return if selection fails
 
 # Gradio Interface
 with gr.Blocks(css=custom_css) as iface:
@@ -48,7 +76,7 @@ with gr.Blocks(css=custom_css) as iface:
 
     with gr.Row():
         with gr.Column(scale=1):
-            input_file = gr.File(label="Upload Image or Video", elem_classes="custom-file-input")
+            input_file = gr.File(label="Upload Image or Video", file_types=["image", "video"], elem_classes="custom-file-input")
             input_type = gr.Radio(choices=["Image", "Video"], label="Input Type", value="Image", elem_classes="custom-radio")
             with gr.Blocks():
                 input_preview_image = gr.Image(label="Input Preview", visible=True, elem_classes="custom-image")
@@ -60,7 +88,7 @@ with gr.Blocks(css=custom_css) as iface:
             # Examples table
             gr.Markdown("### Examples")
             examples_table = gr.Dataframe(
-                value=[[ex["Input File"], ex["Output File"], ex["Input Type"]] for ex in examples],
+                value=[[os.path.basename(ex["Input File"]), os.path.basename(ex["Output File"]), ex["Input Type"]] for ex in examples],
                 headers=["Input File", "Output File", "Input Type"],
                 interactive=True,  # Allow row selection
                 elem_classes="custom-table"
@@ -107,4 +135,4 @@ with gr.Blocks(css=custom_css) as iface:
     )
 
 if __name__ == "__main__":
-    iface.launch(share=True)
+    iface.launch()