change schemas

texttovoice/views.py

@@ -8,12 +8,37 @@ from TTS.api import TTS
 from rest_framework.authentication import TokenAuthentication
 from rest_framework.permissions import IsAuthenticated
 from .serializers import TextToSpeechSerializer
+from rest_framework.parsers import MultiPartParser
+from drf_yasg import openapi
+from drf_yasg.utils import swagger_auto_schema
+
 
 class TextToSpeechCreateView(CreateAPIView):
     serializer_class = TextToSpeechSerializer
     authentication_classes = [TokenAuthentication]  # Apply token authentication
     permission_classes = [IsAuthenticated]  # Require authentication for this view
 
+    parser_classes = [MultiPartParser]
+
+    @swagger_auto_schema(
+            operation_id='Create a document',
+            operation_description='Create a document by providing file and s3_key',
+            manual_parameters=[
+                openapi.Parameter('file', openapi.IN_FORM, type=openapi.TYPE_FILE, description='Document to be uploaded'),
+                openapi.Parameter('s3_key', openapi.IN_FORM, type=openapi.TYPE_STRING, description='S3 Key of the Document '
+                                                                                                   '(folders along with name)')
+            ],
+            responses={
+                status.HTTP_200_OK: openapi.Response(
+                    'Success', schema=openapi.Schema(type=openapi.TYPE_OBJECT, properties={
+                        'doc_id': openapi.Schema(type=openapi.TYPE_STRING, description='Document ID'),
+                        'mime_type': openapi.Schema(type=openapi.TYPE_STRING, description='Mime Type of the Document'),
+                        'version_id': openapi.Schema(type=openapi.TYPE_STRING, description='S3 version ID of the document')
+                    })
+                )
+            }
+        )
+
     def create(self, request, *args, **kwargs):
         serializer = self.get_serializer(data=request.data)
         if serializer.is_valid():