Update app.py

app.py

@@ -6,9 +6,9 @@ import os
 import zipfile
 import hashlib
 import matplotlib.pyplot as plt
-import io
+import tempfile
 import logging
-from IPython.display import display, FileLink
+from pathlib import Path
 
 # Configuración de logs
 logging.basicConfig(level=logging.INFO)
@@ -23,8 +23,11 @@ def obtener_metadatos(imagen):
         
         metadata = {}
         for tag_id, value in exif_data.items():
-            tag = TAGS.get(tag_id, tag_id)
-            metadata[tag] = value
+            try:
+                tag = TAGS.get(tag_id, tag_id)
+                metadata[tag] = value
+            except Exception as e:
+                logger.debug(f"Error al procesar etiqueta EXIF: {str(e)}")
         return metadata
     except Exception as e:
         logger.error(f"Error al obtener metadatos: {str(e)}")
@@ -134,8 +137,8 @@ def realizar_ela(imagen, quality=95, scale=100):
 def procesar_imagen(archivo_imagen):
     """Procesa la imagen completa y genera resultados."""
     # Crear directorio temporal
-    temp_dir = "temp_evidence"
-    os.makedirs(temp_dir, exist_ok=True)
+    temp_dir = tempfile.TemporaryDirectory()
+    logger.info(f"Directorio temporal creado: {temp_dir.name}")
     
     try:
         # Cargar imagen
@@ -188,9 +191,9 @@ def procesar_imagen(archivo_imagen):
         
         # Guardar resultados
         nombre = os.path.splitext(os.path.basename(archivo_imagen))[0]
-        original_path = os.path.join(temp_dir, f"{nombre}_original.jpg")
-        ela_path = os.path.join(temp_dir, f"{nombre}_ela.jpg")
-        text_path = os.path.join(temp_dir, f"{nombre}_analisis.txt")
+        original_path = os.path.join(temp_dir.name, f"{nombre}_original.jpg")
+        ela_path = os.path.join(temp_dir.name, f"{nombre}_ela.jpg")
+        text_path = os.path.join(temp_dir.name, f"{nombre}_analisis.txt")
         
         img.save(original_path)
         cv2.imwrite(ela_path, realizar_ela(img))
@@ -199,23 +202,21 @@ def procesar_imagen(archivo_imagen):
             f.write(analysis_text)
         
         # Crear ZIP
-        zip_path = os.path.join(temp_dir, f"{nombre}_errorELA.zip")
+        zip_path = os.path.join(temp_dir.name, f"{nombre}_errorELA.zip")
         with zipfile.ZipFile(zip_path, "w") as zipf:
             zipf.write(original_path, os.path.basename(original_path))
             zipf.write(ela_path, os.path.basename(ela_path))
             zipf.write(text_path, os.path.basename(text_path))
         
-        # Limpiar archivos temporales
-        os.remove(original_path)
-        os.remove(ela_path)
-        os.remove(text_path)
-        
+        # Limpiar archivos temporales (Gradio se encarga de esto automáticamente)
         logger.info(f"Análisis completado. Archivo ZIP: {zip_path}")
-        return zip_path
+        
+        # Devolver el archivo ZIP como resultado
+        return zip_path, analysis_text
     
     except Exception as e:
         logger.error(f"Error en procesamiento: {str(e)}")
-        return f"Error: {str(e)}"
+        return f"Error: {str(e)}", f"Error al procesar la imagen: {str(e)}"
 
 # Interfaz Gradio
 with gr.Blocks(title="Análisis Forense de Imágenes con ELA", theme=gr.themes.Soft()) as demo:
@@ -256,9 +257,11 @@ with gr.Blocks(title="Análisis Forense de Imágenes con ELA", theme=gr.themes.S
 
 # Configuración de ejecución
 if __name__ == "__main__":
+    # Configuración para Hugging Face Spaces
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
         share=True,
-        inbrowser=True
+        inbrowser=True,
+        favicon_path="https://huggingface.co/datasets/huggingface/logo/resolve/main/hf-logo.png"
     )