Update app.py

app.py

@@ -17,20 +17,20 @@ class BasicAgent:
         # Cargar el modelo BERT preentrenado en SQuAD para tareas de Pregunta y Respuesta
         self.qa_pipeline = pipeline("question-answering", model="bert-large-uncased-whole-word-masking-finetuned-squad")
     
-    def __call__(self, question: str, context: str = None) -> str:
+    def __call__(self, question: str) -> str:
         """
         Procesa la pregunta y devuelve una respuesta basada en el contexto proporcionado.
         Si no se proporciona contexto, devuelve un mensaje de error.
         """
-        if context is None:
-            return "FINAL ANSWER: No context provided."
+        prompt = """You are a general AI assistant. I will ask you a question based on the provided context. 
+Please provide the answer in a clear and concise manner."""
         
         # Crear un prompt dentro del contexto que estructure la tarea más explícitamente
         prompt = f"""
 You are a general AI assistant. I will ask you a question based on the provided context. 
 Please provide the answer in a clear and concise manner.
 Question: {question}
-Context: {context}
+Context: {prompt}
 Answer:
 """