Update app.py
Browse files
app.py
CHANGED
|
@@ -2,80 +2,63 @@ import os
|
|
| 2 |
import time
|
| 3 |
import base64
|
| 4 |
import io
|
| 5 |
-
import traceback
|
| 6 |
import threading
|
| 7 |
import atexit
|
| 8 |
-
import
|
| 9 |
from datetime import datetime
|
| 10 |
-
from typing import Optional,
|
| 11 |
|
| 12 |
import gradio as gr
|
| 13 |
import anthropic
|
| 14 |
-
from playwright.
|
| 15 |
-
from playwright.sync_api import sync_playwright, Browser, Page
|
| 16 |
from PIL import Image
|
| 17 |
-
import requests
|
| 18 |
from dotenv import load_dotenv
|
| 19 |
|
| 20 |
# νκ²½λ³μ λ‘λ
|
| 21 |
load_dotenv()
|
| 22 |
|
| 23 |
-
#
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
is_browser_ready = False
|
| 29 |
-
browser_thread_id = None
|
| 30 |
|
| 31 |
-
# Claude API ν΄λΌμ΄μΈνΈ μ΄κΈ°ν
|
| 32 |
def get_claude_client():
|
|
|
|
| 33 |
api_key = os.getenv("ANTHROPIC_API_KEY")
|
| 34 |
if not api_key:
|
| 35 |
raise ValueError("ANTHROPIC_API_KEY νκ²½λ³μκ° μ€μ λμ§ μμμ΅λλ€.")
|
| 36 |
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
timeout=60.0
|
| 43 |
-
)
|
| 44 |
-
return client
|
| 45 |
-
except Exception as e:
|
| 46 |
-
print(f"Claude ν΄λΌμ΄μΈνΈ μ΄κΈ°ν μ€λ₯: {e}")
|
| 47 |
-
# κΈ°λ³Έ μ΄κΈ°ν λ°©μμΌλ‘ μ¬μλ
|
| 48 |
-
return anthropic.Anthropic(api_key=api_key)
|
| 49 |
|
| 50 |
def install_browsers():
|
| 51 |
-
"""Playwright λΈλΌμ°μ
|
| 52 |
try:
|
| 53 |
-
print("Playwright λΈλΌμ°μ
|
| 54 |
-
# λ³λ ¬ μ€μΉλ‘ μλ ν₯μ
|
| 55 |
os.system("playwright install chromium --force")
|
| 56 |
-
print("
|
| 57 |
-
os.system("apt-get install -y fonts-nanum fonts-noto-cjk &")
|
| 58 |
-
print("λΈλΌμ°μ λ° ν°νΈ μ€μΉ μλ£!")
|
| 59 |
return True
|
| 60 |
except Exception as e:
|
| 61 |
-
print(f"μ€μΉ μ€ν¨: {e}")
|
| 62 |
return False
|
| 63 |
|
| 64 |
-
def
|
| 65 |
-
"""
|
| 66 |
-
|
|
|
|
|
|
|
| 67 |
|
| 68 |
try:
|
| 69 |
-
|
| 70 |
-
print(f"λΈλΌμ°μ WarmUp μμ (μ€λ λ: {browser_thread_id})")
|
| 71 |
-
|
| 72 |
-
# λΉ λ₯Έ λΈλΌμ°μ μ€μΉ
|
| 73 |
-
install_browsers()
|
| 74 |
|
| 75 |
-
# Playwright μΈμ€ν΄μ€ μμ±
|
| 76 |
playwright_instance = sync_playwright().start()
|
| 77 |
|
| 78 |
-
#
|
| 79 |
browser = playwright_instance.chromium.launch(
|
| 80 |
headless=True,
|
| 81 |
args=[
|
|
@@ -84,159 +67,77 @@ def init_browser():
|
|
| 84 |
'--disable-gpu',
|
| 85 |
'--disable-extensions',
|
| 86 |
'--disable-web-security',
|
| 87 |
-
'--disable-features=
|
| 88 |
-
'--disable-background-timer-throttling',
|
| 89 |
-
'--disable-renderer-backgrounding',
|
| 90 |
-
'--disable-plugins',
|
| 91 |
-
'--disable-images', # μ΄λ―Έμ§ λ‘λ© λΉνμ±νλ‘ μλ ν₯μ
|
| 92 |
-
'--disable-javascript', # JS λΉνμ±νλ‘ μλ ν₯μ
|
| 93 |
-
'--disable-css', # CSS λΉνμ±νλ‘ μλ ν₯μ
|
| 94 |
-
'--force-device-scale-factor=1',
|
| 95 |
-
'--aggressive-cache-discard',
|
| 96 |
-
'--memory-pressure-off'
|
| 97 |
]
|
| 98 |
)
|
| 99 |
|
| 100 |
# μ νμ΄μ§ μμ±
|
| 101 |
page = browser.new_page()
|
| 102 |
-
|
| 103 |
-
# λ·°ν¬νΈ μ€μ (μκ² μ€μ νμ¬ μλ ν₯μ)
|
| 104 |
-
page.set_viewport_size({"width": 1200, "height": 800})
|
| 105 |
-
|
| 106 |
-
# λΉ λ₯Έ ν€λ μ€μ
|
| 107 |
page.set_extra_http_headers({
|
| 108 |
-
"Accept-Language": "ko-KR,ko;q=0.9"
|
|
|
|
| 109 |
})
|
| 110 |
|
| 111 |
-
|
| 112 |
-
|
| 113 |
-
|
| 114 |
-
wait_until="domcontentloaded",
|
| 115 |
-
timeout=15000)
|
| 116 |
|
| 117 |
-
|
| 118 |
-
time.sleep(
|
| 119 |
|
| 120 |
-
|
| 121 |
-
|
| 122 |
-
|
| 123 |
-
|
| 124 |
-
|
| 125 |
-
|
| 126 |
-
|
| 127 |
-
|
| 128 |
-
def ensure_browser_ready():
|
| 129 |
-
"""λΈλΌμ°μ κ° μ€λΉλμ΄ μλμ§ νμΈνκ³ , νμμ μ΄κΈ°ν"""
|
| 130 |
-
global is_browser_ready, browser, page, browser_thread_id
|
| 131 |
-
|
| 132 |
-
current_thread_id = threading.current_thread().ident
|
| 133 |
-
|
| 134 |
-
# λΈλΌμ°μ κ° μ€λΉλμ§ μμκ±°λ λ€λ₯Έ μ€λ λμμ μμ±λ κ²½μ°
|
| 135 |
-
if not is_browser_ready or browser_thread_id != current_thread_id or not browser or not page:
|
| 136 |
-
print(f"λΈλΌμ°μ μ¬μ΄κΈ°ν νμ (νμ¬ μ€λ λ: {current_thread_id}, λΈλΌμ°μ μ€λ λ: {browser_thread_id})")
|
| 137 |
-
|
| 138 |
-
# κΈ°μ‘΄ λΈλΌμ°μ μ 리
|
| 139 |
-
cleanup_browser()
|
| 140 |
-
|
| 141 |
-
# νμ¬ μ€λ λμμ μλ‘ μ΄κΈ°ν
|
| 142 |
-
init_browser()
|
| 143 |
-
|
| 144 |
-
return is_browser_ready
|
| 145 |
-
|
| 146 |
-
def cleanup_browser():
|
| 147 |
-
"""λΈλΌμ°μ μ 리"""
|
| 148 |
-
global browser, page, playwright_instance
|
| 149 |
-
|
| 150 |
-
try:
|
| 151 |
-
if page:
|
| 152 |
-
page.close()
|
| 153 |
-
page = None
|
| 154 |
-
if browser:
|
| 155 |
-
browser.close()
|
| 156 |
-
browser = None
|
| 157 |
-
if playwright_instance:
|
| 158 |
-
playwright_instance.stop()
|
| 159 |
-
playwright_instance = None
|
| 160 |
-
print("λΈλΌμ°μ μ 리 μλ£")
|
| 161 |
-
except Exception as e:
|
| 162 |
-
print(f"λΈλΌμ°μ μ 리 μ€ μ€λ₯: {e}")
|
| 163 |
-
|
| 164 |
-
def apply_filters(geo: str, time_range: str, category: str) -> bool:
|
| 165 |
-
"""Google Trends νν° μ μ© (κ³ μ λͺ¨λ)"""
|
| 166 |
-
global page
|
| 167 |
-
|
| 168 |
-
if not ensure_browser_ready():
|
| 169 |
-
print("λΈλΌμ°μ μ€λΉ μ€ν¨")
|
| 170 |
-
return False
|
| 171 |
-
|
| 172 |
-
try:
|
| 173 |
-
print(f"κ³ μ νν° μ μ©: μ§μ={geo}")
|
| 174 |
-
|
| 175 |
-
# μ§μ μ€μ μ μν URL ꡬμ±
|
| 176 |
-
geo_map = {
|
| 177 |
-
"KR": "KR", "US": "US", "JP": "JP",
|
| 178 |
-
"GB": "GB", "DE": "DE", "FR": "FR"
|
| 179 |
-
}
|
| 180 |
-
|
| 181 |
-
base_url = "https://trends.google.com/trending"
|
| 182 |
-
geo_param = geo_map.get(geo, "KR")
|
| 183 |
-
url = f"{base_url}?geo={geo_param}"
|
| 184 |
-
|
| 185 |
-
print(f"λΉ λ₯Έ μ΄λ: {url}")
|
| 186 |
-
# λΉ λ₯Έ λ‘λ©μ μν΄ domcontentloaded μ¬μ©, νμμμ λν λ¨μΆ
|
| 187 |
-
page.goto(url, wait_until="domcontentloaded", timeout=10000)
|
| 188 |
-
|
| 189 |
-
# μ΅μ λκΈ° μκ°
|
| 190 |
-
time.sleep(1)
|
| 191 |
-
|
| 192 |
-
print("κ³ μ νν° μ μ© μλ£")
|
| 193 |
-
return True
|
| 194 |
-
|
| 195 |
-
except Exception as e:
|
| 196 |
-
print(f"νν° μ μ© μ€ν¨: {str(e)}")
|
| 197 |
-
# μ€ν¨ν΄λ νμ¬ νμ΄μ§μμ μ€ν¬λ¦°μ· μλ
|
| 198 |
-
return True # μ€ν¨ν΄λ κ³μ μ§ν
|
| 199 |
-
|
| 200 |
-
def capture_screenshot() -> Optional[Image.Image]:
|
| 201 |
-
"""νμ¬ νμ΄μ§μ μ€ν¬λ¦°μ· μΊ‘μ² (κ³ μ λͺ¨λ)"""
|
| 202 |
-
global page
|
| 203 |
-
|
| 204 |
-
if not ensure_browser_ready():
|
| 205 |
-
print("λΈλΌμ°μ μ€λΉ μ€ν¨")
|
| 206 |
-
return None
|
| 207 |
-
|
| 208 |
-
try:
|
| 209 |
-
print("κ³ μ μ€ν¬λ¦°μ· μΊ‘μ² μ€...")
|
| 210 |
|
| 211 |
-
#
|
| 212 |
-
time.sleep(
|
| 213 |
|
| 214 |
-
#
|
| 215 |
try:
|
| 216 |
-
|
| 217 |
-
|
| 218 |
-
|
| 219 |
-
|
| 220 |
-
print("λ©μΈ μμ κ³ μ μΊ‘μ² μλ£")
|
| 221 |
else:
|
| 222 |
-
|
| 223 |
-
|
| 224 |
-
|
| 225 |
-
|
| 226 |
-
|
| 227 |
-
|
| 228 |
-
|
| 229 |
-
|
| 230 |
-
|
| 231 |
screenshot = Image.open(io.BytesIO(screenshot_bytes))
|
|
|
|
| 232 |
return screenshot
|
| 233 |
-
|
| 234 |
except Exception as e:
|
| 235 |
print(f"μ€ν¬λ¦°μ· μΊ‘μ² μ€ν¨: {str(e)}")
|
| 236 |
return None
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 237 |
|
| 238 |
def analyze_with_claude(image: Image.Image) -> str:
|
| 239 |
-
"""Claude API
|
|
|
|
|
|
|
|
|
|
| 240 |
try:
|
| 241 |
client = get_claude_client()
|
| 242 |
|
|
@@ -245,16 +146,40 @@ def analyze_with_claude(image: Image.Image) -> str:
|
|
| 245 |
image.save(buffer, format="PNG")
|
| 246 |
image_base64 = base64.b64encode(buffer.getvalue()).decode()
|
| 247 |
|
| 248 |
-
#
|
| 249 |
-
prompt = """Google Trends μ€ν¬λ¦°μ·μ
|
| 250 |
-
|
| 251 |
-
|
| 252 |
-
|
| 253 |
-
|
| 254 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 255 |
response = client.messages.create(
|
| 256 |
model="claude-3-5-sonnet-20241022",
|
| 257 |
-
max_tokens=
|
| 258 |
messages=[
|
| 259 |
{
|
| 260 |
"role": "user",
|
|
@@ -276,291 +201,253 @@ def analyze_with_claude(image: Image.Image) -> str:
|
|
| 276 |
]
|
| 277 |
)
|
| 278 |
|
| 279 |
-
|
|
|
|
|
|
|
| 280 |
|
| 281 |
except Exception as e:
|
| 282 |
error_msg = f"Claude API λΆμ μ€ν¨: {str(e)}"
|
| 283 |
print(error_msg)
|
| 284 |
return error_msg
|
| 285 |
|
| 286 |
-
def
|
| 287 |
-
"""
|
| 288 |
-
global
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 289 |
|
| 290 |
try:
|
| 291 |
-
|
|
|
|
| 292 |
|
| 293 |
-
|
| 294 |
-
|
| 295 |
-
|
| 296 |
-
|
| 297 |
-
|
| 298 |
-
|
| 299 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 300 |
else:
|
| 301 |
-
|
| 302 |
-
|
| 303 |
-
|
| 304 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 305 |
else:
|
| 306 |
-
|
| 307 |
|
| 308 |
-
|
| 309 |
-
|
| 310 |
-
|
| 311 |
|
| 312 |
-
def
|
| 313 |
-
"""
|
| 314 |
-
global
|
| 315 |
|
| 316 |
-
if
|
| 317 |
-
|
| 318 |
-
|
| 319 |
-
|
| 320 |
-
|
| 321 |
-
|
| 322 |
-
|
| 323 |
-
|
| 324 |
-
|
| 325 |
-
return "νμ΄μ§ λΉ λ₯Έ μλ‘κ³ μΉ¨ μλ£!"
|
| 326 |
-
except Exception as e:
|
| 327 |
-
return f"νμ΄μ§ μλ‘κ³ μΉ¨ μ€ν¨: {str(e)}"
|
| 328 |
-
|
| 329 |
-
def main_process_screenshot(geo: str, time_range: str, category: str) -> Tuple[Optional[Image.Image], str]:
|
| 330 |
-
"""μ€ν¬λ¦°μ· μΊ‘μ²λ§ μννλ ν¨μ (κ³ μ λͺ¨λ)"""
|
| 331 |
-
try:
|
| 332 |
-
# 1. λΈλΌμ°μ μ€λΉ νμΈ (λΉ λ₯Έ νμΈ)
|
| 333 |
-
if not ensure_browser_ready():
|
| 334 |
-
return None, "λΈλΌμ°μ κ° μ€λΉλμ§ μμμ΅λλ€."
|
| 335 |
-
|
| 336 |
-
# 2. κ³ μ νν° μ μ©
|
| 337 |
-
print(f"κ³ μ μ²λ¦¬: μ§μ={geo}")
|
| 338 |
-
filter_success = apply_filters(geo, time_range, category)
|
| 339 |
-
|
| 340 |
-
# νν° μ€ν¨ν΄λ νμ¬ νμ΄μ§μμ μ€ν¬λ¦°μ· μλ
|
| 341 |
-
if not filter_success:
|
| 342 |
-
print("νν° μ μ© μ€ν¨νμ§λ§ νμ¬ νμ΄μ§μμ μ€ν¬λ¦°μ· μλ")
|
| 343 |
-
|
| 344 |
-
# 3. κ³ μ μ€ν¬λ¦°μ· μΊ‘μ²
|
| 345 |
-
screenshot = capture_screenshot()
|
| 346 |
-
|
| 347 |
-
if screenshot is None:
|
| 348 |
-
return None, "μ€ν¬λ¦°μ· μΊ‘μ²μ μ€ν¨νμ΅λλ€."
|
| 349 |
-
|
| 350 |
-
return screenshot, "κ³ μ μ€ν¬λ¦°μ· μΊ‘μ² μλ£!"
|
| 351 |
|
| 352 |
-
|
| 353 |
-
error_msg = f"κ³ μ μ²λ¦¬ μ€ μ€λ₯: {str(e)}"
|
| 354 |
-
print(error_msg)
|
| 355 |
-
return None, error_msg
|
| 356 |
|
| 357 |
-
def
|
| 358 |
-
"""
|
| 359 |
-
if screenshot is None:
|
| 360 |
-
return "λΆμν μ€ν¬λ¦°μ·μ΄ μμ΅λλ€."
|
| 361 |
|
| 362 |
-
|
| 363 |
-
|
| 364 |
-
|
| 365 |
-
|
| 366 |
-
|
| 367 |
-
|
| 368 |
-
|
| 369 |
-
|
| 370 |
-
|
| 371 |
-
|
| 372 |
-
|
| 373 |
-
|
| 374 |
-
|
| 375 |
-
|
| 376 |
-
|
| 377 |
-
|
| 378 |
-
|
| 379 |
-
|
| 380 |
-
print(f"νν° μ μ© μ€: μ§μ={geo}, μκ°={time_range}, μΉ΄ν
κ³ λ¦¬={category}")
|
| 381 |
-
filter_success = apply_filters(geo, time_range, category)
|
| 382 |
-
|
| 383 |
-
if not filter_success:
|
| 384 |
-
return None, "νν° μ μ©μ μ€ν¨νμ΅λλ€."
|
| 385 |
|
| 386 |
-
|
| 387 |
-
|
| 388 |
-
screenshot = capture_screenshot()
|
| 389 |
|
| 390 |
-
if
|
| 391 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 392 |
|
| 393 |
-
|
| 394 |
-
|
| 395 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 396 |
|
| 397 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 398 |
|
| 399 |
-
|
| 400 |
-
|
| 401 |
-
|
| 402 |
-
|
| 403 |
-
|
| 404 |
-
|
| 405 |
-
def check_browser_status():
|
| 406 |
-
"""λΈλΌμ°μ μν νμΈ (κ³ μ)"""
|
| 407 |
-
global is_browser_ready, page, browser_thread_id
|
| 408 |
-
|
| 409 |
-
current_thread_id = threading.current_thread().ident
|
| 410 |
-
|
| 411 |
-
if not page:
|
| 412 |
-
return "λΈλΌμ°μ λ―Έμ΄κΈ°ν"
|
| 413 |
-
|
| 414 |
-
if not is_browser_ready:
|
| 415 |
-
return "λΈλΌμ°μ μ΄κΈ°ν μ€..."
|
| 416 |
-
|
| 417 |
-
try:
|
| 418 |
-
current_url = page.url
|
| 419 |
-
return f"λΈλΌμ°μ μ€λΉμλ£ - URL: {current_url[:50]}..."
|
| 420 |
-
except Exception as e:
|
| 421 |
-
return f"λΈλΌμ°μ μ°κ²° λ¬Έμ : {str(e)[:30]}..."
|
| 422 |
-
|
| 423 |
-
# Gradio μΈν°νμ΄μ€ μμ±
|
| 424 |
-
def create_interface():
|
| 425 |
-
"""Gradio μΈν°νμ΄μ€ μμ±"""
|
| 426 |
-
|
| 427 |
-
with gr.Blocks(title="Google Trends κ³ μ λΆμκΈ°", theme=gr.themes.Soft()) as interface:
|
| 428 |
-
gr.Markdown("# β‘ Google Trends κ³ μ λΆμκΈ°")
|
| 429 |
-
gr.Markdown("**μ΄κ³ μ λͺ¨λ**: Google Trends μ€μκ° κ²μμ΄λ₯Ό λΉ λ₯΄κ² μΊ‘μ²νκ³ AI λΆμν©λλ€.")
|
| 430 |
|
| 431 |
-
#
|
| 432 |
-
screenshot_state = gr.State(None)
|
| 433 |
|
| 434 |
with gr.Row():
|
| 435 |
with gr.Column(scale=1):
|
| 436 |
-
|
| 437 |
-
choices=["KR", "US", "JP", "GB", "DE", "FR"],
|
| 438 |
-
value="KR",
|
| 439 |
-
label="μ§μ μ ν"
|
| 440 |
-
)
|
| 441 |
|
| 442 |
-
|
| 443 |
-
|
| 444 |
-
value="24μκ°",
|
| 445 |
-
label="κΈ°κ° μ ν"
|
| 446 |
-
)
|
| 447 |
|
| 448 |
-
|
| 449 |
-
|
| 450 |
-
|
| 451 |
-
|
| 452 |
-
)
|
| 453 |
|
| 454 |
-
|
| 455 |
-
capture_btn = gr.Button("β‘ κ³ μ μΊ‘μ²", variant="primary", size="lg")
|
| 456 |
-
analyze_btn = gr.Button("π€ AI λΆμ", variant="secondary", interactive=False)
|
| 457 |
|
| 458 |
-
|
| 459 |
-
|
| 460 |
-
|
| 461 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 462 |
|
| 463 |
with gr.Column(scale=2):
|
| 464 |
-
|
| 465 |
-
|
| 466 |
-
|
| 467 |
-
|
| 468 |
-
max_lines=15,
|
| 469 |
-
placeholder="κ³ μ μΊ‘μ² ν 'AI λΆμ' λ²νΌμ ν΄λ¦νμΈμ."
|
| 470 |
)
|
| 471 |
-
status_output = gr.Textbox(label="μν λ©μμ§")
|
| 472 |
-
|
| 473 |
-
# μ΄λ²€νΈ νΈλ€λ¬
|
| 474 |
|
| 475 |
-
#
|
| 476 |
-
|
| 477 |
-
|
| 478 |
-
|
| 479 |
-
# λΆμ λ²νΌ νμ±ν
|
| 480 |
-
return screenshot, screenshot, message, gr.update(interactive=True)
|
| 481 |
-
else:
|
| 482 |
-
return None, None, message, gr.update(interactive=False)
|
| 483 |
-
|
| 484 |
-
capture_btn.click(
|
| 485 |
-
fn=on_capture_click,
|
| 486 |
-
inputs=[geo_dropdown, time_dropdown, category_dropdown],
|
| 487 |
-
outputs=[screenshot_output, screenshot_state, status_output, analyze_btn]
|
| 488 |
-
)
|
| 489 |
-
|
| 490 |
-
# AI λΆμ λ²νΌ
|
| 491 |
-
def on_analyze_click(screenshot):
|
| 492 |
-
if screenshot is None:
|
| 493 |
-
return "λΆμν μ€ν¬λ¦°μ·μ΄ μμ΅λλ€. λ¨Όμ 'κ³ μ μΊ‘μ²' λ²νΌμ ν΄λ¦νμΈμ."
|
| 494 |
-
|
| 495 |
-
analysis_result = analyze_screenshot_with_claude(screenshot)
|
| 496 |
-
return analysis_result
|
| 497 |
-
|
| 498 |
-
analyze_btn.click(
|
| 499 |
-
fn=on_analyze_click,
|
| 500 |
-
inputs=[screenshot_state],
|
| 501 |
-
outputs=[analysis_output]
|
| 502 |
-
)
|
| 503 |
-
|
| 504 |
-
# κΈ°ν λ²νΌλ€
|
| 505 |
-
refresh_btn.click(
|
| 506 |
-
fn=refresh_trends_page,
|
| 507 |
-
outputs=status_output
|
| 508 |
-
)
|
| 509 |
-
|
| 510 |
-
reinit_btn.click(
|
| 511 |
-
fn=reinit_browser,
|
| 512 |
-
outputs=status_output
|
| 513 |
)
|
| 514 |
|
| 515 |
-
|
| 516 |
-
fn=
|
| 517 |
-
outputs=status_output
|
| 518 |
)
|
| 519 |
|
| 520 |
return interface
|
| 521 |
|
| 522 |
def main():
|
| 523 |
-
"""λ©μΈ ν¨μ
|
| 524 |
-
print("=" *
|
| 525 |
-
print("Google Trends
|
| 526 |
-
print("=" *
|
| 527 |
|
| 528 |
# API ν€ νμΈ
|
| 529 |
api_key = os.getenv("ANTHROPIC_API_KEY")
|
| 530 |
if not api_key:
|
| 531 |
-
print("
|
|
|
|
| 532 |
return
|
| 533 |
else:
|
| 534 |
-
print("
|
| 535 |
-
|
| 536 |
-
print("β Gradio κ³ μ μΈν°νμ΄μ€ μ€λΉ μ€...")
|
| 537 |
-
|
| 538 |
-
# νκ²½ νμΈ
|
| 539 |
-
if os.getenv("SPACE_ID"):
|
| 540 |
-
print("β Hugging Face Space νκ²½μμ μ€ν μ€...")
|
| 541 |
|
| 542 |
-
|
|
|
|
|
|
|
| 543 |
|
| 544 |
-
|
| 545 |
-
|
| 546 |
-
|
| 547 |
-
|
| 548 |
-
# μ’
λ£ μ λΈλΌμ°μ μ 리
|
| 549 |
-
atexit.register(cleanup_browser)
|
| 550 |
|
| 551 |
-
#
|
| 552 |
-
|
|
|
|
|
|
|
|
|
|
| 553 |
|
| 554 |
-
print("
|
|
|
|
| 555 |
|
| 556 |
-
#
|
| 557 |
-
|
| 558 |
-
|
| 559 |
-
|
| 560 |
-
|
| 561 |
-
|
| 562 |
-
|
| 563 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 564 |
|
| 565 |
if __name__ == "__main__":
|
| 566 |
main()
|
|
|
|
| 2 |
import time
|
| 3 |
import base64
|
| 4 |
import io
|
|
|
|
| 5 |
import threading
|
| 6 |
import atexit
|
| 7 |
+
import traceback
|
| 8 |
from datetime import datetime
|
| 9 |
+
from typing import Optional, Dict, Any
|
| 10 |
|
| 11 |
import gradio as gr
|
| 12 |
import anthropic
|
| 13 |
+
from playwright.sync_api import sync_playwright
|
|
|
|
| 14 |
from PIL import Image
|
|
|
|
| 15 |
from dotenv import load_dotenv
|
| 16 |
|
| 17 |
# νκ²½λ³μ λ‘λ
|
| 18 |
load_dotenv()
|
| 19 |
|
| 20 |
+
# λΆμ λ°μ΄ν° μ μ₯μ
|
| 21 |
+
analyzed_data = {}
|
| 22 |
+
last_update_time = None
|
| 23 |
+
update_thread = None
|
| 24 |
+
is_updating = False
|
|
|
|
|
|
|
| 25 |
|
|
|
|
| 26 |
def get_claude_client():
|
| 27 |
+
"""Claude API ν΄λΌμ΄μΈνΈ μ΄κΈ°ν"""
|
| 28 |
api_key = os.getenv("ANTHROPIC_API_KEY")
|
| 29 |
if not api_key:
|
| 30 |
raise ValueError("ANTHROPIC_API_KEY νκ²½λ³μκ° μ€μ λμ§ μμμ΅λλ€.")
|
| 31 |
|
| 32 |
+
return anthropic.Anthropic(
|
| 33 |
+
api_key=api_key,
|
| 34 |
+
max_retries=3,
|
| 35 |
+
timeout=60.0
|
| 36 |
+
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
|
| 38 |
def install_browsers():
|
| 39 |
+
"""Playwright λΈλΌμ°μ μ€μΉ"""
|
| 40 |
try:
|
| 41 |
+
print("Playwright λΈλΌμ°μ μ€μΉ μ€...")
|
|
|
|
| 42 |
os.system("playwright install chromium --force")
|
| 43 |
+
print("λΈλΌμ°μ μ€μΉ μλ£!")
|
|
|
|
|
|
|
| 44 |
return True
|
| 45 |
except Exception as e:
|
| 46 |
+
print(f"λΈλΌμ°μ μ€μΉ μ€ν¨: {e}")
|
| 47 |
return False
|
| 48 |
|
| 49 |
+
def capture_trends_screenshot() -> Optional[Image.Image]:
|
| 50 |
+
"""μ€μκ° Google Trends μ€ν¬λ¦°μ· μΊ‘μ² (λ
립 λΈλΌμ°μ μΈμ
)"""
|
| 51 |
+
playwright_instance = None
|
| 52 |
+
browser = None
|
| 53 |
+
page = None
|
| 54 |
|
| 55 |
try:
|
| 56 |
+
print("μ λΈλΌμ°μ μΈμ
μμ...")
|
|
|
|
|
|
|
|
|
|
|
|
|
| 57 |
|
| 58 |
+
# λ
립μ μΈ Playwright μΈμ€ν΄μ€ μμ±
|
| 59 |
playwright_instance = sync_playwright().start()
|
| 60 |
|
| 61 |
+
# λΈλΌμ°μ μ€ν
|
| 62 |
browser = playwright_instance.chromium.launch(
|
| 63 |
headless=True,
|
| 64 |
args=[
|
|
|
|
| 67 |
'--disable-gpu',
|
| 68 |
'--disable-extensions',
|
| 69 |
'--disable-web-security',
|
| 70 |
+
'--disable-blink-features=AutomationControlled'
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 71 |
]
|
| 72 |
)
|
| 73 |
|
| 74 |
# μ νμ΄μ§ μμ±
|
| 75 |
page = browser.new_page()
|
| 76 |
+
page.set_viewport_size({"width": 1400, "height": 900})
|
|
|
|
|
|
|
|
|
|
|
|
|
| 77 |
page.set_extra_http_headers({
|
| 78 |
+
"Accept-Language": "ko-KR,ko;q=0.9,en;q=0.8",
|
| 79 |
+
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
|
| 80 |
})
|
| 81 |
|
| 82 |
+
# Google Trends μ€μκ° νμ΄μ§ λ‘λ
|
| 83 |
+
url = "https://trends.google.co.kr/trending?geo=KR"
|
| 84 |
+
print(f"νμ΄μ§ λ‘λ: {url}")
|
|
|
|
|
|
|
| 85 |
|
| 86 |
+
page.goto(url, wait_until="domcontentloaded", timeout=30000)
|
| 87 |
+
time.sleep(5)
|
| 88 |
|
| 89 |
+
# μΏ ν€ νμ
μ²λ¦¬
|
| 90 |
+
try:
|
| 91 |
+
accept_button = page.locator('button:has-text("λͺ¨λ μλ½"), button:has-text("Accept all"), button:has-text("μλ½")')
|
| 92 |
+
if accept_button.count() > 0:
|
| 93 |
+
accept_button.first.click(timeout=3000)
|
| 94 |
+
time.sleep(2)
|
| 95 |
+
except:
|
| 96 |
+
pass
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 97 |
|
| 98 |
+
# νμ΄μ§ μμ λ‘λ λκΈ°
|
| 99 |
+
time.sleep(3)
|
| 100 |
|
| 101 |
+
# μ€ν¬λ¦°μ· μΊ‘μ²
|
| 102 |
try:
|
| 103 |
+
main_content = page.locator('main, .main-content, [data-ve-type="main"], .trends-wrapper')
|
| 104 |
+
if main_content.count() > 0 and main_content.first.is_visible():
|
| 105 |
+
screenshot_bytes = main_content.first.screenshot(timeout=10000)
|
| 106 |
+
print("λ©μΈ 컨ν
μΈ μ€ν¬λ¦°μ· μΊ‘μ² μλ£")
|
|
|
|
| 107 |
else:
|
| 108 |
+
screenshot_bytes = page.screenshot(full_page=True, timeout=15000)
|
| 109 |
+
print("μ 체 νμ΄μ§ μ€ν¬λ¦°μ· μΊ‘μ² μλ£")
|
| 110 |
+
except Exception as e:
|
| 111 |
+
print(f"μ€ν¬λ¦°μ· μΊ‘μ² μ¬μλ: {e}")
|
| 112 |
+
screenshot_bytes = page.screenshot(timeout=15000)
|
| 113 |
+
print("μ¬μλ μ€ν¬λ¦°μ· μΊ‘μ² μλ£")
|
| 114 |
+
|
| 115 |
+
# μ΄λ―Έμ§ κ°μ²΄ μμ±
|
|
|
|
| 116 |
screenshot = Image.open(io.BytesIO(screenshot_bytes))
|
| 117 |
+
print("μ€ν¬λ¦°μ· μ²λ¦¬ μλ£")
|
| 118 |
return screenshot
|
| 119 |
+
|
| 120 |
except Exception as e:
|
| 121 |
print(f"μ€ν¬λ¦°μ· μΊ‘μ² μ€ν¨: {str(e)}")
|
| 122 |
return None
|
| 123 |
+
finally:
|
| 124 |
+
# λΈλΌμ°μ μ 리
|
| 125 |
+
try:
|
| 126 |
+
if page:
|
| 127 |
+
page.close()
|
| 128 |
+
if browser:
|
| 129 |
+
browser.close()
|
| 130 |
+
if playwright_instance:
|
| 131 |
+
playwright_instance.stop()
|
| 132 |
+
print("λΈλΌμ°μ μΈμ
μ 리 μλ£")
|
| 133 |
+
except Exception as e:
|
| 134 |
+
print(f"λΈλΌμ°μ μ 리 μ€ μ€λ₯: {e}")
|
| 135 |
|
| 136 |
def analyze_with_claude(image: Image.Image) -> str:
|
| 137 |
+
"""Claude APIλ‘ μ΄λ―Έμ§ λΆμ"""
|
| 138 |
+
if image is None:
|
| 139 |
+
return "λΆμν μ΄λ―Έμ§κ° μμ΅λλ€."
|
| 140 |
+
|
| 141 |
try:
|
| 142 |
client = get_claude_client()
|
| 143 |
|
|
|
|
| 146 |
image.save(buffer, format="PNG")
|
| 147 |
image_base64 = base64.b64encode(buffer.getvalue()).decode()
|
| 148 |
|
| 149 |
+
# λΆμ ν둬ννΈ
|
| 150 |
+
prompt = """Google Trends μ€μκ° μΈκΈ° κ²μμ΄ μ€ν¬λ¦°μ·μ λΆμν΄μ£ΌμΈμ.
|
| 151 |
+
|
| 152 |
+
λ€μ νλͺ©λ€μ μ€μ¬μΌλ‘ νκ΅μ΄λ‘ λΆμν΄μ£ΌμΈμ:
|
| 153 |
+
|
| 154 |
+
### A. μμ κ²μμ΄ μμ (κ²μλ κΈ°μ€)
|
| 155 |
+
κ²μλμ΄ λμ μμλλ‘ μ λ ¬νμ¬ λμ΄:
|
| 156 |
+
**1μ.** κ²μμ΄λͺ
(κ²μλ)
|
| 157 |
+
**2μ.** κ²μμ΄λͺ
(κ²μλ)
|
| 158 |
+
**3μ.** κ²μμ΄λͺ
(κ²μλ)
|
| 159 |
+
(μ΄λ° μμΌλ‘ 10μκΉμ§)
|
| 160 |
+
|
| 161 |
+
### B. μ£Όμ νΈλ λ ν€μλ
|
| 162 |
+
- νΉν μ£Όλͺ©ν λ§ν κ²μμ΄λ€
|
| 163 |
+
- κ²μλ μ¦κ° ν¨ν΄
|
| 164 |
+
|
| 165 |
+
### C. μΉ΄ν
κ³ λ¦¬λ³ νΉμ§
|
| 166 |
+
- μν°ν
μΈλ¨ΌνΈ, λ΄μ€, μ€ν¬μΈ λ± λΆμΌλ³ λν₯
|
| 167 |
+
|
| 168 |
+
### D. νΉμ΄μ¬ν λ° λΆμ
|
| 169 |
+
- κΈμμΉ ν€μλ
|
| 170 |
+
- μκΈ°μ /μ¬νμ λ°°κ²½ μΆμΈ‘
|
| 171 |
+
- μ 체μ μΈ κ΄μ¬μ¬ ν¨ν΄
|
| 172 |
+
|
| 173 |
+
**μ€μ μ§μμ¬ν:**
|
| 174 |
+
- λ§ν¬λ€μ΄ νμμ μ¬μ©νμ¬ **κ΅΅μ κΈμ¨**, *κΈ°μΈμ*, `μ½λ` λ±μΌλ‘ μ€μν λ΄μ©μ κ°μ‘°ν΄μ£ΌμΈμ
|
| 175 |
+
- κ²μμ΄λͺ
, μμΉ, μΉ΄ν
κ³ λ¦¬λͺ
λ± ν΅μ¬ μ 보λ **κ΅΅μ κΈμ¨**λ‘ νμ
|
| 176 |
+
- κΈμμΉλ₯ μ΄λ νΉλ³ν μμΉλ `λ°±ν±`μΌλ‘ κ°μΈμ κ°μ‘°
|
| 177 |
+
- μμ κ²μμ΄λ λ°λμ κ²μλ μμλλ‘ 1μλΆν° 10μκΉμ§ μ λ ¬
|
| 178 |
+
- ### ν€λ©μ μ΄λ―Έ μ 곡λλ λ΄μ©λ§ μμ±ν΄μ£ΌμΈμ"""
|
| 179 |
+
|
| 180 |
response = client.messages.create(
|
| 181 |
model="claude-3-5-sonnet-20241022",
|
| 182 |
+
max_tokens=1500,
|
| 183 |
messages=[
|
| 184 |
{
|
| 185 |
"role": "user",
|
|
|
|
| 201 |
]
|
| 202 |
)
|
| 203 |
|
| 204 |
+
analysis_result = response.content[0].text
|
| 205 |
+
print("Claude λΆμ μλ£")
|
| 206 |
+
return analysis_result
|
| 207 |
|
| 208 |
except Exception as e:
|
| 209 |
error_msg = f"Claude API λΆμ μ€ν¨: {str(e)}"
|
| 210 |
print(error_msg)
|
| 211 |
return error_msg
|
| 212 |
|
| 213 |
+
def perform_analysis():
|
| 214 |
+
"""μ€μκ° νΈλ λ λΆμ μν"""
|
| 215 |
+
global analyzed_data, last_update_time, is_updating
|
| 216 |
+
|
| 217 |
+
is_updating = True
|
| 218 |
+
print(f"\n{'='*60}")
|
| 219 |
+
print(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] μ€μκ° νΈλ λ λΆμ μμ")
|
| 220 |
+
print(f"{'='*60}")
|
| 221 |
|
| 222 |
try:
|
| 223 |
+
# λΈλΌμ°μ μ€μΉ νμΈ
|
| 224 |
+
install_browsers()
|
| 225 |
|
| 226 |
+
# μ€ν¬λ¦°μ· μΊ‘μ²
|
| 227 |
+
print("μ€ν¬λ¦°μ· μΊ‘μ² μ€...")
|
| 228 |
+
screenshot = capture_trends_screenshot()
|
| 229 |
+
|
| 230 |
+
if screenshot:
|
| 231 |
+
print("Claude λΆμ μ€...")
|
| 232 |
+
# Claude λΆμ
|
| 233 |
+
analysis = analyze_with_claude(screenshot)
|
| 234 |
+
|
| 235 |
+
# κ²°κ³Ό μ μ₯
|
| 236 |
+
analyzed_data["μ€μκ°"] = {
|
| 237 |
+
'analysis': analysis,
|
| 238 |
+
'timestamp': datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
| 239 |
+
}
|
| 240 |
+
|
| 241 |
+
last_update_time = datetime.now()
|
| 242 |
+
print(f"λΆμ μλ£ - {last_update_time.strftime('%Y-%m-%d %H:%M:%S')}")
|
| 243 |
+
return True
|
| 244 |
else:
|
| 245 |
+
print("μ€ν¬λ¦°μ· μΊ‘μ² μ€ν¨")
|
| 246 |
+
return False
|
| 247 |
+
|
| 248 |
+
except Exception as e:
|
| 249 |
+
print(f"λΆμ μ€ μ€λ₯: {str(e)}")
|
| 250 |
+
print(f"μμΈ μ€λ₯: {traceback.format_exc()}")
|
| 251 |
+
return False
|
| 252 |
+
finally:
|
| 253 |
+
is_updating = False
|
| 254 |
+
print(f"{'='*60}\n")
|
| 255 |
+
|
| 256 |
+
def auto_update_scheduler():
|
| 257 |
+
"""1μκ°λ§λ€ μλ μ
λ°μ΄νΈ"""
|
| 258 |
+
while True:
|
| 259 |
+
try:
|
| 260 |
+
print(f"[μ€μΌμ€λ¬] λ€μ μ
λ°μ΄νΈκΉμ§ 1μκ° λκΈ° μ€...")
|
| 261 |
+
time.sleep(3600) # 1μκ° λκΈ°
|
| 262 |
+
|
| 263 |
+
if not is_updating:
|
| 264 |
+
print(f"[μ€μΌμ€λ¬] μλ μ
λ°μ΄νΈ μμ")
|
| 265 |
+
perform_analysis()
|
| 266 |
else:
|
| 267 |
+
print(f"[μ€μΌμ€λ¬] μ΄λ―Έ μ
λ°μ΄νΈ μ€μ΄λ―λ‘ κ±΄λλ")
|
| 268 |
|
| 269 |
+
except Exception as e:
|
| 270 |
+
print(f"[μ€μΌμ€λ¬] μ€λ₯: {e}")
|
| 271 |
+
time.sleep(300) # 5λΆ ν μ¬μλ
|
| 272 |
|
| 273 |
+
def get_analysis_result() -> tuple:
|
| 274 |
+
"""μ μ₯λ λΆμ κ²°κ³Ό λ°ν"""
|
| 275 |
+
global analyzed_data, last_update_time
|
| 276 |
|
| 277 |
+
if "μ€μκ°" in analyzed_data:
|
| 278 |
+
data = analyzed_data["μ€μκ°"]
|
| 279 |
+
status = f"μ΅μ λ°μ΄ν° (μ
λ°μ΄νΈ: {data['timestamp']})"
|
| 280 |
+
return data['analysis'], status
|
| 281 |
+
else:
|
| 282 |
+
if last_update_time:
|
| 283 |
+
status = f"λ°μ΄ν° μμ (λ§μ§λ§ μ
λ°μ΄νΈ: {last_update_time.strftime('%H:%M:%S')})"
|
| 284 |
+
else:
|
| 285 |
+
status = "μμ§ λΆμ λ°μ΄ν°κ° μ€λΉλμ§ μμμ΅λλ€."
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 286 |
|
| 287 |
+
return "**λΆμ λ°μ΄ν°κ° μμ§ μ€λΉλμ§ μμμ΅λλ€.**", status
|
|
|
|
|
|
|
|
|
|
| 288 |
|
| 289 |
+
def create_interface():
|
| 290 |
+
"""Gradio μΈν°νμ΄μ€ μμ±"""
|
|
|
|
|
|
|
| 291 |
|
| 292 |
+
def show_cached_analysis():
|
| 293 |
+
"""μ μ₯λ λΆμ κ²°κ³Ό νμ"""
|
| 294 |
+
if "μ€μκ°" in analyzed_data:
|
| 295 |
+
data = analyzed_data["μ€μκ°"]
|
| 296 |
+
status = f"μΊμλ λΆμ κ²°κ³Ό (μ
λ°μ΄νΈ: {data['timestamp']})"
|
| 297 |
+
return data['analysis'], status
|
| 298 |
+
else:
|
| 299 |
+
if last_update_time:
|
| 300 |
+
status = f"μ μ₯λ λ°μ΄ν° μμ (λ§μ§λ§ μ
λ°μ΄νΈ: {last_update_time.strftime('%H:%M:%S')})"
|
| 301 |
+
else:
|
| 302 |
+
status = "μμ§ λΆμ λ°μ΄ν°κ° μ€λΉλμ§ μμμ΅λλ€. μ μ ν λ€μ μλν΄μ£ΌμΈμ."
|
| 303 |
+
|
| 304 |
+
return "**μ μ₯λ λΆμ λ°μ΄ν°κ° μμ΅λλ€.**\n\n'μ€μκ° μ
λ°μ΄νΈ' λ²νΌμ λλ¬μ£ΌμΈμ.", status
|
| 305 |
+
|
| 306 |
+
def perform_live_update():
|
| 307 |
+
"""μ€μκ° μ
λ°μ΄νΈ μν"""
|
| 308 |
+
if is_updating:
|
| 309 |
+
return "**νμ¬ μ
λ°μ΄νΈκ° μ§ν μ€μ
λλ€.**\n\nμ μ ν λ€μ μλν΄μ£ΌμΈμ.", "μ
λ°μ΄νΈ μ§ν μ€..."
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 310 |
|
| 311 |
+
print("μ€μκ° μ
λ°μ΄νΈ μμ²λ¨")
|
| 312 |
+
success = perform_analysis()
|
|
|
|
| 313 |
|
| 314 |
+
if success and analyzed_data:
|
| 315 |
+
data = analyzed_data["μ€μκ°"]
|
| 316 |
+
return data['analysis'], f"μ€μκ° μ
λ°μ΄νΈ μλ£ - {datetime.now().strftime('%H:%M:%S')}"
|
| 317 |
+
else:
|
| 318 |
+
return "**μ
λ°μ΄νΈμ μ€ν¨νμ΅λλ€.**\n\nμ μ ν λ€μ μλν΄μ£ΌμΈμ.", f"μ
λ°μ΄νΈ μ€ν¨ - {datetime.now().strftime('%H:%M:%S')}"
|
| 319 |
+
|
| 320 |
+
# Gradio μΈν°νμ΄μ€ ꡬμ±
|
| 321 |
+
with gr.Blocks(
|
| 322 |
+
title="Google Trends μΈκΈ° κ²μμ΄",
|
| 323 |
+
theme=gr.themes.Soft(),
|
| 324 |
+
css="""
|
| 325 |
+
@import url('https://fonts.googleapis.com/css2?family=Noto+Sans+KR:wght@300;400;500;700&display=swap');
|
| 326 |
+
|
| 327 |
+
* {
|
| 328 |
+
font-family: 'Noto Sans KR', sans-serif !important;
|
| 329 |
+
}
|
| 330 |
|
| 331 |
+
.markdown-content {
|
| 332 |
+
font-size: 14px;
|
| 333 |
+
line-height: 1.6;
|
| 334 |
+
background: #f8f9fa;
|
| 335 |
+
padding: 20px;
|
| 336 |
+
border-radius: 8px;
|
| 337 |
+
border: 1px solid #e9ecef;
|
| 338 |
+
}
|
| 339 |
|
| 340 |
+
.markdown-content h3 {
|
| 341 |
+
color: #2563eb;
|
| 342 |
+
margin-top: 20px;
|
| 343 |
+
margin-bottom: 10px;
|
| 344 |
+
}
|
| 345 |
|
| 346 |
+
.markdown-content strong {
|
| 347 |
+
color: #1d4ed8;
|
| 348 |
+
font-weight: 600;
|
| 349 |
+
}
|
| 350 |
+
"""
|
| 351 |
+
) as interface:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 352 |
|
| 353 |
+
gr.Markdown("# Google Trends μΈκΈ° κ²μμ΄")
|
|
|
|
| 354 |
|
| 355 |
with gr.Row():
|
| 356 |
with gr.Column(scale=1):
|
| 357 |
+
gr.Markdown("### λΆμ μ΅μ
")
|
|
|
|
|
|
|
|
|
|
|
|
|
| 358 |
|
| 359 |
+
with gr.Column():
|
| 360 |
+
cached_btn = gr.Button("μ€μκ° κ²μμ΄ λΆμ", variant="primary", size="lg")
|
|
|
|
|
|
|
|
|
|
| 361 |
|
| 362 |
+
gr.Markdown("---")
|
| 363 |
+
|
| 364 |
+
with gr.Column():
|
| 365 |
+
live_btn = gr.Button("μ€μκ° μ
λ°μ΄νΈ", variant="secondary", size="lg")
|
|
|
|
| 366 |
|
| 367 |
+
gr.Markdown("---")
|
|
|
|
|
|
|
| 368 |
|
| 369 |
+
status_output = gr.Textbox(
|
| 370 |
+
label="μν",
|
| 371 |
+
value="λΆμν μ΅μ
μ μ νν΄μ£ΌμΈμ.",
|
| 372 |
+
interactive=False
|
| 373 |
+
)
|
| 374 |
+
|
| 375 |
+
# Apeach μ΄λ―Έμ§ νμ
|
| 376 |
+
gr.Image(
|
| 377 |
+
value="apeach.png",
|
| 378 |
+
label=None,
|
| 379 |
+
show_label=False,
|
| 380 |
+
interactive=False,
|
| 381 |
+
height=200,
|
| 382 |
+
width=200
|
| 383 |
+
)
|
| 384 |
|
| 385 |
with gr.Column(scale=2):
|
| 386 |
+
analysis_output = gr.Markdown(
|
| 387 |
+
value="**λΆμ λ²νΌμ λλ¬μ£ΌμΈμ.**",
|
| 388 |
+
elem_classes=["markdown-content"],
|
| 389 |
+
height=600
|
|
|
|
|
|
|
| 390 |
)
|
|
|
|
|
|
|
|
|
|
| 391 |
|
| 392 |
+
# μ΄λ²€νΈ μ°κ²°
|
| 393 |
+
cached_btn.click(
|
| 394 |
+
fn=show_cached_analysis,
|
| 395 |
+
outputs=[analysis_output, status_output]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 396 |
)
|
| 397 |
|
| 398 |
+
live_btn.click(
|
| 399 |
+
fn=perform_live_update,
|
| 400 |
+
outputs=[analysis_output, status_output]
|
| 401 |
)
|
| 402 |
|
| 403 |
return interface
|
| 404 |
|
| 405 |
def main():
|
| 406 |
+
"""λ©μΈ ν¨μ"""
|
| 407 |
+
print("=" * 60)
|
| 408 |
+
print("Google Trends μΈκΈ° κ²μμ΄ μμ!")
|
| 409 |
+
print("=" * 60)
|
| 410 |
|
| 411 |
# API ν€ νμΈ
|
| 412 |
api_key = os.getenv("ANTHROPIC_API_KEY")
|
| 413 |
if not api_key:
|
| 414 |
+
print("ANTHROPIC_API_KEY νκ²½λ³μκ° μ€μ λμ§ μμμ΅λλ€.")
|
| 415 |
+
print(" .env νμΌμ ANTHROPIC_API_KEY=your_key_here λ₯Ό μΆκ°νμΈμ.")
|
| 416 |
return
|
| 417 |
else:
|
| 418 |
+
print("Claude API ν€ νμΈλ¨")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 419 |
|
| 420 |
+
# μ΄κΈ° λΆμ μν
|
| 421 |
+
print("μ΄κΈ° λΆμ μμ...")
|
| 422 |
+
initial_success = perform_analysis()
|
| 423 |
|
| 424 |
+
if initial_success:
|
| 425 |
+
print("μ΄κΈ° λΆμ μλ£")
|
| 426 |
+
else:
|
| 427 |
+
print("μ΄κΈ° λΆμ μ€ν¨, μλΉμ€λ κ³μ μ§νλ©λλ€.")
|
|
|
|
|
|
|
| 428 |
|
| 429 |
+
# μλ μ
λ°μ΄νΈ μ€μΌμ€λ¬ μμ
|
| 430 |
+
print("μλ μ
λ°μ΄νΈ μ€μΌμ€λ¬ μμ (1μκ° κ°κ²©)")
|
| 431 |
+
global update_thread
|
| 432 |
+
update_thread = threading.Thread(target=auto_update_scheduler, daemon=True)
|
| 433 |
+
update_thread.start()
|
| 434 |
|
| 435 |
+
print("λͺ¨λ μ€λΉ μλ£!")
|
| 436 |
+
print("=" * 60)
|
| 437 |
|
| 438 |
+
# Gradio μΈν°νμ΄μ€ μ€ν
|
| 439 |
+
try:
|
| 440 |
+
interface = create_interface()
|
| 441 |
+
interface.launch(
|
| 442 |
+
server_name="0.0.0.0",
|
| 443 |
+
server_port=7860,
|
| 444 |
+
share=False,
|
| 445 |
+
show_error=True,
|
| 446 |
+
quiet=False
|
| 447 |
+
)
|
| 448 |
+
except Exception as e:
|
| 449 |
+
print(f"μΈν°νμ΄μ€ μμ μ€ν¨: {e}")
|
| 450 |
+
print(f"μμΈ μ€λ₯: {traceback.format_exc()}")
|
| 451 |
|
| 452 |
if __name__ == "__main__":
|
| 453 |
main()
|