Agregar descarga automática de modelos al directorio raíz

app.py

@@ -1,6 +1,25 @@
 # Simple app.py for Face Detection
+import os
 import streamlit as st
 
+# Comprobar y descargar modelos necesarios
+try:
+    # Verificar si los archivos de modelo existen
+    model_files = [
+        "deploy.prototxt.txt",
+        "res10_300x300_ssd_iter_140000.caffemodel"
+    ]
+    
+    missing_files = [f for f in model_files if not os.path.exists(f)]
+    
+    if missing_files:
+        st.warning("Faltan archivos de modelo. Descargando...")
+        import download_models
+        download_models.main()
+        st.success("¡Archivos de modelo descargados correctamente!")
+except Exception as e:
+    st.error(f"Error al comprobar/descargar modelos: {e}")
+
 try:
     # Importar la aplicación principal
     print("Starting Face Detection Application...")

download_models.py

@@ -17,21 +17,15 @@ def download_file(url, save_path):
         return False
 
 def main():
-    # Crear directorio para modelos si no existe
-    models_dir = "models"
-    if not os.path.exists(models_dir):
-        os.makedirs(models_dir)
-        print(f"Directorio creado: {models_dir}")
-
     # URLs y rutas para los modelos necesarios
     models = [
         {
-            "url": "https://raw.githubusercontent.com/opencv/opencv/master/samples/dnn/face_detector/deploy.prototxt",
-            "path": os.path.join(models_dir, "deploy.prototxt")
+            "url": "https://raw.githubusercontent.com/sr6033/face-detection-with-OpenCV-and-DNN/master/deploy.prototxt.txt",
+            "path": "deploy.prototxt.txt"
         },
         {
-            "url": "https://raw.githubusercontent.com/opencv/opencv_3rdparty/dnn_samples_face_detector_20170830/res10_300x300_ssd_iter_140000.caffemodel",
-            "path": os.path.join(models_dir, "res10_300x300_ssd_iter_140000.caffemodel")
+            "url": "https://raw.githubusercontent.com/sr6033/face-detection-with-OpenCV-and-DNN/master/res10_300x300_ssd_iter_140000.caffemodel",
+            "path": "res10_300x300_ssd_iter_140000.caffemodel"
         }
     ]