feat: adicionando url.py em cada app e mais refatorações

_utils/handle_files.py

@@ -0,0 +1,23 @@
+import tempfile, os
+
+
+def handle_pdf_files_from_serializer(files):
+    listaPDFs = []
+    for file in files:
+        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)
+    return listaPDFs
+
+
+def remove_pdf_temp_files(listaPDFs):
+    for file in listaPDFs:
+        os.remove(file)

gerar_documento/urls.py

@@ -0,0 +1,11 @@
+from django.urls import path
+
+from gerar_documento.views import ResumoSimplesCursorCompletoView
+
+urlpatterns = [
+    path(
+        "gerar-documento",
+        ResumoSimplesCursorCompletoView.as_view(),
+        name="summary-cursor-completo-pdf",
+    ),
+]

gerar_documento/views.py

@@ -1,8 +1,8 @@
 from rest_framework.views import APIView
 from adrf.views import APIView as AsyncAPIView
-import tempfile, os
 from rest_framework.response import Response
 
+from _utils.handle_files import handle_pdf_files_from_serializer, remove_pdf_temp_files
 from _utils.resumo_completo_cursor import (
     get_llm_summary_answer_by_cursor_complete,
 )
@@ -11,6 +11,7 @@ from .serializer import (
 )
 from rest_framework.parsers import MultiPartParser
 from drf_spectacular.utils import extend_schema
+from datetime import datetime
 
 
 class ResumoSimplesCursorCompletoView(AsyncAPIView):
@@ -20,42 +21,19 @@ class ResumoSimplesCursorCompletoView(AsyncAPIView):
         request=ResumoCursorCompeltoSerializer,
     )
     async def post(self, request):
+        print(f"\n\nDATA E HORA DA REQUISIÇÃO: {datetime.now()}")
         serializer = ResumoCursorCompeltoSerializer(data=request.data)
         if serializer.is_valid(raise_exception=True):
-            print("\n\n\n")
-            listaPDFs = []
             data = serializer.validated_data
-            print("\nserializer.validated_data: ", serializer.validated_data)
-
-            for file in serializer.validated_data["files"]:
-                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)
+            print("\n\ndata: ", data)
+
+            listaPDFs = handle_pdf_files_from_serializer(data["files"])
 
             resposta_llm = await get_llm_summary_answer_by_cursor_complete(
                 data, listaPDFs
             )
+            print("\n\nresposta_llm: ", resposta_llm)
 
-            print("\n\n\n")
-            print("resposta_llm: ", resposta_llm)
-
-            final = resposta_llm
-            print("\n\n\n")
-            print("final: ", final)
-
-            for file in listaPDFs:
-                os.remove(file)
+            remove_pdf_temp_files(listaPDFs)
 
-            return Response({"resposta": final})
+            return Response({"resposta": resposta_llm})

modelos_usuarios/urls.py

@@ -0,0 +1,13 @@
+from django.urls import path
+
+from modelos_usuarios.views import (
+    ListCreateModeloUsuarioView,
+    CreateUpdateDeleteModeloUsuarioView,
+    ListModelosPorUsuarioView,
+)
+
+urlpatterns = [
+    path("modelo", ListCreateModeloUsuarioView.as_view()),
+    path("modelo/<int:pk>", CreateUpdateDeleteModeloUsuarioView.as_view()),
+    path("usuario/<int:user_id>/modelos", ListModelosPorUsuarioView.as_view()),
+]

ragas_api/urls.py

@@ -0,0 +1,8 @@
+from django.urls import path
+
+from ragas_api.views import RagasFromTextView, RagasView
+
+urlpatterns = [
+    path("ragas", RagasView.as_view()),
+    path("ragas/no-pdf", RagasFromTextView.as_view()),
+]

setup/settings.py

@@ -14,7 +14,6 @@ BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 # Quick-start development settings - unsuitable for production
 # See https://docs.djangoproject.com/en/4.2/howto/deployment/checklist/
 
-# SECURITY WARNING: keep the secret key used in production secret!
 SECRET_KEY = os.environ.get("SECRET_KEY")
 
 # SECURITY WARNING: don't run with debug turned on in production!

setup/urls.py

@@ -2,37 +2,24 @@ from django.contrib import admin
 from django.urls import path, include
 from rest_framework import routers
 from drf_spectacular.views import SpectacularSwaggerView, SpectacularAPIView
-
-
-from gerar_documento.views import ResumoSimplesCursorCompletoView
 from _antigos.resumos.views import (
     ResumoSimplesCursorView,
 )  # Não sei por quê, mas se remover esta importação, o endpoint de gerar_documentos para de funcionar
-from ragas_api.views import RagasFromTextView, RagasView
-from modelos_usuarios.views import (
-    ListCreateModeloUsuarioView,
-    CreateUpdateDeleteModeloUsuarioView,
-    ListModelosPorUsuarioView,
-)
 
 router = routers.DefaultRouter()
 # router.register("endpoint-teste", EndpointTesteViewSet, basename="Basename do endpoint-teste")
 
-urlpatterns = [
+config_urls = [
     path("api/schema/", SpectacularAPIView.as_view(), name="schema"),
     path(
         "swagger/", SpectacularSwaggerView.as_view(url_name="schema"), name="swagger-ui"
     ),
     path("admin/", admin.site.urls),
     path("", include(router.urls)),
-    path(
-        "gerar-documento",
-        ResumoSimplesCursorCompletoView.as_view(),
-        name="summary-cursor-completo-pdf",
-    ),
-    path("modelo", ListCreateModeloUsuarioView.as_view()),
-    path("modelo/<int:pk>", CreateUpdateDeleteModeloUsuarioView.as_view()),
-    path("usuario/<int:user_id>/modelos", ListModelosPorUsuarioView.as_view()),
-    path("ragas", RagasView.as_view()),
-    path("ragas/no-pdf", RagasFromTextView.as_view()),
+]
+
+urlpatterns = config_urls + [
+    path("", include("gerar_documento.urls")),
+    path("", include("ragas_api.urls")),
+    path("", include("modelos_usuarios.urls")),
 ]