Update app.py

app.py

@@ -93,7 +93,7 @@ def get_video_comments(video_id):
         request = youtube.commentThreads().list(
             part='snippet',
             videoId=video_id,
-            maxResults=100,   #댓글 읽어들이는 수 정의
+            maxResults=10,   #댓글 읽어들이는 수 정의
             textFormat='plainText'
         )
         response = request.execute()
@@ -113,7 +113,7 @@ def get_video_comments(video_id):
                     part='snippet',
                     videoId=video_id,
                     pageToken=response['nextPageToken'],
-                    maxResults=100,  #댓글 읽어들이는 수 정의
+                    maxResults=10,  #댓글 읽어들이는 수 정의
                     textFormat='plainText'
                 )
                 response = request.execute()
@@ -123,7 +123,7 @@ def get_video_comments(video_id):
     except Exception as e:
         return [{'error': str(e)}]
 
-def fetch_comments(video_url, system_prompt):
+def fetch_comments(video_url, system_prompt, reply_mode):
     log_entries = []
     video_id_match = re.search(r'(?:v=|\/)([0-9A-Za-z_-]{11}).*', video_url)
     if video_id_match:
@@ -140,7 +140,10 @@ def fetch_comments(video_url, system_prompt):
         if not new_comments or 'error' in new_comments[0]:
             return "댓글을 찾을 수 없거나 오류가 발생했습니다."
         
-        recent_new_comments = [c for c in new_comments if c['comment_id'] not in {c['comment_id'] for c in existing_comments} and c['reply_count'] == 0]
+        if reply_mode == "Non Reply":
+            recent_new_comments = [c for c in new_comments if c['comment_id'] not in {c['comment_id'] for c in existing_comments} and c['reply_count'] == 0]
+        else:
+            recent_new_comments = [c for c in new_comments if c['comment_id'] not in {c['comment_id'] for c in existing_comments}]
         
         if recent_new_comments:
             for most_recent_comment in recent_new_comments:
@@ -165,7 +168,7 @@ def fetch_comments(video_url, system_prompt):
 
 def background_fetch_comments():
     while not stop_event.is_set():
-        result = fetch_comments("https://www.youtube.com/watch?v=dQw4w9WgXcQ", DEFAULT_SYSTEM_PROMPT)  # URL과 프롬프트 실제 사용 예시
+        result = fetch_comments("https://www.youtube.com/watch?v=dQw4w9WgXcQ", DEFAULT_SYSTEM_PROMPT, "Non Reply")  # URL과 프롬프트 실제 사용 예시
         print(result)
         time.sleep(10)
 
@@ -193,6 +196,9 @@ with demo:
         input_text_url = gr.Textbox(placeholder='YouTube video URL', label='YouTube URL')
         input_text_prompt = gr.Textbox(placeholder='시스템 프롬프트', label='시스템 프롬프트', value=DEFAULT_SYSTEM_PROMPT, lines=5)
     
+    with gr.Row():
+        radio_reply_mode = gr.Radio(['Non Reply', 'Force Reply'], label='Reply Mode', value='Non Reply')
+    
     with gr.Row():
         result_button_transcribe = gr.Button('Transcribe')
         result_button_comments = gr.Button('Fetch Comments and Generate Reply')
@@ -202,7 +208,7 @@ with demo:
         output_text_prompt = gr.Textbox(placeholder='응답 텍스트', label='응답 텍스트', lines=20)
     
     result_button_transcribe.click(get_text, inputs=input_text_url, outputs=output_text_transcribe, api_name="transcribe_api")
-    result_button_comments.click(fetch_comments, inputs=[input_text_url, input_text_prompt], outputs=output_text_prompt, api_name="fetch_comments_api")
+    result_button_comments.click(fetch_comments, inputs=[input_text_url, input_text_prompt, radio_reply_mode], outputs=output_text_prompt, api_name="fetch_comments_api")
 
 # 인터페이스 실행
 demo.launch()