Master

api/apps.py

@@ -1,6 +1,28 @@
+import os
+import sys
 from django.apps import AppConfig
-
+from django.core.management import call_command
 
 class ApiConfig(AppConfig):
     default_auto_field = 'django.db.models.BigAutoField'
     name = 'api'
+
+    def ready(self):
+        # runserver 시에만 실행 (migrate, shell 등에서는 실행 안 됨)
+        if 'runserver' in sys.argv and not self._is_reloading():
+            self.reset_database_on_startup()
+
+    def _is_reloading(self):
+        """Django autoreload로 인한 재시작인지 확인"""
+        return os.environ.get('RUN_MAIN') == 'true'
+
+    def reset_database_on_startup(self):
+        """서버 시작 시 기존 reset_local_db 명령어 실행"""
+        try:
+            print('🔄 서버 시작 시 데이터베이스 자동 초기화...')
+
+            # 기존에 만든 reset_local_db 명령어 실행
+            call_command('reset_local_db')
+
+        except Exception as e:
+            print(f'❌ 자동 초기화 실패: {e}')

api/management/commands/reset_local_db.py

@@ -0,0 +1,29 @@
+import os
+import shutil
+from django.core.management.base import BaseCommand
+from django.conf import settings
+from api.models import Receipt, Participant, ReceiptInfo, Settlement
+
+class Command(BaseCommand):
+    help = '로컬 MySQL 데이터베이스 초기화'
+
+    def handle(self, *args, **options):
+        self.stdout.write('🔄 로컬 데이터베이스 초기화 시작...')
+
+        # 1. 모든 데이터 삭제 (역순으로 - 외래키 때문에)
+        self.stdout.write('🗃️ 데이터베이스 데이터 삭제 중...')
+        # Settlement.objects.all().delete()
+        ReceiptInfo.objects.all().delete()
+        Participant.objects.all().delete()
+        Receipt.objects.all().delete()
+
+        # 2. 미디어 파일 삭제
+        self.stdout.write('📁 업로드된 파일 삭제 중...')
+        media_receipts = os.path.join(settings.MEDIA_ROOT, 'receipts')
+        if os.path.exists(media_receipts):
+            shutil.rmtree(media_receipts)
+        os.makedirs(media_receipts, exist_ok=True)
+
+        self.stdout.write(
+            self.style.SUCCESS('🎉 로컬 데이터베이스 초기화 완료!')
+        )

api/migrations/0002_settlement.py

@@ -0,0 +1,28 @@
+# Generated by Django 5.2.1 on 2025-06-05 15:38
+
+import django.db.models.deletion
+import django.utils.timezone
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('api', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Settlement',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('result', models.JSONField()),
+                ('created_at', models.DateTimeField(default=django.utils.timezone.now)),
+                ('participants', models.ManyToManyField(to='api.participant')),
+                ('receipt', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='api.receipt')),
+            ],
+            options={
+                'db_table': 'settlement',
+            },
+        ),
+    ]

api/migrations/0003_settlement_method.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.2.1 on 2025-06-06 18:18
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('api', '0002_settlement'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='settlement',
+            name='method',
+            field=models.CharField(choices=[('equal', 'Equal Split (N분의1)'), ('item', 'Item Split (항목별)')], default='equal', max_length=10),
+        ),
+    ]

api/models.py

@@ -55,14 +55,19 @@ def __str__(self):
         return f"Participant {self.id}: {self.name}"
 
 class Settlement(models.Model):
+    METHOD_CHOICES = [
+        ('equal', 'Equal Split (N분의1)'),
+        ('item', 'Item Split (항목별)'),
+    ]
 
     receipt = models.ForeignKey('Receipt', on_delete=models.CASCADE)
     participants = models.ManyToManyField('Participant')
     result = models.JSONField()  # {'홍길동': 3000, '김철수': 3000}
+    method = models.CharField(max_length=10, choices=METHOD_CHOICES, default='equal') 
     created_at = models.DateTimeField(default=timezone.now)
 
     class Meta:
         db_table = 'settlement'  # MySQL 테이블 이름 지정
 
     def __str__(self):
-        return f"Settlement for Receipt {self.receipt.id} - {self.method}"
+        return f"Settlement for Receipt {self.receipt.id} - {self.created_at}"

api/ocr_pipeline/dictionary_store_item.json

@@ -0,0 +1,33 @@
+{
+  "stores": {
+    "동국대학교소비자생활협동조합": {
+      "items": [
+        "콘치즈솥밥",
+        "삼겹김치철판",
+        "치즈불닭철판",
+        "잔치국수",
+        "데리야끼치킨솥밥"
+      ]
+    },
+    "교직원식당": {
+      "items": ["식권7000"]
+    },
+    "동국대학교 폴바셋": {
+      "items": ["ICE 카페라떼"]
+    },
+    "동국대 남산학사 리김밥": {
+      "items": ["리라면", "야채김밥", "눈꽃치즈라볶이"]
+    },
+    "coopsket": {
+      "items": [
+        "슈가로로코코제로요구르트",
+        "e)신선함을그대로망고펫500",
+        "썬키스트허니유자펫280ml",
+        "아임이)빅얼음컵230g",
+        "인절미맛마시는빙수340ml",
+        "영진)참치김치삼각김밥2편",
+        "대정)소고기고추장삼각2편"
+      ]
+    }
+  }
+}

api/ocr_pipeline/extract_item2.py

@@ -0,0 +1,185 @@
+import re
+import math
+import os
+from .process_text import TextPostProcessor
+
+def normalize_number(text):
+    if not text:
+        return text
+    text = re.sub(r'O', '0', text)  # O를 0으로 변환 (OCR 오류 보정)
+    text = re.sub(r'[,.\s]', '', text)  # 콤마, 점, 공백 제거
+    return text
+
+def is_number_format(text):
+    """텍스트가 숫자 형식인지 확인"""
+    try:
+        int(normalize_number(text))
+        return True
+    except ValueError:
+        return False
+
+def extract_numbers_from_line(words):
+    """줄에서 숫자 형식의 단어들을 추출"""
+    numbers = []
+    for word in words:
+        if is_number_format(word):
+            try:
+                numbers.append(int(normalize_number(word)))
+            except ValueError:
+                continue
+    return numbers
+
+def extract_menu_items_from_lines(lines):
+    """
+    사전 기반 유사도 매칭을 사용한 메뉴 항목 추출
+    TextPostProcessor를 사용해서 dictionary_store_item.json 로드
+    """
+    # JSON 사전 파일을 사용하는 TextPostProcessor 생성
+    dict_path = os.path.join(os.path.dirname(__file__), 'dictionary_store_item.json')
+    processor = TextPostProcessor(dict_path=dict_path)
+
+    menu_items = []
+    store_name = None
+    store_found = False
+    last_successful_line = -1  # 마지막으로 성공적으로 메뉴가 추가된 줄 번호
+
+    # 1) 맨 위부터 읽으면서 가게명 찾기
+    for i, line in enumerate(lines):
+        line = line.strip()
+        if not line:
+            continue
+
+        if not store_found:
+            # 2) 가게명과 유사도 검사
+            # 전체 줄에서 가게명 매칭 시도
+            match, score = processor.find_best_store_match(line)
+            if match:
+                store_name = match
+                store_found = True
+                print(f"🏪 가게명 발견: {line} → {match} (유사도: {score:.2f})")
+                break
+
+            # 단어별로도 시도
+            words = line.split()
+            for word in words:
+                match, score = processor.find_best_store_match(word)
+                if match:
+                    store_name = match
+                    store_found = True
+                    print(f"🏪 가게명 발견: {word} → {match} (유사도: {score:.2f})")
+                    break
+
+            if store_found:
+                break
+
+    # 3) 가게명을 찾았다면, 이후 다시 txt를 읽어서 메뉴 항목 찾기
+    if store_found and store_name:
+        for i, line in enumerate(lines):
+            line = line.strip()
+            if not line:
+                continue
+
+            words = line.split()
+            if len(words) == 0:
+                continue
+
+            # 연속성 체크: 첫 번째 메뉴가 아니고 이전 성공 줄과 너무 멀면 중단
+            if last_successful_line != -1 and i > last_successful_line + 2:
+                break
+
+            # 4) 첫 번째 단어부터 시작해서 누적적으로 확장하며 최고 유사도 찾기
+            best_match = None
+            best_score = 0
+            best_end_index = -1
+            best_test_phrase = None  # 실제 매칭된 구문 저장
+
+            # 첫 번째 단어부터만 시작
+            for k in range(0, len(words)):
+                # 0부터 k까지의 단어들을 합침
+                if k == 0:
+                    # 첫 번째 단어만
+                    test_phrase = words[0]
+                else:
+                    # 여러 단어를 띄어쓰기로 연결
+                    test_phrase = " ".join(words[0:k+1])
+
+                # 메뉴 사전과 비교
+                match, score = processor.find_best_item_match(test_phrase, store_name)
+
+                if match and score > best_score:
+                    best_match = match
+                    best_score = score
+                    best_end_index = k
+                    best_test_phrase = test_phrase  # 실제 매칭된 구문 저장
+
+                # 숫자가 나타나면 더 이상 확장하지 않음
+                if k < len(words) - 1 and is_number_format(words[k+1]):
+                    break
+
+            # 최고 매칭이 있다면 처리
+            if best_match and best_score >= 0.4:  # 임계값
+                # 매칭된 부분 다음부터 숫자 추출
+                remaining_words = words[best_end_index + 1:]
+                numbers = extract_numbers_from_line(remaining_words)
+
+                menu_added = False
+
+                if len(numbers) == 1:
+                    # 4.1) 숫자 하나: 총액 = 개당 가격
+                    total_amount = numbers[0]
+                    unit_price = total_amount
+                    quantity = 1
+
+                    menu_items.append({
+                        "item_name": best_match,
+                        "unit_price": unit_price,
+                        "total_amount": total_amount,
+                        "quantity": quantity
+                    })
+                    print(f"🍔 메뉴 발견: {best_test_phrase} → {best_match} (유사도: {best_score:.2f})")
+                    print(f"   → 개당: {unit_price}원, 개수: {quantity}개, 총액: {total_amount}원")
+                    menu_added = True
+
+                elif len(numbers) == 2:
+                    # 4.2) 숫자 두개: 첫번째=개당가격, 두번째=총액
+                    unit_price = numbers[0]
+                    total_amount = numbers[1]
+                    quantity = total_amount // unit_price if unit_price > 0 else 1
+
+                    menu_items.append({
+                        "item_name": best_match,
+                        "unit_price": unit_price,
+                        "total_amount": total_amount,
+                        "quantity": quantity
+                    })
+                    print(f"🍔 메뉴 발견: {best_test_phrase} → {best_match} (유사도: {best_score:.2f})")
+                    print(f"   → 개당: {unit_price}원, 개수: {quantity}개, 총액: {total_amount}원")
+                    menu_added = True
+
+                elif len(numbers) >= 3:
+                    # 4.3) 숫자 세개 이상: 첫번째=개당가격, 두번째=개수, 세번째=총액
+                    unit_price = numbers[0]
+                    quantity = numbers[1]
+                    total_amount = numbers[2]
+
+                    menu_items.append({
+                        "item_name": best_match,
+                        "unit_price": unit_price,
+                        "total_amount": total_amount,
+                        "quantity": quantity
+                    })
+                    print(f"🍔 메뉴 발견: {best_test_phrase} → {best_match} (유사도: {best_score:.2f})")
+                    print(f"   → 개당: {unit_price}원, 개수: {quantity}개, 총액: {total_amount}원")
+                    menu_added = True
+
+                # 메뉴가 성공적으로 추가되었으면 마지막 성공 줄 번호 업데이트
+                if menu_added:
+                    last_successful_line = i
+
+    result = {
+        "store_name": store_name,
+        "items": menu_items,
+    }
+
+    print(f"✅ 항목 추출 완료 → {store_name or '상호명 없음'} ({len(menu_items)}개)\n")
+    return result