z

app.py

@@ -1,11 +1,16 @@
 import os
 import shutil
+import time
+
 import cv2
 import base64
 import uuid
 import re
 from flask import Flask
 import gradio as gr
+from google import genai
+from openai import OpenAI as QwenOpenAI
+
 
 # --- Config 클래스 (Gemma, GPT4o 제거, Qwen만 사용) ---
 class Config:
@@ -21,57 +26,44 @@ class Config:
         {"name": "사이다", "image": "images/food6.jpg", "price": 12.00},
     ]
     # 알리바바 Qwen API 키 (기본값은 빈 문자열)
-    QWEN_API_KEY = ""
-
+    QWEN_API_KEY = "sk-2424f0bd26d64fe5a7f3a2bd407adc76"
+    GEMINI_API_key = "AIzaSyCc8lcm2cZo3ZeMgi4QN1IHSvn9BBpLnz4"
     DEFAULT_PROMPT_TEMPLATE = (
-        "### Persona ###\n"
-        "You are an expert tip calculation assistant focusing on service quality observed in a video, and you also consider the user's review, star rating, and recent Google reviews. "
-        "Your role is to evaluate all these aspects evenly by assigning each a score out of 100, then calculate the overall average score to determine the appropriate tip percentage.\n\n"
-
-        "### Task ###\n"
-        "1. **Video Analysis**: Analyze the video frames as described in the 'Video Caption' to observe the staff's actions and interactions. "
-        "Assign a score out of 100 based on the observed service quality.\n\n"
-
-        "2. **Bill Amount Determination**: Determine the bill amount by following these steps:\n"
-        "   - If an explicit bill amount is mentioned in the 'Video Caption', use it.\n"
-        "   - Otherwise, use the 'Calculated Subtotal' provided.\n"
-        "   - If neither is available, assume a default value of $50.\n\n"
-
-        "3. **Overall Service Quality Evaluation**:\n"
-        "   Evaluate the service quality by evenly scoring the following four components, each out of 100:\n"
-        "       a) **Video Service Score**: Service quality observed in the video (from Video Caption).\n"
-        "       b) **Google Review Score**: Please provide an overall analysis of the recent Google reviews and give the general rating score. Highlight any significant social issues mentioned in the reviews, such as racist comments, discriminatory behavior, or other major societal concerns. The analysis should consider all reviews collectively, not individually. Regardless of the rating score, tips should be calculated very low.\n"
-        "       - **Note**: If any review mentions racist, sexist, or any other ethical violations, all Google Review scores should automatically be set to 0. This is to account for the severity of such violations. "
-        "       - Additionally, when such violations are found, the tip should be calculated **very low**, even lower than the typical range for poor service.\n"
-        "       c) **User Review Score**: The user's review (if the review mentions improvements or enhanced service, especially if it indicates that previously reported issues like racism have been resolved, adjust the negative impact accordingly).\n"
-        "       d) **Star Rating Score**: The user's star rating, interpreted on a scale where 5/5 corresponds to 100 points.\n"
-        "   - Calculate the overall average score by taking the mean of these four scores.\n\n"
-
-        "4. **Service Quality Classification and Tip Guidelines**:\n"
-        "   - Based on the overall average score, classify the service quality as follows:\n"
-        "       - Poor Service: Overall average score < 60 (Tip range: 0% ~ 5% of the bill)\n"
-        "       - Average Service: Overall average score ≥ 60 and < 80 (Tip range: 10% ~ 15% of the bill)\n"
-        "       - Good Service: Overall average score ≥ 80 (Tip range: 15% ~ 20% of the bill)\n"
-        "   - Select a specific tip percentage within the appropriate range.\n"
-        "   - Calculate the tip amount by multiplying the determined bill amount by the chosen tip percentage (round to two decimal places).\n"
-        "   - Calculate the final total bill by adding the tip amount to the subtotal (round to two decimal places).\n\n"
-
-        "5. **Review Prioritization and Score Adjustment**:\n"
-        "   - Even though all four factors are evaluated equally, if the user's review explicitly indicates improvements (e.g., stating 'There's no racism!' or similar phrases), then increase the Google Review score by adjusting or restoring it to a value higher than the score initially reduced due to negative comments. In other words, if improvements are noted, add a positive increment (using a '+' adjustment) to the Google Review score, ensuring it is higher than the negatively impacted value.\n"
-        "   - Ensure that the user's review score takes priority when there is a conflict with the negative tone of the Google Reviews.\n\n"
-
-        "### Ethical Violations and Tip Adjustment ###\n"
-        "   - If there are any racist, discriminatory, or offensive remarks in the Google reviews, regardless of the review score, the tip should be significantly reduced. "
-        "   - In these cases, the tip should be calculated at a very low percentage, often close to 0%, to reflect the severity of such behavior. "
-        "   - This adjustment should result in an even lower tip amount than would be calculated for poor service alone.\n\n"
-
-        "### User Context ###\n"
-        "    - Current Country: USA\n"
-        "    - Restaurant Name: The Golden Spoon (Assumed)\n"
-        "    - Calculated Subtotal: ${calculated_subtotal:.2f}\n"
-        "    - User Star Rating: {star_rating} / 5\n"
-        "    - Currently User Review: {user_review}\n\n"
-
+        """
+        ### Persona ###
+        You are an expert tip calculation assistant focusing on service quality observed in video captions, and you also consider the user's review, star rating, and recent Google reviews.
+        Your role is to evaluate all these aspects, assign scores, and calculate the overall average score to determine the appropriate tip percentage, with special handling for ethical violations.
+    
+        ### Task ###
+            1.  **Video Analysis**: Analyze the service quality depicted in the provided **Video Caption(s)**. If multiple captions are given, perform a **holistic evaluation** based on the overall impression conveyed by all captions combined. **Focus on significant staff actions, expressions, and interactions described across the captions.** Assign a single, overall **Video Service Score** out of 100 based on this comprehensive analysis, rather than averaging scores from individual captions.
+    
+            2.  **Bill Amount Determination**: Determine the bill amount by following these steps:
+                * Use the 'Calculated Subtotal' provided.
+    
+            3.  **Overall Service Quality Evaluation**:
+                Evaluate the service quality by evenly scoring the following four components, each out of 100:
+                    a) **Video Service Score**: Service quality derived from the holistic analysis of Video Caption(s).
+                    b) **Google Review Score**: Provide an overall analysis of the recent Google reviews and give the general rating score. Highlight any significant social issues mentioned, such as racist comments or discriminatory behavior. The analysis should consider all reviews collectively.
+                       * **Note**: If any review mentions racist, sexist, or any other ethical violations, the Google Review score must automatically be set to **0**.
+                    c) **User Review Score**: The user's review.
+                    d) **Star Rating Score**: The user's star rating, interpreted on a scale where 5/5 corresponds to 100 points.
+                * Calculate the overall average score by taking the mean of these four scores.
+    
+            4.  **Service Quality Classification and Tip Guidelines**:
+                * Based on the overall average score, classify the service quality as follows:
+                    * Poor Service: Overall average score < 60 (Tip range: 0% ~ 5% of the bill)
+                    * Average Service: Overall average score ≥ 60 and < 80 (Tip range: 10% ~ 15% of the bill)
+                    * Good Service: Overall average score ≥ 80 (Tip range: 15% ~ 20% of the bill)
+                * Select a specific tip percentage within the appropriate range. **Exception**: If ethical violations were noted (Google Review Score is 0), the Final Tip Percentage **must be 0%**.
+                * Calculate the tip amount by multiplying the determined bill amount by the chosen tip percentage (round to two decimal places).
+                * Calculate the final total bill by adding the tip amount to the subtotal (round to two decimal places).
+    
+            5.  **Review Prioritization and Score Adjustment**:
+                * Even though all four factors are evaluated equally, if the user's review **explicitly states that specific issues previously mentioned in Google Reviews (especially ethical violations like racism) have been resolved or are no longer present**, then the Google Review score should be adjusted upwards from its potentially reduced value (e.g., from 0 back towards a score reflecting the *current* situation described by the user). The user's direct, specific, and current assessment takes precedence over older or contradicted Google Review points in such cases. Ensure the User Review Score reflects the user's sentiment.
+    
+        ### Ethical Violations and Tip Adjustment ###
+            * If there are any racist, discriminatory, or offensive remarks found in the Google reviews, regardless of the overall review sentiment or scores from other factors, the Google Review Score is automatically set to **0**. Consequently, the **Final Tip Percentage must be set to 0%** to strongly reflect the severity and unacceptability of such behavior.
+    
         "### Recent Google Review ###\n\n"
 
         "#### Google Review 1 ####\n"
@@ -80,45 +72,50 @@ class Config:
         "#### Google Review 2 ####\n"
         "[4.0 stars] A steak restaurant with an American-style atmosphere. Thick tenderloin and strips served? The sound stimulated my appetite, but there was a lot of food, so I left it behind. It was a bit difficult to eat because it was undercooked in the middle. I also had stir-fried kimchi as a garnish, and it was a little sweet, so it tasted like Southeast Asia. The ice cream I had for dessert was delicious, and there was a lot of food left over."
         "\n\n"
-
-        "### Input ###\n"
-        "Video Caption:\n{{caption_text}}\n\n"
-
-        "### Output ###\n"
-        "Return your answer in the exact format below:\n"
-        "Video Text Analysis: [Summary of the observed actions and interactions of the staff in the video along with the assigned score (out of 100) based on video analysis.]\n"
-        "Recent Google Review Analysis: [Summary of the insights from the Google Reviews, including any negative or racist comments, with the assigned score (out of 100).]\n"
-        "User Review Analysis: [Summary of the user's review including any improvements or enhanced service mentions, with the assigned score (out of 100).]\n"
-        "Star Rating Analysis: [Interpret the user's star rating (e.g., converting 5/5 to 100 points) and include the assigned score (out of 100).]\n"
-        "Overall Analysis: [Step-by-step explanation detailing:\n"
-        "           - How the bill amount was determined;\n"
-        "           - How each of the four components was scored and how any negative scores (e.g., for racism) were adjusted based on improvement indications in the user's review;\n"
-        "           - If any review mentions racist, sexist, or other ethical violations, the Google Review score is automatically set to 0, and a **very low** tip percentage is calculated. This reflects the severity of such violations and ensures that the tip is significantly reduced.\n"
-        "           - How the overall average score was calculated;\n"
-        "           - The reasoning for the final service quality classification based on the average score;\n"
-        "           - How the tip percentage was chosen within the guideline range and the detailed calculation for the tip amount and final total bill, taking into account the 0% tip percentage due to the ethical violations.]\n\n"
-
-
-        "### Example Output Indicators (for reference only) ###\n"
-        "**Final Tip Percentage**: 12.5%\n"
-        "**Final Tip Amount**: $6.25\n"
-        "**Final Total Bill**: $56.25\n\n"
-
-        "### Example Output Indicators (for reference only) ###\n"
-        "**Final Tip Percentage**: 12.5%\n"
-        "**Final Tip Amount**: $6.25\n"
-        "**Final Total Bill**: $56.25\n\n"
-
-        "### Output Indicators ###\n"
-        "**Final Tip Percentage**: [X]% (only floating point)\n"
-        "**Final Tip Amount**: $[Calculated Tip]\n"
-        "**Final Total Bill**: $[Subtotal + Tip]\n"
-
-        "### GUIDE ###"
-
-        "\n\nIn Final Answer,The ### Output Indicators ### must strictly follow the format of ### Example Output Indicators (for reference only) ###\n\n"
-        "### FINAL ANSWER: **THIS INSTRUCTION ENSURES THAT ONLY THE NECESSARY SPECIAL CHARACTERS THAT ARE PART OF THE REQUIRED OUTPUT FORMAT (E.G., **, $, %) ARE USED, AND ANY OTHER SPECIAL CHARACTERS SUCH AS \\textbf, LATEX COMMANDS, OR ANY NON-STANDARD CHARACTERS SHOULD BE EXCLUDED.**"
-        "Complete all the ### Task ### and ### Output ### first, and perform the ### Output Indicators ### last. Do not perform the ### Output Indicators ### in advance."
+    
+        ### Video Caption Input ###
+        # Multiple captions can be provided under this section
+        Video Caption(s):
+        {{caption_text}}
+    
+        ### Output ###
+        Return your answer in the exact format below, providing the text analyses first, followed by the JSON block containing the final calculations:
+    
+        Video Text Analysis: [Provide a summary of the **significant staff actions, expressions, and interactions** observed across all provided **Video Caption(s)**. Explain how these observations contribute to the overall service impression. Include the single, **holistically determined Video Service Score** (out of 100) based on this combined analysis.]
+        Recent Google Review Analysis: [Summary of the insights from the Google Reviews, including any negative or racist comments, with the assigned score (out of 100). If ethical violations result in a 0 score, state this clearly.]
+        User Review Analysis: [Summary of the user's review including any improvements or enhanced service mentions, with the assigned score (out of 100). Note if this review led to adjustments in the Google Review score based on explicit statements about resolved issues.]
+        Star Rating Analysis: [Interpret the user's star rating (e.g., converting 5/5 to 100 points) and include the assigned score (out of 100).]
+        Overall Analysis: [Step-by-step explanation detailing:
+                   - How the bill amount was determined;
+                   - How each of the four components was scored, including the holistic evaluation of video captions and any adjustments made to the Google Review score based on specific user statements about resolved issues;
+                   - If any review mentioned racist, sexist, or other ethical violations, explicitly state that the Google Review score was set to 0 and the **Final Tip Percentage is consequently set to 0%**, reflecting the severity;
+                   - How the overall average score was calculated;
+                   - The reasoning for the final service quality classification based on the average score;
+                   - How the final tip percentage was chosen (either based on the guideline range or mandatorily set to 0% due to ethical violations) and the detailed calculation for the tip amount and final total bill.
+    
+        ### Final Calculation Results (JSON Format) ###
+        ```json
+        {{{{
+          "final_tip_percentage": <calculated_percentage_int>,
+          "final_tip_amount": <calculated_tip_float>,
+          "final_total_bill": <calculated_total_bill_float>
+        }}}}
+        ```
+    
+        ### GUIDE ###
+    
+        In Final Answer, The ### Final Calculation Results (JSON Format) ### section must strictly follow the JSON structure provided above, containing only the JSON object and its calculated numerical values (use floating-point numbers for all values). **DO NOT include the placeholders like <calculated_percentage_float> in the actual final output; replace them with the real calculated numbers.** Ensure no other special characters like **, $, % are used *within* the JSON structure itself, only standard JSON syntax (keys in double quotes, string values in double quotes, numbers without quotes). The text analysis sections preceding the JSON should still follow their specified format.
+        Complete all the text analysis and overall analysis sections first, and generate the ### Final Calculation Results (JSON Format) ### section last.
+        **THIS INSTRUCTION ENSURES THAT ONLY THE NECESSARY SPECIAL CHARACTERS THAT ARE PART OF THE REQUIRED OUTPUT FORMAT (E.G., standard JSON syntax) ARE USED, AND ANY OTHER SPECIAL CHARACTERS SUCH AS **, $, %, LATEX COMMANDS, OR ANY NON-STANDARD CHARACTERS SHOULD BE EXCLUDED FROM THE FINAL JSON BLOCK.**
+    
+        ### User Context ###
+            * Current Country: USA
+            * Currently Restaurant Name: The Golden Spoon (Assumed)
+            * Currently Calculated Subtotal: ${calculated_subtotal:.2f}
+            * Currently User Star Rating: {star_rating} / 5 (5 is the maximum)
+            * Currently User Review: {user_review}
+    
+        """
     )
 
     CUSTOM_CSS = """
@@ -156,6 +153,7 @@ class ModelClients:
             api_key=config.QWEN_API_KEY,
             base_url="https://dashscope-intl.aliyuncs.com/compatible-mode/v1",
         )
+        self.gemini_client = genai.Client(api_key=config.GEMINI_API_key)
 
     def encode_video_qwen(self, video_path):
         with open(video_path, "rb") as video_file:
@@ -368,6 +366,86 @@ Task 2: Provide a short chronological summary of the entire scene.
         analysis, tip_percentage, tip_amount, output_text = self.parse_llm_output(final_text)
         return analysis, tip_percentage, tip_amount, [], None, output_text
 
+    def process_tip_gemini(self, video_file_path, star_rating, user_review, calculated_subtotal, custom_prompt=None):
+
+        if not video_file_path or not os.path.exists(video_file_path):
+            return "비디오 파일 경로가 유효하지 않습니다.", [], None
+        image_captioning_prompt = '''
+        Task 1: Describe the actions of any waiters or staff visible in these restaurant video. Note any specific interactions, mistakes, or positive actions.
+        Task 2: Provide a concise overall summary of the scene depicted in the frames, in chronological order if possible.
+
+        Task 1 Output:
+        Task 2 Output:
+        '''
+
+        video_file = self.model_clients.gemini_client.files.upload(file=video_file_path)
+        print(f"Uploaded file info: {video_file}")
+
+        # 동영상은 처리 중(Processing) 상태이므로, ACTIVE 상태가 될 때까지 대기합니다.
+        while video_file.state.name == "PROCESSING":
+            print("동영상 처리 중...")
+            time.sleep(1)  # 예: 5초마다 상태 점검
+            video_file = self.model_clients.gemini_client.files.get(name=video_file.name)
+
+        # 처리 실패 여부 확인
+        if video_file.state.name == "FAILED":
+            raise ValueError(f"파일 처리 실패: {video_file.state.name}")
+
+        # 비디오를 프롬프트에 포함해서 추론 요청
+        try:
+            caption_summary = self.model_clients.gemini_client.models.generate_content(
+                model="gemini-2.0-flash-lite",  # 예시: Gemini 2.0 Flash 모델 사용
+                contents=[video_file, image_captioning_prompt]
+            )
+            caption_summary = caption_summary.text
+
+        except Exception as e:
+            print(f"로컬 모델 프레임 처리 오류: {e}")
+            return f"Error in processing frames with local model: {e}", None, None
+
+        user_review = user_review.strip() if user_review and user_review.strip() else "(No user review provided)"
+
+        if custom_prompt is None:
+            prompt = self.config.DEFAULT_PROMPT_TEMPLATE.format(
+                calculated_subtotal=calculated_subtotal, star_rating=star_rating, user_review=user_review
+            )
+        else:
+            try:
+                prompt = custom_prompt.format(
+                    calculated_subtotal=calculated_subtotal, star_rating=star_rating, user_review=user_review
+                )
+            except KeyError as e:
+                print(f"경고: 커스텀 프롬프트에 필요한 키가 없습니다: {e}. 기본 템플릿을 사용합니다.")
+                prompt = self.config.DEFAULT_PROMPT_TEMPLATE.format(
+                    calculated_subtotal=calculated_subtotal, star_rating=star_rating, user_review=user_review
+                )
+
+        final_prompt = prompt.replace("{caption_text}", caption_summary)
+        messages = final_prompt
+
+        video_file = self.model_clients.gemini_client.files.upload(file=video_file_path)
+        print(f"Uploaded file info: {video_file}")
+
+        # 동영상은 처리 중(Processing) 상태이므로, ACTIVE 상태가 될 때까지 대기합니다.
+        while video_file.state.name == "PROCESSING":
+            print("동영상 처리 중...")
+            time.sleep(1)  # 예: 5초마다 상태 점검
+            video_file = self.model_clients.gemini_client.files.get(name=video_file.name)
+
+        # 처리 실패 여부 확인
+        if video_file.state.name == "FAILED":
+            raise ValueError(f"파일 처리 실패: {video_file.state.name}")
+
+
+        response = self.model_clients.gemini_client.models.generate_content(
+            model="gemini-2.0-flash-lite",  # 예시: Gemini 2.0 Flash 모델 사용
+            contents=[video_file, messages]
+        )
+        llm_output = response.text
+        analysis, tip_percentage, tip_amount, output_text = self.parse_llm_output(llm_output)
+        return analysis, tip_percentage, tip_amount, [], None, output_text
+
+
     def calculate_manual_tip(self, tip_percent, subtotal):
         tip_amount = subtotal * (tip_percent / 100)
         total_bill = subtotal + tip_amount
@@ -401,7 +479,7 @@ class UIHandler:
         updated_prompt = updated_prompt.replace("{caption_text}", "{{caption_text}}")
         return calculated_subtotal, updated_prompt
 
-    def compute_tip(self, alibaba_key, video_file_obj, subtotal, star_rating, user_review, custom_prompt_text):
+    def compute_tip(self, type, video_file_obj, subtotal, star_rating, user_review, custom_prompt_text):
         analysis_output = "계산을 시작합니다..."
         tip_percentage = 0.0
         tip_output = "$0.00"
@@ -409,12 +487,6 @@ class UIHandler:
         if video_file_obj is None:
             return "오류: 비디오 파일을 업로드해주세요.", "$0.00", total_bill_output, custom_prompt_text, gr.update(value=None)
         try:
-            if alibaba_key and alibaba_key.strip():
-                from openai import OpenAI as QwenOpenAI
-                self.tip_calculator.model_clients.qwen_client = QwenOpenAI(
-                    api_key=alibaba_key,
-                    base_url="https://dashscope-intl.aliyuncs.com/compatible-mode/v1"
-                )
             temp_video_path = f"temp_video_{uuid.uuid4().hex}.mp4"
             original_path = video_file_obj.name if hasattr(video_file_obj, 'name') else video_file_obj
             shutil.copyfile(original_path, temp_video_path)
@@ -424,9 +496,14 @@ class UIHandler:
             return f"오류: 비디오 파일을 처리할 수 없습니다: {e}", "$0.00", total_bill_output, custom_prompt_text, None
         frame_folder = None
         try:
-            analysis, tip_percentage, tip_amount, _, _, output_text = self.tip_calculator.process_tip_qwen(
-                temp_video_path, star_rating, user_review, subtotal, custom_prompt_text
-            )
+            if type == 'qwen':
+                analysis, tip_percentage, tip_amount, _, _, output_text = self.tip_calculator.process_tip_qwen(
+                    temp_video_path, star_rating, user_review, subtotal, custom_prompt_text
+                )
+            else:
+                analysis, tip_percentage, tip_amount, _, _, output_text = self.tip_calculator.process_tip_gemini(
+                    temp_video_path, star_rating, user_review, subtotal, custom_prompt_text
+                )
             if "Error" in analysis:
                 analysis_output = analysis
                 tip_amount = 0.0
@@ -444,9 +521,9 @@ class UIHandler:
             self.video_processor.cleanup_temp_files(temp_video_path, frame_folder)
         return analysis_output, tip_output, total_bill_output, custom_prompt_text, gr.update(value=None)
 
-    def auto_tip_and_invoice(self, alibaba_key, video_file_obj, subtotal, star_rating, review, prompt, *quantities):
+    def auto_tip_and_invoice(self, type, video_file_obj, subtotal, star_rating, review, prompt, *quantities):
         analysis, tip_disp, total_bill_disp, prompt_out, vid_out = self.compute_tip(
-            alibaba_key, video_file_obj, subtotal, star_rating, review, prompt
+            type, video_file_obj, subtotal, star_rating, review, prompt
         )
         invoice = self.update_invoice_summary(*quantities, tip_disp, total_bill_disp)
         return analysis, tip_disp, total_bill_disp, prompt_out, vid_out, invoice
@@ -507,8 +584,9 @@ class App:
             review_input, rating_input = None, None
             btn_5, btn_10, btn_15, btn_20, btn_25 = None, None, None, None, None
             qwen_btn = None
+            gemini_btn = None
             tip_display, total_bill_display, payment_btn, payment_result = None, None, None, None
-            alibaba_key_input, video_input = None, None
+            video_input = None
             analysis_display, order_summary_display = None, None
             prompt_editor = None  # 프롬프트 에디터는 다른 탭에 정의될 것임
 
@@ -549,6 +627,8 @@ class App:
                                 btn_25 = gr.Button("25%")
                             with gr.Row():
                                 # Qwen 버튼 정의를 여기로 이동!
+                                gemini_btn = gr.Button("Google-Gemini AI Tip Calculation", variant="primary",
+                                                     elem_id="Gemini-button")
                                 qwen_btn = gr.Button("Alibaba-Qwen AI Tip Calculation", variant="primary",
                                                      elem_id="qwen-button")
                             gr.Markdown("### 4. Results")
@@ -561,8 +641,7 @@ class App:
                         # --- 오른쪽 열 (원래대로, 프롬프트 디스플레이 제외) ---
                         with gr.Column(scale=1):
                             gr.Markdown("### 5. Upload & Prompt Access")
-                            alibaba_key_input = gr.Textbox(label="Alibaba API Key", placeholder="Enter Key", lines=1,
-                                                           type="password")
+
                             video_input = gr.Video(label="Upload Service Video")
 
                             gr.Markdown("### 6. AI Analysis")
@@ -589,10 +668,10 @@ class App:
                 gr.Examples(
                     examples=[
                         # 입력 순서: [Alibaba API Key, Video, Subtotal, Star Rating, Review] + [각 음식의 Qty]
-                        ["", "video/sample.mp4", 0.0, 1, "He drop the tray..so bad", 0, 0, 0, 0, 0, 2, 0, 0],
-                        ["", "video/sample2.mp4", 0.0, 5, "Good service!", 0, 0, 0, 0, 0, 2, 0, 0]
+                        ["video/sample.mp4", 0.0, 1, "He drop the tray..so bad", 0, 0, 0, 0, 0, 2, 0, 0],
+                        ["video/sample2.mp4", 0.0, 5, "Good service!", 0, 0, 0, 0, 0, 2, 0, 0]
                     ],
-                    inputs=[alibaba_key_input, video_input, subtotal_display, rating_input,
+                    inputs=[video_input, subtotal_display, rating_input,
                             review_input] + quantity_inputs,
                     outputs=[analysis_display, tip_display, total_bill_display, video_input, order_summary_display],
                     label="Example: Bad Service, Good Service"
@@ -603,8 +682,8 @@ class App:
             # (실제 코드에서는 컴포넌트 None 체크 또는 더 나은 구조화 필요)
 
             if all([subtotal_display, subtotal_visible_display_output, review_input, rating_input, prompt_editor,
-                    order_summary_display, alibaba_key_input, video_input, analysis_display, tip_display,
-                    total_bill_display, payment_result, qwen_btn] + quantity_inputs):
+                    order_summary_display, video_input, analysis_display, tip_display,
+                    total_bill_display, payment_result, qwen_btn, gemini_btn] + quantity_inputs):
 
                 # 1. 소계 업데이트 (숨겨진 Number -> 보이는 Textbox)
                 subtotal_display.change(
@@ -634,14 +713,21 @@ class App:
                 # 4. AI 계산 버튼('Main Interface' 탭) 클릭 시
                 # 입력: 메인 탭의 컴포넌트들 + 'Edit Prompt' 탭의 prompt_editor
                 # 출력: 메인 탭의 컴포넌트들 + 'Edit Prompt' 탭의 prompt_editor (업데이트 될 수 있으므로)
-                qwen_compute_inputs = [alibaba_key_input, video_input, subtotal_display, rating_input, review_input,
-                                       prompt_editor] + quantity_inputs
-                qwen_compute_outputs = [analysis_display, tip_display, total_bill_display, prompt_editor, video_input,
-                                        order_summary_display]
                 qwen_btn.click(
-                    fn=self.ui_handler.auto_tip_and_invoice,
-                    inputs=qwen_compute_inputs,
-                    outputs=qwen_compute_outputs
+                    fn=lambda vid, sub, rat, rev, prom, *qty: self.ui_handler.auto_tip_and_invoice('qwen', vid, sub,
+                                                                                                   rat, rev, prom,
+                                                                                                   *qty),
+                    inputs=[video_input, subtotal_display, rating_input, review_input, prompt_editor] + quantity_inputs,
+                    outputs=[analysis_display, tip_display, total_bill_display, prompt_editor, video_input,
+                             order_summary_display]
+                )
+                gemini_btn.click(
+                    fn=lambda vid, sub, rat, rev, prom, *qty: self.ui_handler.auto_tip_and_invoice('gemini', vid, sub,
+                                                                                                   rat, rev, prom,
+                                                                                                   *qty),
+                    inputs=[video_input, subtotal_display, rating_input, review_input, prompt_editor] + quantity_inputs,
+                    outputs=[analysis_display, tip_display, total_bill_display, prompt_editor, video_input,
+                             order_summary_display]
                 )
 
                 # 5. 수동 팁 버튼('Main Interface' 탭) 클릭 시

requirements.txt

@@ -1,4 +1,6 @@
 gradio
 opencv-python
 flask
-openai
+openai
+genai
+google-generativeai