Update app.py

app.py

@@ -24,7 +24,11 @@ def gerar_cdu(palavras_chave):
     resultados = {}
     for palavra_chave in palavras_chave:
         cdu_encontrada = consultar_cdu(palavra_chave)
-        cdu_formatada = [re.search(r"\[(.*?)\]", cdu).group(1) for cdu in cdu_encontrada]
+        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
     return resultados
 
@@ -38,7 +42,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: " + cdu for cdu in cdu_encontrada])
+            formatted_resultados[palavra_chave] = "\n".join(cdu_encontrada)
         return formatted_resultados
     else:
         return {}
@@ -46,6 +50,5 @@ 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."
 
-gr.Interface(fn=generate_cdu, inputs=input_text, outputs=output_text, title=title, description=description).launch()
-
-
+iface = gr.Interface(fn=generate_cdu, inputs=input_text, outputs=output_text, title=title, description=description)
+iface.launch()