Update app.py

app.py

@@ -44,7 +44,27 @@ Estamos mergulhando em alguns trechos de um processo judicial e queremos que voc
 O objetivo é criar um texto único e fluido que englobe todas essas informações de maneira compreensível para qualquer pessoa, mesmo aquelas com apenas a educação básica. Certifique-se de não adicionar informações novas e de não omitir nenhum nome mencionado nos trechos.
 """
 
-integro_prompt = PromptTemplate.from_template(reduce_template)
+resumo_template="""Sentenças do Caso: {docs}
+Este documento contém uma série de sentenças relacionados a um caso jurídico. Primeiramente, vamos identificar as partes envolvidas e compreender suas motivações. Em seguida, sintetizaremos os aspectos cruciais de cada resumo, preservando os detalhes essenciais e o contexto.
+
+Partes Envolvidas e Contexto:
+Identifique claramente quem são as pessoas ou entidades principais do caso e por que elas estão envolvidas, fornecendo um breve contexto.
+
+Análise e Síntese:
+Consolide os resumos examinando os pontos principais e as evidências apresentadas. Destaque elementos chave que são comuns ou significativos em toda a documentação.
+
+Texto Unificado:
+Com base nos elementos destacados, redija um resumo que combine todas as informações essenciais. Este deve incluir: 
+1. Relatório: um resumo do caso, apresentando os fatos principais.
+2. Fundamentação: as razões principais que suportam a decisão, explicando o raciocínio por trás das conclusões.
+3. Decisão: o veredito final ou as conclusões do caso.
+
+Lembre-se de manter a objetividade, sem adicionar informações não presentes nos originais. Evite jargões jurídicos complexos, mas mantenha a precisão dos termos quando necessário para a compreensão do caso. O objetivo é criar um resumo claro, direto e informativo.
+
+Instruções Finais:
+Reveja o texto para garantir que seja compreensível e que não omita informações críticas. O resumo final deve permitir a qualquer pessoa entender os aspectos fundamentais do caso jurídico sem necessidade de consulta aos textos originais.
+"""
+
 
 model_used = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 
@@ -70,18 +90,40 @@ def greet(uploaded_file, template_choice):
 
     if template_choice == "Template 2":
         prompt_template = PromptTemplate.from_template(integro_template)
+        llm = Together(
+          model=model_used,
+          temperature=0.5,
+          max_tokens=10000,
+          top_k=60,
+          together_api_key=token
+        )
     elif template_choice == "Template 1 (menos resumido)":
         prompt_template = PromptTemplate.from_template(sem_resumir_template)
+        llm = Together(
+          model=model_used,
+          temperature=0.5,
+          max_tokens=10000,
+          top_k=60,
+          together_api_key=token
+        )
+    elif template_choice == "Template 3 (Resumo Apenas)":
+        prompt_template = PromptTemplate.from_template(resumo_template)
+        llm = Together(
+          model=model_used,
+          temperature=0,
+          max_tokens=10000,
+          top_k=60,
+          together_api_key=token
+        )
     else:
         prompt_template = PromptTemplate.from_template(reduce_template)
-
-    llm = Together(
-      model=model_used,
-      temperature=0.5,
-      max_tokens=10000,
-      top_k=60,
-      together_api_key=token
-    )
+        llm = Together(
+          model=model_used,
+          temperature=0.5,
+          max_tokens=10000,
+          top_k=60,
+          together_api_key=token
+        )
     
     integro_chain = LLMChain(llm=llm, prompt=prompt_template)
     
@@ -92,7 +134,7 @@ def greet(uploaded_file, template_choice):
 iface = gr.Interface(
     fn=greet,
     inputs=[gr.File(file_types=['txt'], show_label=True),
-           gr.Radio(["Template 1 (mais resumido)", "Template 1 (menos resumido)", "Template 2"], label="Escolha o Template")],
+           gr.Radio(["Template 1 (mais resumido)", "Template 1 (menos resumido)", "Template 2", "Template 3 (Resumo Apenas)"], label="Escolha o Template")],
     outputs=["text"],
     description="Esta ferramenta simplifica textos jurídicos para uma linguagem mais acessível. Escolha um template e carregue um arquivo .txt com o texto que deseja processar.",
 )