Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import PyPDF2
-import re
 
 def extrair_texto_pdf():
     with open("dados-cdu.pdf", "rb") as pdf_file:
@@ -24,12 +23,7 @@ def gerar_cdu(palavras_chave):
     resultados = {}
     for palavra_chave in palavras_chave:
         cdu_encontrada = consultar_cdu(palavra_chave)
-        cdu_formatada = []
-        for cdu in cdu_encontrada:
-            match = re.search(r"\[(.*?)\]", cdu)
-            if match:
-                cdu_formatada.append("CDU: " + match.group(1))
-        resultados[palavra_chave] = cdu_formatada
+        resultados[palavra_chave] = cdu_encontrada
     return resultados
 
 # Interface Gradio
@@ -42,7 +36,7 @@ def generate_cdu(palavras_chave):
         resultados = gerar_cdu(palavras_chave)
         formatted_resultados = {}
         for palavra_chave, cdu_encontrada in resultados.items():
-            formatted_resultados[palavra_chave] = "\n".join(cdu_encontrada)
+            formatted_resultados[palavra_chave] = "\n".join(["CDU: " + cdu for cdu in cdu_encontrada])
         return formatted_resultados
     else:
         return {}
@@ -50,5 +44,4 @@ def generate_cdu(palavras_chave):
 title = "Gerador de CDU"
 description = "Insira palavras-chave separadas por vírgulas e clique em 'Gerar CDU' para obter a classificação relacionada."
 
-iface = gr.Interface(fn=generate_cdu, inputs=input_text, outputs=output_text, title=title, description=description)
-iface.launch()
+gr.Interface(fn=generate_cdu, inputs=input_text, outputs=output_text, title=title, description=description).launch()