Update app.py

app.py

@@ -1,9 +1,6 @@
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
-import datetime
 import requests
-import pytz
-import yaml 
-import PyPDF2
+yaml
 import gradio as gr
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
@@ -26,38 +23,6 @@ def classify_alert(service_criticity: str, disruption: str, affectation_time: st
     else:
         return "P3 - Menor"
 
-# Herramienta para generar resúmenes de reportes de infraestructura
-@tool
-def summarize_report(report_text: str) -> str:
-    """Genera un resumen de un reporte sobre la infraestructura de la plataforma.
-    Args:
-        report_text: Texto del reporte.
-    """
-    api_url = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
-    headers = {"Authorization": "Bearer YOUR_HF_API_KEY"}
-    payload = {"inputs": report_text}
-    
-    response = requests.post(api_url, headers=headers, json=payload)
-    if response.status_code == 200:
-        return response.json()[0]['summary_text']
-    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(
@@ -74,7 +39,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, classify_alert, summarize_report, summarize_pdf_report],
+    tools=[final_answer, classify_alert],
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -83,16 +48,3 @@ agent = CodeAgent(
     description=None,
     prompt_templates=prompt_templates
 )
-
-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()
-
-