Merge Develop to Main by youyeon11 · Pull Request #11 · DearBelly/DearBelly-BE-FastAPI

youyeon11 · 2025-08-27T06:33:16Z

📌 작업 목적

여태까지의 작업을 main에 merge

Feat/add model (#2)

Feat/add api (#4)

Feat/redis job (#8)

gemini-code-assist

Summary of Changes

Hello @youyeon11, I'm Gemini Code Assist¹! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

이 PR은 개발 브랜치에서 메인 브랜치로의 병합을 통해 AI 기반 알약 식별 서비스의 핵심 기능을 도입합니다. 이 서비스는 FastAPI를 사용하여 구축되었으며, Redis 스트림을 통해 비동기 작업 처리를 관리하고, S3에서 이미지를 다운로드하며, PyTorch 모델을 사용하여 알약을 식별합니다. 이는 프로젝트의 아키텍처를 크게 업데이트하고 새로운 기능을 추가하는 중요한 변경 사항입니다.

Highlights

AI 기반 알약 식별 서비스 도입: FastAPI를 활용하여 이미지를 통한 알약 식별 기능을 제공합니다.
비동기 작업 처리 시스템 구축: Redis 스트림을 사용하여 이미지 처리 작업을 비동기적으로 관리하고, 작업 큐 및 결과 스트림을 통해 효율적인 처리를 가능하게 합니다.
S3 연동을 통한 이미지 처리: Presigned URL을 통해 S3에서 이미지를 다운로드하고 처리하는 기능을 구현했습니다.
PyTorch 모델 통합: LightCNN 모델을 사용하여 이미지에서 알약을 예측하고 식별하는 로직을 포함합니다.
프로젝트 구조 및 설정 업데이트: .github 이슈 템플릿, .gitignore 파일, README.md를 업데이트하고, .idea IDE 설정 파일을 제거하며, 새로운 app 디렉토리 구조로 파일을 재구성했습니다.

Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point in your pull request via creating an issue comment (i.e. comment on the pull request page) using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands.

Feature	Command	Description
Code Review	`/gemini review`	Performs a code review for the current pull request in its current state.
Pull Request Summary	`/gemini summary`	Provides a summary of the current pull request in its current state.
Comment	@gemini-code-assist	Responds in comments when explicitly tagged, both in issue comments and review comments.
Help	`/gemini help`	Displays a list of available commands.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for Github and other Google products, sign up here.

You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension.

Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution. ↩

gemini-code-assist

Code Review

이 PR은 기존 프로젝트 구조를 FastAPI 기반의 애플리케이션으로 재구성하고, Redis Stream을 이용한 비동기 작업 큐 시스템을 도입하는 등 많은 변경 사항을 포함하고 있습니다. 전반적으로 좋은 구조를 갖추고 있지만, 몇 가지 중요한 수정이 필요해 보입니다. 특히 모델 로딩 시의 보안 취약점, 워커 로직의 버그, 설정 관리 방식 등에서 개선이 필요합니다. 자세한 내용은 각 파일에 남긴 코멘트를 참고해 주세요.

gemini-code-assist · 2025-08-27T06:35:44Z

app/services/predictor_service.py

+    def _load_model(self, model_path: Path) -> LightCNN:
+        import __main__
+        __main__.LightCNN = LightCNN
+        model = torch.load(model_path, map_location=self.device, weights_only=False)


torch.load 함수에 weights_only=False를 사용하는 것은 심각한 보안 취약점을 유발할 수 있습니다. 신뢰할 수 없는 소스의 모델 파일을 로드할 경우 임의의 코드가 실행될 수 있습니다. 모델을 저장할 때 torch.save(model.state_dict(), PATH)와 같이 state_dict만 저장하고, 로드할 때는 model = TheModelClass(*args, **kwargs)로 모델 인스턴스를 먼저 생성한 뒤 model.load_state_dict(torch.load(PATH))를 사용하는 방식으로 변경하는 것이 안전하고 권장됩니다. 현재의 __main__ 의존성 문제도 함께 해결할 수 있습니다.

gemini-code-assist · 2025-08-27T06:35:44Z

app/worker/worker.py

+
+import asyncio
+import redis.asyncio as redis
+from app.core.config import Settings


Settings 클래스를 import하고 있지만, 코드에서는 settings 인스턴스를 사용하고 있습니다. from app.core.config import settings로 수정해야 NameError가 발생하지 않습니다.

Suggested change

from app.core.config import Settings

from app.core.config import settings

gemini-code-assist · 2025-08-27T06:35:44Z

app/worker/worker.py

+                            await process_image_scan(job, redis_client)
+                            # 처리 성공 시에만 ack
+                            await redis_client.xack(settings.STREAM_JOB, settings.GROUP_NAME, msg_id)


run 메소드 내에서 redis_client 변수가 정의되지 않았습니다. 클래스 인스턴스의 redis 클라이언트를 사용해야 하므로 self.redis_client로 수정해야 합니다. 그렇지 않으면 NameError가 발생합니다.

Suggested change

await process_image_scan(job, redis_client)

# 처리 성공 시에만 ack

await redis_client.xack(settings.STREAM_JOB, settings.GROUP_NAME, msg_id)

await process_image_scan(job, self.redis_client)

# 처리 성공 시에만 ack

await self.redis_client.xack(settings.STREAM_JOB, settings.GROUP_NAME, msg_id)

gemini-code-assist · 2025-08-27T06:35:44Z

app/core/config.py

+    REDIS_URL: str = os.environ.get("REDIS_SERVER_URL", "redis://localhost:6379")
+    S3_REGION: str = os.environ.get("S3_REGION", "ap-northeast-2")
+    S3_ENDPOINT_URL: str | None = os.environ.get("S3_ENDPOINT_URL")
+    BUCKET_NAME: str = os.environ.get("BUCKET_NAME")


BUCKET_NAME 환경 변수가 설정되지 않은 경우 os.environ.get("BUCKET_NAME")은 None을 반환합니다. 하지만 타입은 str로 지정되어 있어 None이 할당될 경우 Pydantic 유효성 검사나 런타임에서 오류가 발생할 수 있습니다. BUCKET_NAME이 필수 설정이라면 os.environ.get을 사용하지 않고 BUCKET_NAME: str로 선언하여, 환경 변수가 없을 때 Pydantic이 시작 시점에 오류를 발생시키도록 하는 것이 더 안전합니다.

Suggested change

BUCKET_NAME: str = os.environ.get("BUCKET_NAME")

BUCKET_NAME: str

gemini-code-assist · 2025-08-27T06:35:45Z

app/worker/redis_client.py

+import redis
+from dotenv import load_dotenv
+import os
+from pydantic import BaseModel, Field
+from typing import Any, Dict
+from app.core.config import settings
+
+load_dotenv()
+
+redis_host = os.environ.get("REDIS_SERVER_HOST")
+
+redis_client = redis.Redis(
+    host=redis_host, port=6379, encoding="UTF-8", decode_responses=True
+)
+
+class PublishRequest(BaseModel):
+    stream: str = Field(default=settings.JOB_STREAM, description="Redis Stream Job name")
+    payload: Dict[str, Any]


이 파일은 여러 문제를 가지고 있어 보입니다.

애플리케이션의 다른 부분에서는 비동기 redis.asyncio를 사용하는데, 여기서는 동기 redis.Redis 클라이언트를 생성하고 있습니다.

settings.JOB_STREAM을 사용하려 하지만, app/core/config.py에는 STREAM_JOB으로 정의되어 있어 AttributeError가 발생합니다.

Redis 호스트를 REDIS_SERVER_HOST 환경 변수에서 가져오는데, config.py에서는 REDIS_URL을 사용하므로 설정이 일관되지 않습니다.

이 파일이 현재 프로젝트에서 사용되지 않는다면 혼란을 피하기 위해 제거하는 것이 좋습니다. 만약 사용되는 파일이라면 위 문제들을 수정해야 합니다.

gemini-code-assist · 2025-08-27T06:35:45Z

app/api/endpoints/predictions.py

+        correlationId=correlationId,
+        presignedUrl=job_request.presigned_url,
+        replyQueue=settings.STREAM_RESULT,
+        callbackUrl=None,
+        contentType="image/jpeg",
+        createdAt=datetime.utcnow().isoformat(),
+        ttlSec=3600,
+    )


contentType과 ttlSec 값이 하드코딩되어 있습니다. contentType은 클라이언트 요청에 따라 달라질 수 있으므로 JobRequest에 포함하는 것이 좋습니다. ttlSec 값은 app/core/config.py의 settings에서 관리하는 것이 유지보수에 더 용이합니다.

gemini-code-assist · 2025-08-27T06:35:45Z

app/schemas/job.py

+class ImageJob(BaseModel):
+    correlationId: str
+    presignedUrl: str
+    replyQueue: str
+    callbackUrl: str | None = None
+    contentType: str
+    createdAt: str
+    ttlSec: int


ImageJob 모델의 필드명(correlationId, presignedUrl 등)이 camelCase로 작성되어 있습니다. Python에서는 PEP 8 스타일 가이드에 따라 snake_case (correlation_id, presigned_url) 사용을 권장합니다. 코드 전체의 일관성을 위해 snake_case로 변경하는 것을 고려해 보세요. 이 변경 사항을 적용하면 이 모델을 사용하는 다른 코드도 함께 수정해야 합니다.

Suggested change

class ImageJob(BaseModel):

correlationId: str

presignedUrl: str

replyQueue: str

callbackUrl: str | None = None

contentType: str

createdAt: str

ttlSec: int

class ImageJob(BaseModel):

correlation_id: str

presigned_url: str

reply_queue: str

callback_url: str | None = None

content_type: str

created_at: str

ttl_sec: int

gemini-code-assist · 2025-08-27T06:35:45Z

app/services/predictor_service.py

+
+
+HERE = Path(__file__).resolve().parent.parent
+MODEL_PATH = HERE / "models" / "models" / "best_model_0823.pt"


모델 경로에 models가 중복으로 포함되어 있습니다 (HERE / "models" / "models" / ...). 이는 오타일 가능성이 있으며, 실제 프로젝트 구조와 일치하는지 확인이 필요합니다. 의도된 것이 아니라면 경로를 수정해야 합니다.

Suggested change

MODEL_PATH = HERE / "models" / "models" / "best_model_0823.pt"

MODEL_PATH = HERE / "models" / "best_model_0823.pt"

gemini-code-assist · 2025-08-27T06:35:45Z

app/worker/tasks.py

+        print(f"[task] Image scan finished for job_id={correlation_id}")
+
+    except Exception as e:
+        print(f"[task] Failed to process job_id={correlation_id}: {e}")


print 함수를 사용하여 로그를 남기고 있습니다. Python의 logging 모듈을 사용하면 로그 레벨(DEBUG, INFO, ERROR 등)에 따른 제어, 로그 포맷팅, 파일 또는 외부 서비스로의 로그 전송 등 더 유연하고 강력한 로깅이 가능합니다. logging 모듈을 사용하도록 리팩토링하는 것을 권장합니다.

gemini-code-assist · 2025-08-27T06:35:46Z

app/worker/worker.py

+        self.redis_client = redis_client
+
+    async def run(self):
+        print(f"[worker] start consumer={settings.CONSUMER_NAME} group={settings.GROUP_NAME} stream={settings.STREAM_JOB}")


print 함수를 사용하여 로그를 남기고 있습니다. Python의 logging 모듈을 사용하면 로그 레벨(DEBUG, INFO, ERROR 등)에 따른 제어, 로그 포맷팅, 파일 또는 외부 서비스로의 로그 전송 등 더 유연하고 강력한 로깅이 가능합니다. logging 모듈을 사용하도록 리팩토링하는 것을 권장합니다.

youyeon11 and others added 30 commits August 25, 2025 16:52

chore: Initial Settings

e8991e3

feat: CV model 업로드

0d0d63a

fix: github 환경 변수 GITHUB -> GIT으로 변경

acdad12

chore: 기본적인 패키지 구조 설정

3687b3d

fix: GithubActions python version 3.11.13 수정

cd980a7

fix: gitignore 수정

e3902e5

fix: import 수정 (BaseModel -> pydantic)

882ee11

Merge pull request #3 from DearBelly/feat/add-model-2

22dc3f1

Feat/add model (#2)

fix: gitignore 대상 추가

fc8ec6d

fix: gitignore 대상 추가

f2f64fe

refactor: 패키지 구조 조정

c1bd631

refactor: 패키지 구조 조정

e4fa50f

feat: 이미지 경로 기반 사진 분석 API 추가

224a6e7

fix: gitignore 대상 추가

5d4bfaa

fix: python 라이브러리 설치 명령어 추가

db05ce1

chore: 템플릿 MD 업로드

78f2050

del: model_loader.py 삭제

eb3ecd6

fix: 사진 파일 gitignore에 추가

4c49622

fix: requirements.txt 모델 버전에 맞게 수정

0fb223b

Merge pull request #6 from DearBelly/feat/add-api-4

9278568

Feat/add api (#4)

refactor: 전반적인 아키텍처 구조 변경

f23a7aa

refactor: 전반적인 아키텍처 구조 변경

ffdbac1

feat: Redis 및 S3 config 파일 추가

c9582db

feat: RedisStream 수명주기 생성

c908bb9

feat: Redis 연결 추가

002ee54

feat: S3 download 서비스 코드 추가

a72cf58

feat: RedisStream 대한 Job 발행 추가

c9f758c

feat: 예측 서비스 코드 및 API 추가

35106d1

fix: requirements.txt 수정

4060037

fix: 피드백 반영하여 수정

3820074

youyeon11 and others added 2 commits August 27, 2025 15:30

fix: 피드백 반영하여 수정

d96aaa7

Merge pull request #9 from DearBelly/feat/redis-job-8

e1b29a2

Feat/redis job (#8)

gemini-code-assist bot reviewed Aug 27, 2025

View reviewed changes

youyeon11 merged commit 1790a22 into main Aug 27, 2025
1 check failed

gemini-code-assist bot reviewed Aug 27, 2025

View reviewed changes

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Merge Develop to Main#11

Merge Develop to Main#11
youyeon11 merged 32 commits intomainfrom
develop

youyeon11 commented Aug 27, 2025

Uh oh!

gemini-code-assist bot left a comment

Uh oh!

Uh oh!

gemini-code-assist bot left a comment

Uh oh!

gemini-code-assist bot Aug 27, 2025

Uh oh!

gemini-code-assist bot Aug 27, 2025

Uh oh!

gemini-code-assist bot Aug 27, 2025

Uh oh!

gemini-code-assist bot Aug 27, 2025

Uh oh!

gemini-code-assist bot Aug 27, 2025

Uh oh!

gemini-code-assist bot Aug 27, 2025

Uh oh!

gemini-code-assist bot Aug 27, 2025

Uh oh!

gemini-code-assist bot Aug 27, 2025

Uh oh!

gemini-code-assist bot Aug 27, 2025

Uh oh!

gemini-code-assist bot Aug 27, 2025

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

1 participant

	from app.core.config import Settings
	from app.core.config import settings

	BUCKET_NAME: str = os.environ.get("BUCKET_NAME")
	BUCKET_NAME: str



		HERE = Path(__file__).resolve().parent.parent
		MODEL_PATH = HERE / "models" / "models" / "best_model_0823.pt"

	MODEL_PATH = HERE / "models" / "models" / "best_model_0823.pt"
	MODEL_PATH = HERE / "models" / "best_model_0823.pt"

Conversation

youyeon11 commented Aug 27, 2025

📌 작업 목적

Uh oh!

gemini-code-assist bot left a comment

Choose a reason for hiding this comment

Summary of Changes

Highlights

Footnotes

Uh oh!

Uh oh!

gemini-code-assist bot left a comment

Choose a reason for hiding this comment

Code Review

Uh oh!

gemini-code-assist bot Aug 27, 2025

Choose a reason for hiding this comment

Uh oh!

gemini-code-assist bot Aug 27, 2025

Choose a reason for hiding this comment

Uh oh!

gemini-code-assist bot Aug 27, 2025

Choose a reason for hiding this comment

Uh oh!

gemini-code-assist bot Aug 27, 2025

Choose a reason for hiding this comment

Uh oh!

gemini-code-assist bot Aug 27, 2025

Choose a reason for hiding this comment

Uh oh!

gemini-code-assist bot Aug 27, 2025

Choose a reason for hiding this comment

Uh oh!

gemini-code-assist bot Aug 27, 2025

Choose a reason for hiding this comment

Uh oh!

gemini-code-assist bot Aug 27, 2025

Choose a reason for hiding this comment

Uh oh!

gemini-code-assist bot Aug 27, 2025

Choose a reason for hiding this comment

Uh oh!

gemini-code-assist bot Aug 27, 2025

Choose a reason for hiding this comment

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

1 participant