Update app.py

app.py

@@ -20,8 +20,12 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
         tipo_peca (str): Tipo de peça jurídica a ser gerada (ex: habeas corpus, petição inicial).
     
     Retorna:
-        str: Peça jurídica gerada.
+        str: Peça jurídica gerada com formatação adequada.
     """
+    # Verifica se o tipo de peça foi selecionado
+    if not tipo_peca:
+        return "Por favor, selecione o tipo de peça jurídica."
+
     mensagens = [
         {
             "role": "system",
@@ -43,7 +47,9 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
         max_tokens=1000
     )
 
-    return resposta.choices[0].message["content"]
+    # Formatação da peça jurídica
+    peca = resposta.choices[0].message["content"]
+    return f"**Peça Jurídica: {tipo_peca.capitalize()}**\n\n{peca}"
 
 # Função para redefinir os campos
 def redefinir_campos():
@@ -153,4 +159,4 @@ def gerar_interface_gradio():
 # Executa o app
 if __name__ == "__main__":
     app = gerar_interface_gradio()
-    app.launch()
+    app.launch()