Update app.py

app.py

@@ -172,134 +172,3 @@ process_button.click(
 
 app.launch()
 
-
-
-
-
-def extract_info_block(article_text):
-    pattern = r'(KBS2.*?<모델명>.*?)(?:\n\n|$)'
-    match = re.search(pattern, article_text, re.DOTALL)
-
-    if match:
-        return match.group(1).strip()
-    else:
-        return None
-
-# 제품명 추출 함수 수정 (여러 모델명 처리)
-
-def extract_product_info(article_text):
-    brand_match = re.search(r'<브랜드>[ \t]*([^\n]+)', article_text)
-    brand = brand_match.group(1).strip() if brand_match else None
-
-    model_matches = re.findall(r'<모델명>[ \t]*([^\n]+)', article_text)
-
-    result = []
-
-    if brand and model_matches:
-        brand_words = re.findall(r'[A-Za-z]+', brand)
-        selected_brand = ' '.join(brand_words[:2])
-
-        for model in model_matches:
-            model = model.strip()
-            search_query = f"{selected_brand} {model}"
-            search_url = f"https://www.coupang.com/np/search?component=&q={search_query.replace(' ', '+')}"
-            result.append((model, search_url))
-
-    return result
-
-# process_url 함수 수정 (모델명 여러개 대응)
-
-def process_url(tistory_url, output_dir):
-    result = []
-    try:
-        response = requests.get(tistory_url)
-        soup = BeautifulSoup(response.text, 'html.parser')
-
-        img_tags = soup.find_all('img')
-        img_urls = [img['src'] for img in img_tags if 'src' in img.attrs]
-
-        # 이미지 처리
-        for idx, img_url in enumerate(img_urls):
-            base_name = os.path.basename(img_url).split('?')[0]
-            if base_name.startswith('new_ico'):
-                continue
-
-            img_data = requests.get(img_url).content
-            input_img = Image.open(io.BytesIO(img_data)).convert('RGBA')
-
-            input_bytes = io.BytesIO()
-            input_img.save(input_bytes, format='PNG')
-            removed = remove(input_bytes.getvalue())
-            img_no_bg = Image.open(io.BytesIO(removed)).convert('RGBA')
-
-            final_img = add_watermark(img_no_bg)
-
-            os.makedirs(output_dir, exist_ok=True)
-            name, ext = os.path.splitext(base_name)
-            save_path = os.path.join(output_dir, f"{name}_{idx+1}_processed.png")
-            final_img.save(save_path)
-            result.append(f"✔️ 저장 완료: {save_path}")
-
-        # 텍스트 추출
-        article_text = soup.get_text()
-
-        # 프로그램 정보 블록 추출
-        info_block = extract_info_block(article_text)
-
-        # 제품 정보 추출 (여러 모델명 처리)
-        item_info_list = extract_product_info(article_text)
-
-        # 정리된 출력 포맷
-        if info_block:
-            result.append("\n===== 프로그램 정보 블록 =====")
-            result.append(info_block)
-            result.append("================================\n")
-
-        for model_name, search_url in item_info_list:
-            result.append(f"🛍️ 아이템 : {model_name}")
-            result.append(f"🔗 쿠팡 링크: {search_url}")
-
-    except Exception as e:
-        result.append(f"❌ URL 처리 실패: {tistory_url} / 에러: {e}")
-
-    return result
-
-
-# 여러 URL 처리
-def process_multiple_urls(urls_text, output_dir):
-    urls = [url.strip() for url in urls_text.strip().splitlines() if url.strip()]
-    all_results = []
-    for url in urls:
-        all_results.extend(process_url(url, output_dir))
-        all_results.append("-" * 50)
-    
-    final_text = "\n".join(all_results)
-    
-    # 결과를 파일로 저장
-    os.makedirs(output_dir, exist_ok=True)
-    result_file_path = os.path.join(output_dir, "result_log.txt")
-    with open(result_file_path, 'w', encoding='utf-8') as f:
-        f.write(final_text)
-    
-    return final_text, result_file_path
-# Gradio UI
-# Gradio 앱
-with gr.Blocks() as app:
-    gr.Markdown("# ✨ 티스토리 자동 처리기 ✨\n- 이미지 배경 제거 + 워터마크 삽입\n- 제품명 추출 후 쿠팡 검색 링크 생성\n- 다운로드 기능 추가!")
-
-    with gr.Row():
-        urls_input = gr.Textbox(label="티스토리 게시글 URL 여러 개 (줄바꿈해서 입력)", lines=5, placeholder="https://example1.com\nhttps://example2.com")
-    with gr.Row():
-        output_folder = gr.Textbox(label="저장할 폴더 경로", value=DEFAULT_DOWNLOAD_DIR)
-
-    process_button = gr.Button("처리 시작 🚀")
-    output_text = gr.Textbox(label="결과", lines=20)
-    download_file = gr.File(label="결과 다운로드")  # 다운로드 버튼 추가
-
-    process_button.click(
-        fn=process_multiple_urls, 
-        inputs=[urls_input, output_folder], 
-        outputs=[output_text, download_file]   # 출력 두 개 연결
-    )
-
-app.launch()