Update app.py

app.py

@@ -188,17 +188,51 @@ async def request_gpt4o_completion(image_id: str, user_credit: int):
 
         # Step 4: Prepare the prompt for Gemini
         prompt = f'''
-        Image content: {extracted_text}
-
-        Please analyze the above text extracted from an image and perform the following tasks:
-        1. Identify if the image contains receipts or business cards.
-        2. For each receipt found, extract the following information: {', '.join(receipt_sheet_headers)}
-        3. For each business card found, extract the following information: {', '.join(business_card_sheet_headers)}
-        4. Format the extracted information as a JSON object with two main keys: 'receipts' and 'busi_cards'.
-        5. Each key should contain an array of objects, where each object represents a single receipt or business card.
-        6. Ensure the total number of receipts and business cards combined does not exceed 500.
+        #Direction
+        You are an employee of finance department of a company. For each task well done, you will be rewarded $1,000 bonus. 
+        The boss has come to your office and gave you a task of transfering the content of receipts and business cards onto a google spreadsheet.
+        He hands you two things. 
+        One, is a paper containing guideline and example.
+        Two, is an OCR text.
+
+        #Guideline
+        -strictly adhere to the given content of OCR text.
+        -it could be of receipt(s) or business card(s) but not both
+        -any other info that couldn't be classified is sorted into "기타"
+
+        #Example
+        ##Receipts
+        Input
+
+        Output
+        -발행일: 2024/02/23 20:10
+        -상호: 은성마켓
+        -사업자번호: 727-27-01455
+        -합계금액: 2,800원
+        -내역: 
+            느타리버섯 1,000원 1개
+            양배추 1,800원 1개
+        -카드번호: 4673-09**-****-515*
+        -카드종류: KB국민체크
+        -승인번호: 30014507
+        -기타:
+
+        ##Business Cards
+        Input
+
+        Output
+        -이름: 박장원
+        -회사명: 아이프리마
+        -소속 및 직위: 대표
+        -주소: 경기도 화성시 정남면 만년로 98번길 55 (우편번호) 18523
+        -전화: (031) 000-0000
+        -휴대폰: (031) 000-0000
+        -팩스: 010-0000-0000
+        -e-mail: iprima@gmail.com
+        -홈페이지: www.sol2u.co.kr
+        -기타:
 
-        Please provide the response in valid JSON format.
+        Image content: {extracted_text}
         '''
         logging.info("Prompt prepared for Gemini model: " + prompt)