feat: adicionado endpoint de resumo

langchain_backend/main.py

@@ -34,4 +34,13 @@ def get_llm_answer(system_prompt, user_prompt, pdf_url, model, embedding):
   vectorstore.delete_collection()
   utils.allIds = []
   print('utils.allIds: ', utils.allIds)
+  return results
+
+def get_llm_answer_summary(system_prompt, user_prompt, pdf_url, model):
+  print('model: ', model)
+  pages = getPDF(pdf_url)
+  rag_chain = create_prompt_llm_chain(system_prompt, model)
+
+  results = rag_chain.invoke({"input": user_prompt, "context": pages})
+
   return results

resumos/admin.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

resumos/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class ResumosConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'resumos'

resumos/models.py

@@ -0,0 +1,3 @@
+from django.db import models
+
+# Create your models here.

resumos/serializer.py

@@ -0,0 +1,24 @@
+from rest_framework import serializers
+
+# Exemplo de retorno que devo enviar ao frontend:
+# {
+#   "nome_do_memorial": "[b]Memorial de Defesa em Ação de Indenização por Danos Morais[/b]",
+#   "argumentos": "[b]Argumentos:[/b]\n[i]• Responsabilidade Civil do Réu:[/i] [i]O réu agiu de forma negligente ao causar o dano. Há dever de indenizar baseado no artigo 186 do Código Civil Brasileiro.[/i]\n[i]• Dano Moral Comprovado:[/i] [i]O dano sofrido pela parte autora é evidente, gerando sofrimento e abalo psicológico. O nexo causal entre a ação do réu e o dano sofrido é claro.[/i]",
+#   "jurisprudencia": "[b]Jurisprudência:[/b]\n[i]• STJ, REsp 123456/DF -[/i] [i]O Superior Tribunal de Justiça entendeu que a indenização por danos morais deve ser fixada de acordo com a gravidade do ato ilícito.[/i]\n[i]• STJ, REsp 654321/SP -[/i] [i]A jurisprudência confirma que o réu tem o dever de reparar integralmente o dano causado.[/i]",
+#   "doutrina": "[b]Doutrina:[/b]\n[i]• Carlos Roberto Gonçalves, Responsabilidade Civil -[/i] [i]A responsabilidade civil é objetiva quando há risco para os direitos da personalidade da vítima.[/i]\n[i]• Maria Helena Diniz, Curso de Direito Civil -[/i] [i]O dano moral é configurado pela violação dos direitos da personalidade.[/i]",
+#   "argumentos_faltantes": "[b]Argumentos Faltantes:[/b]\n[i]• Prova pericial de impacto psicológico.[/i]\n[i]• Estudo comparativo com casos análogos para quantificação do valor da indenização.[/i]",
+#   "palavras_chave": [
+#     "[i]Responsabilidade civil[/i]",
+#     "[i]Dano moral[/i]",
+#     "[i]Nexo causal[/i]",
+#     "[i]Indenização[/i]"
+#   ]
+# }
+
+# pecam para a AI formatar em BBcode
+
+class ResumoPDFSerializer(serializers.Serializer):
+    files = serializers.ListField(child=serializers.FileField(), required=True)
+    system_prompt = serializers.CharField(required=True)
+    user_message = serializers.CharField(required=False)
+    model = serializers.CharField(required=False)

resumos/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

resumos/views.py

@@ -0,0 +1,42 @@
+from rest_framework.views import APIView
+import tempfile, os
+from rest_framework.response import Response
+from .serializer import ResumoPDFSerializer
+from langchain_backend.main import get_llm_answer_summary
+from setup.environment import default_model
+from rest_framework.parsers import MultiPartParser
+from drf_spectacular.utils import extend_schema
+
+
+class ResumoView(APIView):
+  parser_classes = [MultiPartParser]
+
+  @extend_schema(
+    request=ResumoPDFSerializer,
+  )
+  def post(self, request):
+    serializer = ResumoPDFSerializer(data=request.data)
+    if serializer.is_valid(raise_exception=True):
+        listaPDFs = []
+        data = request.data
+        model = serializer.validated_data.get("model", default_model)
+        user_message = data.get("user_message", "")
+
+        for file in serializer.validated_data['files']:
+            print("file: ", file)
+            file.seek(0)
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_file: # Create a temporary file to save the uploaded PDF
+                for chunk in file.chunks(): # Write the uploaded file content to the temporary file
+                    temp_file.write(chunk)
+                temp_file_path = temp_file.name  # Get the path of the temporary file
+                listaPDFs.append(temp_file_path)
+            # print('listaPDFs: ', listaPDFs)
+
+        resposta_llm = get_llm_answer_summary(data["system_prompt"], user_message, listaPDFs, model=model)
+
+        for file in listaPDFs:
+            os.remove(file)
+
+        return Response({
+            resposta_llm
+        })

setup/settings.py

@@ -46,7 +46,8 @@ INSTALLED_APPS = [
     "django_filters",
     "corsheaders",
     "endpoint_teste",
-    "drf_spectacular"
+    "drf_spectacular",
+    "resumos"
 ]
 
 MIDDLEWARE = [

setup/urls.py

@@ -5,6 +5,7 @@ from drf_spectacular.views import SpectacularSwaggerView, SpectacularAPIView
 
 
 from endpoint_teste.views import EndpointTesteViewSet, getTeste, getPDF
+from resumos.views import ResumoView
 
 router = routers.DefaultRouter()
 # router.register("endpoint-teste", EndpointTesteViewSet, basename="Basename do endpoint-teste")
@@ -15,5 +16,6 @@ urlpatterns = [
     path("admin/", admin.site.urls),
     path('', include(router.urls)),
     path("teste", getTeste),
-    path('pdf', getPDF, name='upload-pdf')  
+    path('pdf', getPDF, name='upload-pdf'),
+    path('resumo', ResumoView.as_view(), name='summary-pdf')
 ]