Update app.py

app.py

@@ -2,7 +2,9 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, t
 import datetime
 import requests
 import pytz
-import yaml
+yaml
+import PyPDF2
+import gradio as gr
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
@@ -41,6 +43,21 @@ def summarize_report(report_text: str) -> str:
     else:
         return "Error al generar el resumen."
 
+# Herramienta para extraer texto de un archivo PDF y generar un resumen
+@tool
+def summarize_pdf_report(pdf_path: str) -> str:
+    """Extrae texto de un archivo PDF y genera un resumen del contenido.
+    Args:
+        pdf_path: Ruta al archivo PDF.
+    """
+    try:
+        with open(pdf_path, "rb") as file:
+            reader = PyPDF2.PdfReader(file)
+            text = "\n".join([page.extract_text() for page in reader.pages if page.extract_text()])
+            return summarize_report(text)
+    except Exception as e:
+        return f"Error procesando el PDF: {str(e)}"
+
 final_answer = FinalAnswerTool()
 
 model = HfApiModel(
@@ -57,7 +74,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, classify_alert, summarize_report],
+    tools=[final_answer, classify_alert, summarize_report, summarize_pdf_report],
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -67,5 +84,15 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-GradioUI(agent).launch()
+def process_pdf(file):
+    return summarize_pdf_report(file.name)
+
+demo = gr.Interface(
+    fn=process_pdf,
+    inputs=gr.File(label="Sube un archivo PDF"),
+    outputs=gr.Textbox(label="Resumen del reporte")
+)
+
+demo.launch()
+