leonett commited on
Commit
4c2f9ec
verified
1 Parent(s): 32450de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -148,16 +148,19 @@ with gr.Blocks(title="Extracci贸n de Fotogramas Forenses") as demo:
148
 
149
  # Al cargar un video: si es v谩lido, se activa el bot贸n "Procesar Fotogramas" y se desactiva "DESCARGAR FOTOGRAMAS"
150
  def on_video_change(video):
151
- if video is None:
 
152
  return gr.Button.update(interactive=False), gr.Button.update(interactive=False)
153
- if isinstance(video, dict):
154
- original_name = video.get("name", "")
155
- else:
156
  original_name = os.path.basename(video)
 
 
157
  allowed_extensions = ('.mp4', '.avi', '.mov', '.mkv')
158
  if not original_name.lower().endswith(allowed_extensions):
159
  raise gr.Error("Solo se permiten archivos de video (mp4, avi, mov, mkv)")
160
- # Habilitar "Procesar Fotogramas" y deshabilitar "DESCARGAR FOTOGRAMAS"
161
  return gr.Button.update(interactive=True), gr.Button.update(interactive=False)
162
 
163
  video_input.change(
 
148
 
149
  # Al cargar un video: si es v谩lido, se activa el bot贸n "Procesar Fotogramas" y se desactiva "DESCARGAR FOTOGRAMAS"
150
  def on_video_change(video):
151
+ # Verificar que se haya cargado algo y que contenga informaci贸n
152
+ if not video:
153
  return gr.Button.update(interactive=False), gr.Button.update(interactive=False)
154
+ if isinstance(video, dict) and video.get("name"):
155
+ original_name = video["name"]
156
+ elif isinstance(video, str) and video:
157
  original_name = os.path.basename(video)
158
+ else:
159
+ return gr.Button.update(interactive=False), gr.Button.update(interactive=False)
160
  allowed_extensions = ('.mp4', '.avi', '.mov', '.mkv')
161
  if not original_name.lower().endswith(allowed_extensions):
162
  raise gr.Error("Solo se permiten archivos de video (mp4, avi, mov, mkv)")
163
+ # Al tener un video v谩lido, activa "Procesar Fotogramas" y desactiva "DESCARGAR FOTOGRAMAS"
164
  return gr.Button.update(interactive=True), gr.Button.update(interactive=False)
165
 
166
  video_input.change(