feat: adicionando possibilidade de enviar arquivo de PDF

endpoint_teste/serializer.py

@@ -8,7 +8,12 @@ class EndpointTesteSerializer(serializers.ModelSerializer):
         fields = "__all__"
     
 class TesteSerializer(serializers.Serializer):
-    system_prompt = serializers.CharField(required=True)
     history_chat = serializers.ListField(read_only=True, required=False)
+    system_prompt = serializers.CharField(required=True)
     user_message = serializers.CharField(required=True)
-    pdf_url = serializers.CharField(required=False)
+    pdf_url = serializers.CharField(required=False)
+
+class PDFUploadSerializer(serializers.Serializer):
+    file = serializers.FileField()
+    system_prompt = serializers.CharField(required=True)
+    user_message = serializers.CharField(required=True)

endpoint_teste/views.py

@@ -1,12 +1,13 @@
+import tempfile, os
 from rest_framework import viewsets, filters
 from django_filters.rest_framework import DjangoFilterBackend
 from endpoint_teste.models import EndpointTesteModel
-from endpoint_teste.serializer import EndpointTesteSerializer
+from endpoint_teste.serializer import EndpointTesteSerializer, PDFUploadSerializer
 
 from rest_framework.decorators import api_view
 from rest_framework.response import Response
 
-from langchain_backend.main import get_llm_answer
+from langchain_backend.main import get_llm_answer, get_llm_answer_from_pdf_file
 from .serializer import TesteSerializer
 
 class EndpointTesteViewSet(viewsets.ModelViewSet):
@@ -30,4 +31,39 @@ def getTeste(request):
                 "Resposta": resposta_llm
             })
     if request.method == "GET":
-        return Response("Fazendo um GET")
+        return Response("Fazendo um GET")
+
+@api_view(["POST"])
+def getPDF(request):
+    if request.method == "POST":
+        serializer = PDFUploadSerializer(data=request.data)
+        if serializer.is_valid(raise_exception=True):
+            # Access the uploaded file
+            data = request.data
+            print('data: ', data)
+            pdf_file = serializer.validated_data['file']
+            pdf_file.seek(0)
+            # print(dir(pdf_file))
+            # print('pdf_file: ', pdf_file.read())
+            # pdf_content = pdf_file.read()
+            # Save the file or process it as needed
+            # For example, you can save it to a specific location
+            # with open(f'endpoint_teste/media/uploads/{pdf_file.name}', 'wb+') as destination:
+            #     for chunk in pdf_file.chunks():
+            #         destination.write(chunk)
+            # return Response({"message": "File uploaded successfully."})
+
+            # Create a temporary file to save the uploaded PDF
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_file:
+                # Write the uploaded file content to the temporary file
+                for chunk in pdf_file.chunks():
+                    temp_file.write(chunk)
+                temp_file_path = temp_file.name  # Get the path of the temporary file
+            print('temp_file_path: ', temp_file_path)
+            resposta_llm = get_llm_answer(data["system_prompt"], data["user_message"], temp_file_path)
+
+            os.remove(temp_file_path)
+
+            return Response({
+                "Resposta": resposta_llm
+            })

setup/urls.py

@@ -2,8 +2,7 @@ from django.contrib import admin
 from django.urls import path, include
 from rest_framework import routers
 
-from endpoint_teste.views import EndpointTesteViewSet
-from endpoint_teste.views import getTeste
+from endpoint_teste.views import EndpointTesteViewSet, getTeste, getPDF
 
 router = routers.DefaultRouter()
 # router.register("endpoint-teste", EndpointTesteViewSet, basename="Basename do endpoint-teste")
@@ -11,6 +10,6 @@ router = routers.DefaultRouter()
 urlpatterns = [
     path("admin/", admin.site.urls),
     path('', include(router.urls)),
-    
-    path("teste", getTeste)
+    path("teste", getTeste),
+    path('pdf', getPDF, name='upload-pdf')  
 ]