Update app.py

app.py

@@ -137,18 +137,21 @@ def validate_row_data(row_data):
             return False
     return True
 
-def extract_data_to_excel_and_html(page, board_select):
+def extract_data_to_excel_and_html(page, board_select, custom_url=""):
     try:
         if not isinstance(page, (int, float)) or page < 1 or page > 50:
             return None, "<p>페이지 수는 1-50 사이여야 합니다.</p>"
         
         session = setup_session()
-        base_url = get_base_url(board_select)
-        if base_url == "Invalid board selected":
-            return "Invalid board selected", ""
-        
-        korea_time = datetime.now(pytz.timezone('Asia/Seoul'))
-        filename = f'{board_select}_{korea_time.strftime("%Y%m%d_%H%M%S")}.xlsx'
+        # 사용자가 직접 링크를 입력한 경우, 해당 링크를 base_url로 사용
+        if custom_url.strip():
+            base_url = custom_url.strip()
+            filename = f'custom_{datetime.now(pytz.timezone("Asia/Seoul")).strftime("%Y%m%d_%H%M%S")}.xlsx'
+        else:
+            base_url = get_base_url(board_select)
+            if base_url == "Invalid board selected":
+                return "Invalid board selected", ""
+            filename = f'{board_select}_{datetime.now(pytz.timezone("Asia/Seoul")).strftime("%Y%m%d_%H%M%S")}.xlsx'
         
         workbook = xlsxwriter.Workbook(filename)
         worksheet = workbook.add_worksheet()
@@ -268,6 +271,7 @@ def extract_data_to_excel_and_html(page, board_select):
         
         for p in range(1, page + 1):
             try:
+                # 기본 URL에 페이지 번호 파라미터 추가
                 url = f"{base_url}&search.page={p}"
                 logger.info(f"[CRAWL] Fetching page {p}: {url}")
                 response = session.get(url)
@@ -358,9 +362,9 @@ def extract_data_to_excel_and_html(page, board_select):
         logger.error(f"[CRAWL] 전체 크롤링 실패: {str(e)}")
         return None, f"<p style='color: #dc3545; padding: 10px; background-color: #f8d7da; border-radius: 4px;'>{error_message}</p>"
 
-def crawl_with_progress(board, pages):
+def crawl_with_progress(board, pages, custom_url):
     try:
-        excel_file, html_output = extract_data_to_excel_and_html(pages, board)
+        excel_file, html_output = extract_data_to_excel_and_html(pages, board, custom_url)
         if excel_file:
             return excel_file, html_output, "수집 완료"  # status 메시지 추가
         else:
@@ -511,26 +515,34 @@ tr:hover {
 with gr.Blocks(css=css) as demo:
     gr.Markdown("# N사 Cafe 핫딜 게시판 크롤링")
     gr.Markdown("""
-    페이지를 입력하면 결과를 출력합니다.
+    페이지 수를 입력하거나, 직접 URL을 입력하시면 결과를 출력합니다.
+    (게시판 선택은 기본 URL을 사용합니다. 직접 URL 입력시 해당 링크로 수집합니다.)
     최대 페이지수는 50페이지 입니다.
     """)
     
     with gr.Row():
         board_select = gr.Radio(
             choices=["맘이베베", "맘스홀릭", "광주맘", "쇼핑지름신", "부산맘", "진희맘"],
-            label="게시판을 선택하세요",
+            label="게시판을 선택하세요 (직접 링크 입력이 없을 경우)",
             container=True
         )
         
     with gr.Row():
         inp = gr.Number(
-            label="수집을 원하시는 페이지 수를 입력하세요(최대 50페이지)",
+            label="수집할 페이지 수를 입력하세요 (최대 50페이지)",
             value=1,
             minimum=1,
             maximum=50,
             container=True
         )
     
+    with gr.Row():
+        custom_url = gr.Textbox(
+            label="직접 링크 입력 (옵션)",
+            placeholder="예: https://cafe.naver.com/ArticleList.nhn?...",
+            container=True
+        )
+    
     status = gr.Textbox(
         label="상태",
         value="대기 중...",
@@ -544,7 +556,7 @@ with gr.Blocks(css=css) as demo:
     
     btn.click(
         fn=crawl_with_progress,
-        inputs=[board_select, inp],
+        inputs=[board_select, inp, custom_url],
         outputs=[output_file, output_html, status]
     )