Files
orc_api/sumasen_easyocr/views.py

32 lines
1.2 KiB
Python
Raw Normal View History

2021-12-08 19:13:03 +05:30
from django.shortcuts import render
2021-12-07 13:25:14 +05:30
from rest_framework.views import APIView
from rest_framework.parsers import MultiPartParser
from rest_framework.response import Response
from rest_framework import status
import easyocr
2021-12-08 19:13:03 +05:30
def UploaderView(request):
return render(request, 'sumasen_easyocr/index.html')
2021-12-07 13:25:14 +05:30
class FileUploadView(APIView):
parser_classes = [MultiPartParser,]
def post(self, request, format='jpg'):
up_file = request.FILES['file']
2021-12-07 13:32:55 +05:30
destination = open('/home/sumasen/django_easyocr_api/orc_api/sumasen_easyocr/uploaded' + up_file.name, 'wb+')
2021-12-07 13:25:14 +05:30
for chunk in up_file.chunks():
destination.write(chunk)
destination.close()
try:
reader = easyocr.Reader(['ja','en'])
2021-12-07 13:32:55 +05:30
result = reader.readtext('/home/sumasen/django_easyocr_api/orc_api/sumasen_easyocr/uploaded' + up_file.name)
2021-12-07 13:25:14 +05:30
#print('@@@@@@@@@@')
#print(result)
#print('@@@@@@@@@@')
return Response(result, status.HTTP_201_CREATED)
#return Response({''}, status.HTTP_201_CREATED)
except Exception as e:
print(e)
return Response({'Error': "Error occured"}, status.HTTP_201_CREATED)