Update app.py

app.py

@@ -3,9 +3,13 @@ import base64
 import vertexai
 from vertexai.generative_models import GenerativeModel, Part, SafetySetting
 
-# Configuración del modelo y parámetros globales
-vertexai.init(project="deploygpt", location="us-central1")
+# Configuración de la clave de API de Google Cloud
+def configurar_api_key(api_key):
+    """Configura la API Key para Vertex AI."""
+    import os
+    os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = api_key
 
+# Configuración del modelo y parámetros globales
 generation_config = {
     "max_output_tokens": 8192,
     "temperature": 0,
@@ -31,9 +35,15 @@ safety_settings = [
     ),
 ]
 
-def revisar_examen(pdf_docente, pdf_alumno):
+def revisar_examen(api_key, pdf_docente, pdf_alumno):
     """Función principal para comparar los PDFs del docente y del alumno."""
     try:
+        # Configurar la API Key de Google Cloud
+        configurar_api_key(api_key)
+
+        # Inicializar Vertex AI
+        vertexai.init(project="deploygpt", location="us-central1")
+
         # Leer los datos de los PDFs desde la ruta proporcionada por Gradio
         with open(pdf_docente.name, "rb") as docente_file:
             docente_data = docente_file.read()
@@ -94,14 +104,14 @@ def revisar_examen(pdf_docente, pdf_alumno):
 interface = gr.Interface(
     fn=revisar_examen,
     inputs=[
+        gr.Textbox(label="API Key de Google Cloud", placeholder="Ingresa tu API Key aquí"),
         gr.File(label="PDF del Docente (Preguntas y Respuestas)"),
         gr.File(label="PDF del Alumno (Respuestas)")
     ],
     outputs=gr.Textbox(label="Retroalimentación del Examen"),
     title="Revisión Automática de Exámenes",
-    description="Sube el PDF del docente y el PDF del alumno para recibir una evaluación detallada."
+    description="Sube el PDF del docente y el PDF del alumno para recibir una evaluación detallada. Ingresa tu API Key de Google Cloud para autenticarte."
 )
 
 # Lanzar la interfaz
 interface.launch(debug=True)
-