Add security

app.py

@@ -3,6 +3,22 @@ from io import BytesIO
 from fastapi import FastAPI, Form
 from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import StreamingResponse, JSONResponse
+from fastapi import FastAPI, Depends, HTTPException, status
+from fastapi.security.api_key import APIKeyHeader
+
+API_KEY = "tu_super_secreta_clave_de_api"
+api_key_header = APIKeyHeader(name="X-API-Key", auto_error=False)
+
+async def get_api_key(api_key: str = Depends(api_key_header)):
+    if not api_key or api_key != API_KEY:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="Credenciales de autenticación inválidas",
+        )
+    return api_key
+
+#Y manda ésto en los headers:
+#"X-API-Key": apiKey
 
 app = FastAPI()
 
@@ -23,7 +39,10 @@ async def health_check():
           description="Test endpoint que recibe y regresa la misma imagen, para probar envío, recepción y problemas con api o red.",
           summary="Summary"
           )
-async def echo_image(image: UploadFile = File(...)):
+async def echo_image(
+                     image: UploadFile = File(...), 
+                     #api_key: str = Depends(get_api_key)
+                     ):
     if not image.content_type.startswith("image/"):
         return {"error": "El archivo no es una imagen"}
     contents = await image.read()

campos/ine_actual.py

@@ -2,20 +2,23 @@ import herramientas
 import tools.ine_actual as ine_actual_tools
 import time
 
-def vigencia(textos_extraidos, textos_extraidos_limpios):
+def vigencia(textos_extraidos_limpios):
 
     conceptos_busqueda = [
     {'preferida': 'vigencia', 'alternativas': ['igencia']},   
     ]
 
     indice, textos_extraidos_corregidos = herramientas.buscaIndexMultiplesPalabras(textos_extraidos_limpios, conceptos_busqueda)
-
     textos_extraidos_limpios = textos_extraidos_corregidos
-    print("El índice de vigencia es: ", indice)
+
+    if indice is None:            
+            return None
     
-    texto_vigencia = textos_extraidos_limpios[indice]
-    print("Y su correspondiente texto es: " , texto_vigencia)    
+    print("El índice de vigencia es: ", indice)
 
     vigencia = ine_actual_tools.obtener_vigencia(indice, textos_extraidos_limpios) 
-    
-    return vigencia
+
+    if vigencia is not None:            
+            return vigencia
+    else: 
+          return None   

campos/ine_antes.py

@@ -1,20 +1,24 @@
 import herramientas
 import tools.ine_antes as ine_antes_tools
 
-def vigencia(textos_extraidos, textos_extraidos_limpios):
+def vigencia(textos_extraidos_limpios):
 
     conceptos_busqueda = [
-    {'preferida': 'vigencia', 'alternativas': ['igencia']},   
+    {'preferida': 'vigencia', 'alternativas': ['igencia', 'vigenca']},   
     ]
 
     indice, textos_extraidos_corregidos = herramientas.buscaIndexMultiplesPalabras(textos_extraidos_limpios, conceptos_busqueda)
-
     textos_extraidos_limpios = textos_extraidos_corregidos
-    print("El índice de genero es: ", indice)
 
-    texto_genero = textos_extraidos_limpios[indice]
-    print("Y su correspondiente texto es: " , texto_genero)
+    if indice is None:            
+            return None    
 
-    vigencia = ine_antes_tools.obtener_vigencia(texto_genero) 
+    print("El índice de genero es: ", indice) 
+    texto = textos_extraidos_limpios[indice]   
+    
+    vigencia = ine_antes_tools.obtener_vigencia(texto) 
     
-    return vigencia
+    if vigencia is not None:            
+        return vigencia
+    else: 
+        return None   

conceptos_busqueda.py

@@ -8,7 +8,10 @@ pasaporte = [
         #{'preferida': 'identidad', 'alternativas': []}    
         ]
 
-ine = [
-        {'preferida': 'Instituto Nacional Electoral', 'alternativas': ['']},
-        {'preferida': 'MEXICO', 'alternativas': ['']}    
-        ]
+ine = [{
+        'preferida': 'Instituto Nacional Electoral', 
+        #'alternativas': ['']
+         },
+        {'preferida': 'MEXICO', 
+         #'alternativas': [''] #Cuando no hay altarnativas, tienen que ser comentadas.
+        }]

documentos.py

@@ -27,17 +27,20 @@ async def pasaporte(textos_extraidos, textos_extraidos_simplificados):
     return nombre, apellido, identificacion
 
 #Funciones para Zurich México
-
-async def ine_actual(textos_extraidos, textos_extraidos_simplificados):
-    
+async def ine_actual(textos_extraidos_simplificados):    
     #Campos Buscados
-    vigencia = doc_ine_actual.vigencia(textos_extraidos, textos_extraidos_simplificados)
+    vigencia = doc_ine_actual.vigencia(textos_extraidos_simplificados)
 
-    return vigencia
+    if vigencia is None:
+        return "Vigencia no localizada"
+    else:
+        return vigencia
 
-async def ine_antes(textos_extraidos, textos_extraidos_simplificados):
-    
+async def ine_antes(textos_extraidos_simplificados):    
     #Campos Buscados
-    vigencia = doc_ine_antes.vigencia(textos_extraidos, textos_extraidos_simplificados)
+    vigencia = doc_ine_antes.vigencia(textos_extraidos_simplificados)
 
-    return vigencia
+    if vigencia is None:
+        return "Vigencia no localizada"
+    else:
+        return vigencia

funciones.py

@@ -72,9 +72,13 @@ async def procesa_ine(image):
 		tipo_ine = identificador.identifica_ine(textos_extraidos_simplificados)
 		print("Tipo ine es : ", tipo_ine)
 		if tipo_ine == 'ine_actual':
-			vigencia = await documentos.ine_actual(textos_extraidos, textos_extraidos_simplificados)
+			vigencia = await documentos.ine_actual(textos_extraidos_simplificados)
 		else:
-			vigencia = await documentos.ine_antes(textos_extraidos, textos_extraidos_simplificados)
+			vigencia = await documentos.ine_antes(textos_extraidos_simplificados)
 		return {
+		"nombre": "-----",
+		"domicilio": "-----",
+		"clave_elector": "-----",
+		"sexo": "-----",	
 		"vigencia": vigencia,
 	}

herramientas.py

@@ -118,7 +118,6 @@ def buscaIndexMultiplesPalabras(arreglo_textos, conceptos_requeridos):
                 if forma in texto_linea_lower:
                     print("Encontrado línea es True!")
                     encontrado_en_esta_linea = True
-                    time.sleep(7)
                     
                     # Guardamos qué forma se encontró y cuál es la preferida
                     palabras_encontradas_en_linea[forma] = concepto['preferida']
@@ -127,11 +126,9 @@ def buscaIndexMultiplesPalabras(arreglo_textos, conceptos_requeridos):
                     # Hemos encontrado una forma para este concepto, pasamos al siguiente concepto
                     break #Éste break entonces si se sale del for. De no hacerlo, volvería arriba.
 
-            print("Paso if encontrado_en_esta_linea es True...")
             if encontrado_en_esta_linea:
                 es_coincidencia_completa = True
                 print(f"Línea {i}: Concepto '{concepto['preferida']}' : Encontrado.")
-                print("Estoy por hacer break por que es_coincidencia_completa es True...")
                 break # Se sale del for grande.
             
             

tools/ine_actual.py

@@ -11,4 +11,7 @@ def obtener_vigencia(indice, textos_extraidos_libres):
     dato_vigencia = textos_extraidos_libres[indice+3]
     print("El dato numérico de la vigencia es: " , dato_vigencia)
 
-    return dato_vigencia
+    if dato_vigencia is not None:            
+            return dato_vigencia
+    else: 
+          return None