Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import PyPDF2
+import re
 
 def extrair_texto_pdf():
     with open("dados-cdu.pdf", "rb") as pdf_file:
@@ -23,40 +24,28 @@ def gerar_cdu(palavras_chave):
     resultados = {}
     for palavra_chave in palavras_chave:
         cdu_encontrada = consultar_cdu(palavra_chave)
-        resultados[palavra_chave] = cdu_encontrada
-
-    # Formatação dos resultados
-    resultados_formatados = {}
-    for palavra_chave, cdu_encontrada in resultados.items():
-        cdu_formatada = [cdu.replace("{}", "").strip() for cdu in cdu_encontrada]
-        resultados_formatados[palavra_chave] = cdu_formatada
-
-    return resultados_formatados
-
+        cdu_formatada = [re.search(r"\[(.*?)\]", cdu).group(1) for cdu in cdu_encontrada]
+        resultados[palavra_chave] = cdu_formatada
+    return resultados
 
 # Interface Gradio
-input_text = gr.inputs.Textbox(label="Inserir o assunto da sua obra (Ex.: Geografia cultural)")
+input_text = gr.inputs.Textbox(label="Inserir palavras-chave separadas por vírgulas")
 button_label = "Gerar CDU"
-output_text = gr.outputs.Textbox(label="Resultados CDU:")
+output_text = gr.outputs.Textbox(label="Resultado")
 
 def generate_cdu(palavras_chave):
     if palavras_chave:
         resultados = gerar_cdu(palavras_chave)
         formatted_resultados = {}
-        for cdu_encontrada in resultados.items():
-            formatted_resultados[palavra_chave] = (["CDU: " + cdu for cdu in cdu_encontrada])
+        for palavra_chave, cdu_encontrada in resultados.items():
+            formatted_resultados[palavra_chave] = "\n".join(["CDU: " + cdu for cdu in cdu_encontrada])
         return formatted_resultados
     else:
         return {}
 
-#def generate_cdu(palavras_chave):
-    #if palavras_chave:
-        #resultados = gerar_cdu(palavras_chave)
-        #return resultados
-    #else:
-        #return {}
-
 title = "Gerador de CDU"
-description = "Insira sua palavra chave ou sua área, por exemplo: Geografia cultural e clique em 'ENVIAR' para obter as opções classificação CDU relacionada."
+description = "Insira palavras-chave separadas por vírgulas e clique em 'Gerar CDU' para obter a classificação relacionada."
+
+gr.Interface(fn=generate_cdu, inputs=input_text, outputs=output_text, title=title, description=description).launch()
+
 
-gr.Interface(fn=generate_cdu, inputs=input_text, outputs=output_text, title=title, description=description).launch()