Dall-E 로 생성한 이미지 파일 PostgreSQL 에 잘 저장되었는지 확인하는 방법

지난주에 Dall-E API 연결을 끝마치고, postman으로 prompt를 작성하여 이미지 url을 받아오는 것까지 확인했다. 생성된 이미지는 데이터베이스에 저장하도록 view에 작성해놓았는데, postgresql 에서 이 이미지들이 잘 저장되었는지 확인하려고 한다. 결국에는 이 이미지들을 post로 업로드하고 싶어, 그 과정을 진행하고 있다. 우선은 postgresql 에서 dall-e를 통해 생성된 이미지를 확인해보도록 하자.

 

우선, 지난번에 했던 것과 같이 table 이름을 알아봐야 한다. 

SELECT table_name
FROM information_schema.tables
WHERE table_schema = 'public';

 

 

28번째 table 에 posts_generatedimage 테이블이 있는걸 확인할 수 있다. 이 테이블은 백엔드 views.py에서 다음과 같이 설정해주었기 때문이다. 

class DalleAPIView(APIView):
    def post(self, request, *args, **kwargs):
        prompt = request.data.get('prompt')
        if not prompt:
            return Response({"error": "No prompt provided"}, status=400)

        image_url = generate_image(prompt)

        # 생성된 이미지 정보를 데이터베이스에 저장
        generated_image = GeneratedImage(prompt=prompt, image_url=image_url)
        generated_image.save()
        return Response({"image": str(image_url)})

 

이제, 다음 명령어를 입력하여 지금까지 생성된 이미지를 확인해보자.

SELECT * FROM posts_generatedimage;

 

이미지 id 와 prompt 내용, image_url 까지 확인할 수 있다. 

 

이제는 이 데이터베이스에 저장된 이미지들을 어떻게 가져와서 게시물로 Post할 수 있을지 알아보려고 한다. To be continued...