Add admin features, responsive UI, user reviews, visit stats, and channel-colored markers
- Admin: video management with Google Maps match status, manual restaurant mapping, restaurant remap on name change - Admin: user management tab with favorites/reviews detail - Admin: channel deletion fix for IDs with slashes - Frontend: responsive mobile layout (map top, list bottom, 2-row header) - Frontend: channel-colored map markers with legend - Frontend: my reviews list, favorites toggle, visit counter overlay - Frontend: force light mode for dark theme devices - Backend: visit tracking (site_visits table), user reviews endpoint - Backend: bulk transcript/extract streaming, geocode key fixes - Nginx config for production deployment Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -1,9 +1,11 @@
|
|||||||
"""FastAPI application entry point."""
|
"""FastAPI application entry point."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
from fastapi import FastAPI
|
from fastapi import FastAPI
|
||||||
from fastapi.middleware.cors import CORSMiddleware
|
from fastapi.middleware.cors import CORSMiddleware
|
||||||
|
|
||||||
from api.routes import restaurants, channels, videos, search, auth, reviews
|
from api.routes import restaurants, channels, videos, search, auth, reviews, admin_users, stats
|
||||||
|
|
||||||
app = FastAPI(
|
app = FastAPI(
|
||||||
title="Tasteby API",
|
title="Tasteby API",
|
||||||
@@ -25,6 +27,8 @@ app.include_router(videos.router, prefix="/api/videos", tags=["videos"])
|
|||||||
app.include_router(search.router, prefix="/api/search", tags=["search"])
|
app.include_router(search.router, prefix="/api/search", tags=["search"])
|
||||||
app.include_router(auth.router, prefix="/api/auth", tags=["auth"])
|
app.include_router(auth.router, prefix="/api/auth", tags=["auth"])
|
||||||
app.include_router(reviews.router, prefix="/api", tags=["reviews"])
|
app.include_router(reviews.router, prefix="/api", tags=["reviews"])
|
||||||
|
app.include_router(admin_users.router, prefix="/api/admin/users", tags=["admin-users"])
|
||||||
|
app.include_router(stats.router, prefix="/api/stats", tags=["stats"])
|
||||||
|
|
||||||
|
|
||||||
@app.get("/api/health")
|
@app.get("/api/health")
|
||||||
|
|||||||
93
backend/api/routes/admin_users.py
Normal file
93
backend/api/routes/admin_users.py
Normal file
@@ -0,0 +1,93 @@
|
|||||||
|
"""Admin user management API routes."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from fastapi import APIRouter, Query
|
||||||
|
|
||||||
|
from core.db import conn
|
||||||
|
|
||||||
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("")
|
||||||
|
def list_users(
|
||||||
|
limit: int = Query(50, le=200),
|
||||||
|
offset: int = Query(0, ge=0),
|
||||||
|
):
|
||||||
|
"""List all users with favorite/review counts."""
|
||||||
|
sql = """
|
||||||
|
SELECT u.id, u.email, u.nickname, u.avatar_url, u.provider, u.created_at,
|
||||||
|
NVL(fav.cnt, 0) AS favorite_count,
|
||||||
|
NVL(rev.cnt, 0) AS review_count
|
||||||
|
FROM tasteby_users u
|
||||||
|
LEFT JOIN (
|
||||||
|
SELECT user_id, COUNT(*) AS cnt FROM user_favorites GROUP BY user_id
|
||||||
|
) fav ON fav.user_id = u.id
|
||||||
|
LEFT JOIN (
|
||||||
|
SELECT user_id, COUNT(*) AS cnt FROM user_reviews GROUP BY user_id
|
||||||
|
) rev ON rev.user_id = u.id
|
||||||
|
ORDER BY u.created_at DESC
|
||||||
|
OFFSET :off ROWS FETCH NEXT :lim ROWS ONLY
|
||||||
|
"""
|
||||||
|
count_sql = "SELECT COUNT(*) FROM tasteby_users"
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(count_sql)
|
||||||
|
total = cur.fetchone()[0]
|
||||||
|
cur.execute(sql, {"off": offset, "lim": limit})
|
||||||
|
cols = [d[0].lower() for d in cur.description]
|
||||||
|
rows = [dict(zip(cols, row)) for row in cur.fetchall()]
|
||||||
|
for r in rows:
|
||||||
|
if r.get("created_at"):
|
||||||
|
r["created_at"] = r["created_at"].isoformat()
|
||||||
|
return {"users": rows, "total": total}
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/{user_id}/favorites")
|
||||||
|
def get_user_favorites(user_id: str):
|
||||||
|
"""Get a user's favorite restaurants."""
|
||||||
|
sql = """
|
||||||
|
SELECT r.id, r.name, r.address, r.region, r.cuisine_type,
|
||||||
|
r.rating, r.business_status, f.created_at
|
||||||
|
FROM user_favorites f
|
||||||
|
JOIN restaurants r ON r.id = f.restaurant_id
|
||||||
|
WHERE f.user_id = :u
|
||||||
|
ORDER BY f.created_at DESC
|
||||||
|
"""
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(sql, {"u": user_id})
|
||||||
|
cols = [d[0].lower() for d in cur.description]
|
||||||
|
rows = [dict(zip(cols, row)) for row in cur.fetchall()]
|
||||||
|
for r in rows:
|
||||||
|
if r.get("created_at"):
|
||||||
|
r["created_at"] = r["created_at"].isoformat()
|
||||||
|
return rows
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/{user_id}/reviews")
|
||||||
|
def get_user_reviews(user_id: str):
|
||||||
|
"""Get a user's reviews with restaurant names."""
|
||||||
|
sql = """
|
||||||
|
SELECT r.id, r.restaurant_id, r.rating, r.review_text,
|
||||||
|
r.visited_at, r.created_at,
|
||||||
|
rest.name AS restaurant_name
|
||||||
|
FROM user_reviews r
|
||||||
|
LEFT JOIN restaurants rest ON rest.id = r.restaurant_id
|
||||||
|
WHERE r.user_id = :u
|
||||||
|
ORDER BY r.created_at DESC
|
||||||
|
"""
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(sql, {"u": user_id})
|
||||||
|
cols = [d[0].lower() for d in cur.description]
|
||||||
|
rows = [dict(zip(cols, row)) for row in cur.fetchall()]
|
||||||
|
for r in rows:
|
||||||
|
# Handle CLOB
|
||||||
|
if hasattr(r.get("review_text"), "read"):
|
||||||
|
r["review_text"] = r["review_text"].read()
|
||||||
|
if r.get("visited_at"):
|
||||||
|
r["visited_at"] = r["visited_at"].isoformat()
|
||||||
|
if r.get("created_at"):
|
||||||
|
r["created_at"] = r["created_at"].isoformat()
|
||||||
|
return rows
|
||||||
@@ -1,16 +1,24 @@
|
|||||||
"""Channel API routes."""
|
"""Channel API routes."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
from concurrent.futures import ThreadPoolExecutor
|
||||||
|
|
||||||
from fastapi import APIRouter, HTTPException
|
from fastapi import APIRouter, HTTPException
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
|
|
||||||
from core import youtube
|
from core import youtube
|
||||||
|
|
||||||
|
_executor = ThreadPoolExecutor(max_workers=4)
|
||||||
|
|
||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
class ChannelCreate(BaseModel):
|
class ChannelCreate(BaseModel):
|
||||||
channel_id: str
|
channel_id: str
|
||||||
channel_name: str
|
channel_name: str
|
||||||
|
title_filter: str | None = None
|
||||||
|
|
||||||
|
|
||||||
@router.get("")
|
@router.get("")
|
||||||
@@ -21,7 +29,7 @@ def list_channels():
|
|||||||
@router.post("", status_code=201)
|
@router.post("", status_code=201)
|
||||||
def create_channel(body: ChannelCreate):
|
def create_channel(body: ChannelCreate):
|
||||||
try:
|
try:
|
||||||
row_id = youtube.add_channel(body.channel_id, body.channel_name)
|
row_id = youtube.add_channel(body.channel_id, body.channel_name, body.title_filter)
|
||||||
return {"id": row_id, "channel_id": body.channel_id}
|
return {"id": row_id, "channel_id": body.channel_id}
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
if "UQ_CHANNELS_CID" in str(e).upper():
|
if "UQ_CHANNELS_CID" in str(e).upper():
|
||||||
@@ -29,18 +37,53 @@ def create_channel(body: ChannelCreate):
|
|||||||
raise
|
raise
|
||||||
|
|
||||||
|
|
||||||
@router.post("/{channel_id}/scan")
|
def _do_scan(channel_id: str, full: bool):
|
||||||
def scan_channel(channel_id: str):
|
"""Sync scan logic, runs in thread pool."""
|
||||||
"""Trigger a scan for new videos from this channel."""
|
|
||||||
channels = youtube.get_active_channels()
|
channels = youtube.get_active_channels()
|
||||||
ch = next((c for c in channels if c["channel_id"] == channel_id), None)
|
ch = next((c for c in channels if c["channel_id"] == channel_id), None)
|
||||||
if not ch:
|
if not ch:
|
||||||
raise HTTPException(404, "Channel not found")
|
return None
|
||||||
|
|
||||||
videos = youtube.fetch_channel_videos(channel_id, max_results=50)
|
after = None if full else youtube.get_latest_video_date(ch["id"])
|
||||||
new_count = 0
|
title_filter = ch.get("title_filter")
|
||||||
for v in videos:
|
existing_vids = youtube.get_existing_video_ids(ch["id"])
|
||||||
row_id = youtube.save_video(ch["id"], v)
|
|
||||||
if row_id:
|
candidates = []
|
||||||
new_count += 1
|
total_fetched = 0
|
||||||
return {"total_fetched": len(videos), "new_videos": new_count}
|
for videos_page in youtube.fetch_channel_videos_iter(channel_id, published_after=after):
|
||||||
|
total_fetched += len(videos_page)
|
||||||
|
new_in_page = 0
|
||||||
|
for v in videos_page:
|
||||||
|
if title_filter and title_filter not in v["title"]:
|
||||||
|
continue
|
||||||
|
if v["video_id"] in existing_vids:
|
||||||
|
continue
|
||||||
|
candidates.append(v)
|
||||||
|
new_in_page += 1
|
||||||
|
if not full and new_in_page == 0 and total_fetched > 50:
|
||||||
|
break
|
||||||
|
|
||||||
|
new_count = youtube.save_videos_batch(ch["id"], candidates)
|
||||||
|
return {"total_fetched": total_fetched, "new_videos": new_count}
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/{channel_id}/scan")
|
||||||
|
async def scan_channel(channel_id: str, full: bool = False):
|
||||||
|
"""Trigger a scan for new videos from this channel (non-blocking)."""
|
||||||
|
loop = asyncio.get_event_loop()
|
||||||
|
result = await loop.run_in_executor(_executor, _do_scan, channel_id, full)
|
||||||
|
if result is None:
|
||||||
|
raise HTTPException(404, "Channel not found")
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
|
@router.delete("/{channel_id:path}")
|
||||||
|
def delete_channel(channel_id: str):
|
||||||
|
"""Deactivate a channel. Accepts channel_id or DB id."""
|
||||||
|
deleted = youtube.deactivate_channel(channel_id)
|
||||||
|
if not deleted:
|
||||||
|
# Try by DB id
|
||||||
|
deleted = youtube.deactivate_channel_by_db_id(channel_id)
|
||||||
|
if not deleted:
|
||||||
|
raise HTTPException(404, "Channel not found")
|
||||||
|
return {"ok": True}
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
"""Restaurant API routes."""
|
"""Restaurant API routes."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
from fastapi import APIRouter, HTTPException, Query
|
from fastapi import APIRouter, HTTPException, Query
|
||||||
|
|
||||||
from core import restaurant
|
from core import restaurant
|
||||||
@@ -13,8 +15,9 @@ def list_restaurants(
|
|||||||
offset: int = Query(0, ge=0),
|
offset: int = Query(0, ge=0),
|
||||||
cuisine: str | None = None,
|
cuisine: str | None = None,
|
||||||
region: str | None = None,
|
region: str | None = None,
|
||||||
|
channel: str | None = None,
|
||||||
):
|
):
|
||||||
return restaurant.get_all(limit=limit, offset=offset, cuisine=cuisine, region=region)
|
return restaurant.get_all(limit=limit, offset=offset, cuisine=cuisine, region=region, channel=channel)
|
||||||
|
|
||||||
|
|
||||||
@router.get("/{restaurant_id}")
|
@router.get("/{restaurant_id}")
|
||||||
@@ -25,6 +28,45 @@ def get_restaurant(restaurant_id: str):
|
|||||||
return r
|
return r
|
||||||
|
|
||||||
|
|
||||||
|
@router.put("/{restaurant_id}")
|
||||||
|
def update_restaurant(restaurant_id: str, body: dict):
|
||||||
|
from core.db import conn
|
||||||
|
r = restaurant.get_by_id(restaurant_id)
|
||||||
|
if not r:
|
||||||
|
raise HTTPException(404, "Restaurant not found")
|
||||||
|
|
||||||
|
allowed = ("name", "address", "region", "cuisine_type", "price_range",
|
||||||
|
"phone", "website", "latitude", "longitude")
|
||||||
|
sets = []
|
||||||
|
params: dict = {"rid": restaurant_id}
|
||||||
|
for field in allowed:
|
||||||
|
if field in body:
|
||||||
|
sets.append(f"{field} = :{field}")
|
||||||
|
params[field] = body[field] if body[field] != "" else None
|
||||||
|
if not sets:
|
||||||
|
raise HTTPException(400, "No fields to update")
|
||||||
|
sets.append("updated_at = SYSTIMESTAMP")
|
||||||
|
sql = f"UPDATE restaurants SET {', '.join(sets)} WHERE id = :rid"
|
||||||
|
with conn() as c:
|
||||||
|
c.cursor().execute(sql, params)
|
||||||
|
return {"ok": True}
|
||||||
|
|
||||||
|
|
||||||
|
@router.delete("/{restaurant_id}")
|
||||||
|
def delete_restaurant(restaurant_id: str):
|
||||||
|
from core.db import conn
|
||||||
|
r = restaurant.get_by_id(restaurant_id)
|
||||||
|
if not r:
|
||||||
|
raise HTTPException(404, "Restaurant not found")
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute("DELETE FROM restaurant_vectors WHERE restaurant_id = :rid", {"rid": restaurant_id})
|
||||||
|
cur.execute("DELETE FROM user_reviews WHERE restaurant_id = :rid", {"rid": restaurant_id})
|
||||||
|
cur.execute("DELETE FROM video_restaurants WHERE restaurant_id = :rid", {"rid": restaurant_id})
|
||||||
|
cur.execute("DELETE FROM restaurants WHERE id = :rid", {"rid": restaurant_id})
|
||||||
|
return {"ok": True}
|
||||||
|
|
||||||
|
|
||||||
@router.get("/{restaurant_id}/videos")
|
@router.get("/{restaurant_id}/videos")
|
||||||
def get_restaurant_videos(restaurant_id: str):
|
def get_restaurant_videos(restaurant_id: str):
|
||||||
r = restaurant.get_by_id(restaurant_id)
|
r = restaurant.get_by_id(restaurant_id)
|
||||||
|
|||||||
@@ -95,3 +95,74 @@ def list_my_reviews(
|
|||||||
):
|
):
|
||||||
"""List current user's reviews."""
|
"""List current user's reviews."""
|
||||||
return review.get_user_reviews(current_user["sub"], limit=limit, offset=offset)
|
return review.get_user_reviews(current_user["sub"], limit=limit, offset=offset)
|
||||||
|
|
||||||
|
|
||||||
|
# --- Favorites ---
|
||||||
|
|
||||||
|
@router.get("/restaurants/{restaurant_id}/favorite")
|
||||||
|
def get_favorite_status(
|
||||||
|
restaurant_id: str,
|
||||||
|
current_user: dict = Depends(get_current_user),
|
||||||
|
):
|
||||||
|
"""Check if current user has favorited this restaurant."""
|
||||||
|
from core.db import conn
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(
|
||||||
|
"SELECT id FROM user_favorites WHERE user_id = :u AND restaurant_id = :r",
|
||||||
|
{"u": current_user["sub"], "r": restaurant_id},
|
||||||
|
)
|
||||||
|
return {"favorited": cur.fetchone() is not None}
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/restaurants/{restaurant_id}/favorite")
|
||||||
|
def toggle_favorite(
|
||||||
|
restaurant_id: str,
|
||||||
|
current_user: dict = Depends(get_current_user),
|
||||||
|
):
|
||||||
|
"""Toggle favorite. Returns new state."""
|
||||||
|
from core.db import conn
|
||||||
|
user_id = current_user["sub"]
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(
|
||||||
|
"SELECT id FROM user_favorites WHERE user_id = :u AND restaurant_id = :r",
|
||||||
|
{"u": user_id, "r": restaurant_id},
|
||||||
|
)
|
||||||
|
row = cur.fetchone()
|
||||||
|
if row:
|
||||||
|
cur.execute("DELETE FROM user_favorites WHERE id = :fid", {"fid": row[0]})
|
||||||
|
return {"favorited": False}
|
||||||
|
else:
|
||||||
|
cur.execute(
|
||||||
|
"INSERT INTO user_favorites (user_id, restaurant_id) VALUES (:u, :r)",
|
||||||
|
{"u": user_id, "r": restaurant_id},
|
||||||
|
)
|
||||||
|
return {"favorited": True}
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/users/me/favorites")
|
||||||
|
def list_my_favorites(
|
||||||
|
current_user: dict = Depends(get_current_user),
|
||||||
|
):
|
||||||
|
"""List current user's favorite restaurants."""
|
||||||
|
from core.db import conn
|
||||||
|
sql = """
|
||||||
|
SELECT r.id, r.name, r.address, r.region, r.latitude, r.longitude,
|
||||||
|
r.cuisine_type, r.price_range, r.google_place_id,
|
||||||
|
r.business_status, r.rating, r.rating_count,
|
||||||
|
f.created_at
|
||||||
|
FROM user_favorites f
|
||||||
|
JOIN restaurants r ON r.id = f.restaurant_id
|
||||||
|
WHERE f.user_id = :u
|
||||||
|
ORDER BY f.created_at DESC
|
||||||
|
"""
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(sql, {"u": current_user["sub"]})
|
||||||
|
cols = [d[0].lower() for d in cur.description]
|
||||||
|
rows = [dict(zip(cols, row)) for row in cur.fetchall()]
|
||||||
|
for r in rows:
|
||||||
|
if r.get("created_at"):
|
||||||
|
r["created_at"] = r["created_at"].isoformat()
|
||||||
|
return rows
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
"""Search API routes — keyword + semantic search."""
|
"""Search API routes — keyword + semantic search."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
from fastapi import APIRouter, Query
|
from fastapi import APIRouter, Query
|
||||||
|
|
||||||
from core import restaurant, vector
|
from core import restaurant, vector
|
||||||
|
|||||||
43
backend/api/routes/stats.py
Normal file
43
backend/api/routes/stats.py
Normal file
@@ -0,0 +1,43 @@
|
|||||||
|
"""Site visit statistics API."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from fastapi import APIRouter
|
||||||
|
|
||||||
|
from core.db import conn
|
||||||
|
|
||||||
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/visit")
|
||||||
|
def record_visit():
|
||||||
|
"""Record a page visit. Increments today's count."""
|
||||||
|
sql = """
|
||||||
|
MERGE INTO site_visits sv
|
||||||
|
USING (SELECT TRUNC(SYSDATE) AS d FROM dual) src
|
||||||
|
ON (sv.visit_date = src.d)
|
||||||
|
WHEN MATCHED THEN UPDATE SET sv.visit_count = sv.visit_count + 1
|
||||||
|
WHEN NOT MATCHED THEN INSERT (visit_date, visit_count) VALUES (src.d, 1)
|
||||||
|
"""
|
||||||
|
with conn() as c:
|
||||||
|
c.cursor().execute(sql)
|
||||||
|
return {"ok": True}
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/visits")
|
||||||
|
def get_visits():
|
||||||
|
"""Return today's visit count and all-time total."""
|
||||||
|
sql_today = """
|
||||||
|
SELECT NVL(visit_count, 0) FROM site_visits WHERE visit_date = TRUNC(SYSDATE)
|
||||||
|
"""
|
||||||
|
sql_total = """
|
||||||
|
SELECT NVL(SUM(visit_count), 0) FROM site_visits
|
||||||
|
"""
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(sql_today)
|
||||||
|
row = cur.fetchone()
|
||||||
|
today = int(row[0]) if row else 0
|
||||||
|
cur.execute(sql_total)
|
||||||
|
total = int(cur.fetchone()[0])
|
||||||
|
return {"today": today, "total": total}
|
||||||
@@ -1,17 +1,29 @@
|
|||||||
"""Video API routes."""
|
"""Video API routes."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
import json as _json
|
||||||
|
import logging
|
||||||
|
import random
|
||||||
|
import time
|
||||||
|
from concurrent.futures import ThreadPoolExecutor
|
||||||
|
|
||||||
from fastapi import APIRouter, Query
|
from fastapi import APIRouter, Query
|
||||||
|
from fastapi.responses import StreamingResponse
|
||||||
|
|
||||||
from core.db import conn
|
from core.db import conn
|
||||||
from core.pipeline import process_pending
|
from core.pipeline import process_pending
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
|
_executor = ThreadPoolExecutor(max_workers=4)
|
||||||
|
|
||||||
|
|
||||||
@router.get("")
|
@router.get("")
|
||||||
def list_videos(
|
def list_videos(
|
||||||
status: str | None = None,
|
status: str | None = None,
|
||||||
limit: int = Query(50, le=200),
|
limit: int = Query(50, le=500),
|
||||||
offset: int = Query(0, ge=0),
|
offset: int = Query(0, ge=0),
|
||||||
):
|
):
|
||||||
conditions = []
|
conditions = []
|
||||||
@@ -23,7 +35,11 @@ def list_videos(
|
|||||||
where = ("WHERE " + " AND ".join(conditions)) if conditions else ""
|
where = ("WHERE " + " AND ".join(conditions)) if conditions else ""
|
||||||
sql = f"""
|
sql = f"""
|
||||||
SELECT v.id, v.video_id, v.title, v.url, v.status,
|
SELECT v.id, v.video_id, v.title, v.url, v.status,
|
||||||
v.published_at, c.channel_name
|
v.published_at, c.channel_name,
|
||||||
|
CASE WHEN v.transcript_text IS NOT NULL AND dbms_lob.getlength(v.transcript_text) > 0 THEN 1 ELSE 0 END as has_transcript,
|
||||||
|
CASE WHEN v.llm_raw_response IS NOT NULL AND dbms_lob.getlength(v.llm_raw_response) > 0 THEN 1 ELSE 0 END as has_llm,
|
||||||
|
(SELECT COUNT(*) FROM video_restaurants vr WHERE vr.video_id = v.id) as restaurant_count,
|
||||||
|
(SELECT COUNT(*) FROM video_restaurants vr JOIN restaurants r ON r.id = vr.restaurant_id WHERE vr.video_id = v.id AND r.google_place_id IS NOT NULL) as matched_count
|
||||||
FROM videos v
|
FROM videos v
|
||||||
JOIN channels c ON c.id = v.channel_id
|
JOIN channels c ON c.id = v.channel_id
|
||||||
{where}
|
{where}
|
||||||
@@ -41,12 +57,604 @@ def list_videos(
|
|||||||
d = dict(zip(cols, row))
|
d = dict(zip(cols, row))
|
||||||
if d.get("published_at"):
|
if d.get("published_at"):
|
||||||
d["published_at"] = d["published_at"].isoformat()
|
d["published_at"] = d["published_at"].isoformat()
|
||||||
|
d["has_transcript"] = bool(d.get("has_transcript"))
|
||||||
|
d["has_llm"] = bool(d.get("has_llm"))
|
||||||
|
d["restaurant_count"] = d.get("restaurant_count", 0)
|
||||||
|
d["matched_count"] = d.get("matched_count", 0)
|
||||||
results.append(d)
|
results.append(d)
|
||||||
return results
|
return results
|
||||||
|
|
||||||
|
|
||||||
|
def _get_unprocessed_videos() -> list[dict]:
|
||||||
|
"""Get videos that have transcripts but no LLM extraction."""
|
||||||
|
sql = """
|
||||||
|
SELECT v.id, v.video_id, v.title, v.url, v.transcript_text
|
||||||
|
FROM videos v
|
||||||
|
WHERE v.transcript_text IS NOT NULL
|
||||||
|
AND dbms_lob.getlength(v.transcript_text) > 0
|
||||||
|
AND (v.llm_raw_response IS NULL OR dbms_lob.getlength(v.llm_raw_response) = 0)
|
||||||
|
AND v.status != 'skip'
|
||||||
|
ORDER BY v.published_at DESC
|
||||||
|
"""
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(sql)
|
||||||
|
rows = cur.fetchall()
|
||||||
|
result = []
|
||||||
|
for r in rows:
|
||||||
|
transcript = r[4]
|
||||||
|
if hasattr(transcript, "read"):
|
||||||
|
transcript = transcript.read()
|
||||||
|
result.append({
|
||||||
|
"id": r[0], "video_id": r[1], "title": r[2], "url": r[3],
|
||||||
|
"transcript": transcript,
|
||||||
|
})
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/bulk-extract/pending")
|
||||||
|
def bulk_extract_pending_count():
|
||||||
|
"""Get count of videos pending LLM extraction."""
|
||||||
|
videos = _get_unprocessed_videos()
|
||||||
|
return {"count": len(videos), "videos": [{"id": v["id"], "title": v["title"]} for v in videos]}
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/bulk-extract")
|
||||||
|
def bulk_extract():
|
||||||
|
"""Process all unextracted videos with random delays. Streams SSE progress."""
|
||||||
|
from core.pipeline import process_video_extract
|
||||||
|
|
||||||
|
videos = _get_unprocessed_videos()
|
||||||
|
|
||||||
|
def generate():
|
||||||
|
total = len(videos)
|
||||||
|
total_restaurants = 0
|
||||||
|
yield f"data: {_json.dumps({'type': 'start', 'total': total})}\n\n"
|
||||||
|
|
||||||
|
for i, v in enumerate(videos):
|
||||||
|
# Random delay (3-8 seconds) between requests to avoid bot detection
|
||||||
|
if i > 0:
|
||||||
|
delay = random.uniform(3.0, 8.0)
|
||||||
|
yield f"data: {_json.dumps({'type': 'wait', 'index': i, 'delay': round(delay, 1)})}\n\n"
|
||||||
|
time.sleep(delay)
|
||||||
|
|
||||||
|
yield f"data: {_json.dumps({'type': 'processing', 'index': i, 'title': v['title']})}\n\n"
|
||||||
|
|
||||||
|
try:
|
||||||
|
count = process_video_extract(
|
||||||
|
{"id": v["id"], "video_id": v["video_id"], "title": v["title"], "url": v["url"]},
|
||||||
|
v["transcript"],
|
||||||
|
)
|
||||||
|
total_restaurants += count
|
||||||
|
yield f"data: {_json.dumps({'type': 'done', 'index': i, 'title': v['title'], 'restaurants': count})}\n\n"
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("Bulk extract error for %s: %s", v["video_id"], e)
|
||||||
|
yield f"data: {_json.dumps({'type': 'error', 'index': i, 'title': v['title'], 'message': str(e)})}\n\n"
|
||||||
|
|
||||||
|
yield f"data: {_json.dumps({'type': 'complete', 'total': total, 'total_restaurants': total_restaurants})}\n\n"
|
||||||
|
|
||||||
|
return StreamingResponse(generate(), media_type="text/event-stream")
|
||||||
|
|
||||||
|
|
||||||
|
def _get_no_transcript_videos() -> list[dict]:
|
||||||
|
"""Get videos that have no transcript yet."""
|
||||||
|
sql = """
|
||||||
|
SELECT v.id, v.video_id, v.title
|
||||||
|
FROM videos v
|
||||||
|
WHERE (v.transcript_text IS NULL OR dbms_lob.getlength(v.transcript_text) = 0)
|
||||||
|
AND v.status != 'skip'
|
||||||
|
ORDER BY v.published_at DESC
|
||||||
|
"""
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(sql)
|
||||||
|
return [{"id": r[0], "video_id": r[1], "title": r[2]} for r in cur.fetchall()]
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/bulk-transcript/pending")
|
||||||
|
def bulk_transcript_pending_count():
|
||||||
|
"""Get count of videos without transcripts."""
|
||||||
|
videos = _get_no_transcript_videos()
|
||||||
|
return {"count": len(videos), "videos": [{"id": v["id"], "title": v["title"]} for v in videos]}
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/bulk-transcript")
|
||||||
|
def bulk_transcript():
|
||||||
|
"""Fetch transcripts for all videos missing them. Streams SSE progress."""
|
||||||
|
from core.youtube import get_transcript
|
||||||
|
|
||||||
|
videos = _get_no_transcript_videos()
|
||||||
|
|
||||||
|
def generate():
|
||||||
|
total = len(videos)
|
||||||
|
success = 0
|
||||||
|
yield f"data: {_json.dumps({'type': 'start', 'total': total})}\n\n"
|
||||||
|
|
||||||
|
for i, v in enumerate(videos):
|
||||||
|
# Random delay (5-15 seconds) to avoid bot detection — longer than LLM since this hits YouTube
|
||||||
|
if i > 0:
|
||||||
|
delay = random.uniform(5.0, 15.0)
|
||||||
|
yield f"data: {_json.dumps({'type': 'wait', 'index': i, 'delay': round(delay, 1)})}\n\n"
|
||||||
|
time.sleep(delay)
|
||||||
|
|
||||||
|
yield f"data: {_json.dumps({'type': 'processing', 'index': i, 'title': v['title']})}\n\n"
|
||||||
|
|
||||||
|
try:
|
||||||
|
transcript, source = get_transcript(v["video_id"])
|
||||||
|
if transcript:
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(
|
||||||
|
"UPDATE videos SET transcript_text = :txt WHERE id = :vid",
|
||||||
|
{"txt": transcript, "vid": v["id"]},
|
||||||
|
)
|
||||||
|
success += 1
|
||||||
|
yield f"data: {_json.dumps({'type': 'done', 'index': i, 'title': v['title'], 'source': source, 'length': len(transcript)})}\n\n"
|
||||||
|
else:
|
||||||
|
yield f"data: {_json.dumps({'type': 'error', 'index': i, 'title': v['title'], 'message': 'No transcript available'})}\n\n"
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("Bulk transcript error for %s: %s", v["video_id"], e)
|
||||||
|
yield f"data: {_json.dumps({'type': 'error', 'index': i, 'title': v['title'], 'message': str(e)})}\n\n"
|
||||||
|
|
||||||
|
yield f"data: {_json.dumps({'type': 'complete', 'total': total, 'success': success})}\n\n"
|
||||||
|
|
||||||
|
return StreamingResponse(generate(), media_type="text/event-stream")
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/extract/prompt")
|
||||||
|
def get_extract_prompt():
|
||||||
|
"""Get the current LLM extraction prompt template."""
|
||||||
|
from core.extractor import _EXTRACT_PROMPT
|
||||||
|
return {"prompt": _EXTRACT_PROMPT}
|
||||||
|
|
||||||
|
|
||||||
|
def _do_process(limit: int):
|
||||||
|
return {"restaurants_extracted": process_pending(limit)}
|
||||||
|
|
||||||
|
|
||||||
@router.post("/process")
|
@router.post("/process")
|
||||||
def trigger_processing(limit: int = Query(5, le=20)):
|
async def trigger_processing(limit: int = Query(5, le=20)):
|
||||||
"""Manually trigger processing of pending videos."""
|
"""Manually trigger processing of pending videos (non-blocking)."""
|
||||||
count = process_pending(limit)
|
loop = asyncio.get_event_loop()
|
||||||
return {"restaurants_extracted": count}
|
return await loop.run_in_executor(_executor, _do_process, limit)
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/{video_db_id}")
|
||||||
|
def get_video_detail(video_db_id: str):
|
||||||
|
"""Get video detail including transcript and extracted restaurants."""
|
||||||
|
from fastapi import HTTPException
|
||||||
|
import json
|
||||||
|
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute("""
|
||||||
|
SELECT v.id, v.video_id, v.title, v.url, v.status,
|
||||||
|
v.published_at, v.transcript_text,
|
||||||
|
c.channel_name
|
||||||
|
FROM videos v
|
||||||
|
JOIN channels c ON c.id = v.channel_id
|
||||||
|
WHERE v.id = :vid
|
||||||
|
""", {"vid": video_db_id})
|
||||||
|
row = cur.fetchone()
|
||||||
|
if not row:
|
||||||
|
raise HTTPException(404, "Video not found")
|
||||||
|
|
||||||
|
transcript = row[6]
|
||||||
|
if hasattr(transcript, "read"):
|
||||||
|
transcript = transcript.read()
|
||||||
|
|
||||||
|
# Get extracted restaurants for this video
|
||||||
|
cur.execute("""
|
||||||
|
SELECT r.id, r.name, r.address, r.cuisine_type, r.price_range, r.region,
|
||||||
|
vr.foods_mentioned, vr.evaluation, vr.guests,
|
||||||
|
r.google_place_id, r.latitude, r.longitude
|
||||||
|
FROM video_restaurants vr
|
||||||
|
JOIN restaurants r ON r.id = vr.restaurant_id
|
||||||
|
WHERE vr.video_id = :vid
|
||||||
|
""", {"vid": video_db_id})
|
||||||
|
|
||||||
|
restaurants = []
|
||||||
|
for rr in cur.fetchall():
|
||||||
|
foods_raw = rr[6].read() if hasattr(rr[6], "read") else rr[6]
|
||||||
|
eval_raw = rr[7].read() if hasattr(rr[7], "read") else rr[7]
|
||||||
|
guests_raw = rr[8].read() if hasattr(rr[8], "read") else rr[8]
|
||||||
|
|
||||||
|
def parse_json(val, default):
|
||||||
|
if val is None:
|
||||||
|
return default
|
||||||
|
if isinstance(val, (list, dict)):
|
||||||
|
return val
|
||||||
|
try:
|
||||||
|
return json.loads(val)
|
||||||
|
except (json.JSONDecodeError, ValueError):
|
||||||
|
return default
|
||||||
|
|
||||||
|
restaurants.append({
|
||||||
|
"restaurant_id": rr[0],
|
||||||
|
"name": rr[1],
|
||||||
|
"address": rr[2],
|
||||||
|
"cuisine_type": rr[3],
|
||||||
|
"price_range": rr[4],
|
||||||
|
"region": rr[5],
|
||||||
|
"foods_mentioned": parse_json(foods_raw, []),
|
||||||
|
"evaluation": parse_json(eval_raw, {}),
|
||||||
|
"guests": parse_json(guests_raw, []),
|
||||||
|
"google_place_id": rr[9],
|
||||||
|
"has_location": rr[10] is not None and rr[11] is not None,
|
||||||
|
})
|
||||||
|
|
||||||
|
return {
|
||||||
|
"id": row[0],
|
||||||
|
"video_id": row[1],
|
||||||
|
"title": row[2],
|
||||||
|
"url": row[3],
|
||||||
|
"status": row[4],
|
||||||
|
"published_at": row[5].isoformat() if row[5] else None,
|
||||||
|
"transcript": transcript,
|
||||||
|
"channel_name": row[7],
|
||||||
|
"restaurants": restaurants,
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def _do_fetch_transcript(video_db_id: str, mode: str):
|
||||||
|
from core.youtube import get_transcript
|
||||||
|
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute("SELECT video_id FROM videos WHERE id = :vid", {"vid": video_db_id})
|
||||||
|
row = cur.fetchone()
|
||||||
|
if not row:
|
||||||
|
return {"error": 404, "detail": "Video not found"}
|
||||||
|
video_id = row[0]
|
||||||
|
|
||||||
|
transcript, source = get_transcript(video_id, mode=mode)
|
||||||
|
if not transcript:
|
||||||
|
return {"error": 422, "detail": "Transcript unavailable for this video"}
|
||||||
|
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(
|
||||||
|
"UPDATE videos SET transcript_text = :txt WHERE id = :vid",
|
||||||
|
{"txt": transcript, "vid": video_db_id},
|
||||||
|
)
|
||||||
|
|
||||||
|
return {"ok": True, "length": len(transcript), "source": source}
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/{video_db_id}/fetch-transcript")
|
||||||
|
async def fetch_transcript(video_db_id: str, mode: str = Query("auto")):
|
||||||
|
"""Fetch and save transcript for a video (non-blocking)."""
|
||||||
|
from fastapi import HTTPException
|
||||||
|
|
||||||
|
if mode not in ("auto", "manual", "generated"):
|
||||||
|
raise HTTPException(400, "mode must be auto, manual, or generated")
|
||||||
|
|
||||||
|
loop = asyncio.get_event_loop()
|
||||||
|
result = await loop.run_in_executor(_executor, _do_fetch_transcript, video_db_id, mode)
|
||||||
|
if "error" in result:
|
||||||
|
raise HTTPException(result["error"], result["detail"])
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
|
def _do_extract(video_db_id: str, custom_prompt: str | None):
|
||||||
|
from core.pipeline import process_video_extract
|
||||||
|
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(
|
||||||
|
"SELECT id, video_id, title, url, transcript_text FROM videos WHERE id = :vid",
|
||||||
|
{"vid": video_db_id},
|
||||||
|
)
|
||||||
|
row = cur.fetchone()
|
||||||
|
if not row:
|
||||||
|
return {"error": 404, "detail": "Video not found"}
|
||||||
|
transcript = row[4]
|
||||||
|
if hasattr(transcript, "read"):
|
||||||
|
transcript = transcript.read()
|
||||||
|
if not transcript:
|
||||||
|
return {"error": 422, "detail": "No transcript available for this video"}
|
||||||
|
|
||||||
|
count = process_video_extract(
|
||||||
|
{"id": row[0], "video_id": row[1], "title": row[2], "url": row[3]},
|
||||||
|
transcript,
|
||||||
|
custom_prompt=custom_prompt,
|
||||||
|
)
|
||||||
|
return {"ok": True, "restaurants_extracted": count}
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/{video_db_id}/extract")
|
||||||
|
async def extract_restaurants_from_video(video_db_id: str, body: dict = None):
|
||||||
|
"""Run LLM extraction on an existing transcript (non-blocking)."""
|
||||||
|
from fastapi import HTTPException
|
||||||
|
custom_prompt = body.get("prompt") if body else None
|
||||||
|
loop = asyncio.get_event_loop()
|
||||||
|
result = await loop.run_in_executor(_executor, _do_extract, video_db_id, custom_prompt)
|
||||||
|
if "error" in result:
|
||||||
|
raise HTTPException(result["error"], result["detail"])
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/{video_db_id}/skip")
|
||||||
|
def skip_video(video_db_id: str):
|
||||||
|
"""Mark a video as skipped."""
|
||||||
|
from fastapi import HTTPException
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(
|
||||||
|
"UPDATE videos SET status = 'skip' WHERE id = :vid",
|
||||||
|
{"vid": video_db_id},
|
||||||
|
)
|
||||||
|
if cur.rowcount == 0:
|
||||||
|
raise HTTPException(404, "Video not found")
|
||||||
|
return {"ok": True}
|
||||||
|
|
||||||
|
|
||||||
|
@router.delete("/{video_db_id}")
|
||||||
|
def delete_video(video_db_id: str):
|
||||||
|
"""Delete a video and its related data."""
|
||||||
|
from core.db import conn as get_conn
|
||||||
|
with get_conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
# Delete vector embeddings for restaurants only linked to this video
|
||||||
|
cur.execute("""
|
||||||
|
DELETE FROM restaurant_vectors
|
||||||
|
WHERE restaurant_id IN (
|
||||||
|
SELECT vr.restaurant_id FROM video_restaurants vr
|
||||||
|
WHERE vr.video_id = :vid
|
||||||
|
AND NOT EXISTS (
|
||||||
|
SELECT 1 FROM video_restaurants vr2
|
||||||
|
WHERE vr2.restaurant_id = vr.restaurant_id
|
||||||
|
AND vr2.video_id != :vid
|
||||||
|
)
|
||||||
|
)
|
||||||
|
""", {"vid": video_db_id})
|
||||||
|
# Delete reviews for restaurants only linked to this video
|
||||||
|
cur.execute("""
|
||||||
|
DELETE FROM user_reviews
|
||||||
|
WHERE restaurant_id IN (
|
||||||
|
SELECT vr.restaurant_id FROM video_restaurants vr
|
||||||
|
WHERE vr.video_id = :vid
|
||||||
|
AND NOT EXISTS (
|
||||||
|
SELECT 1 FROM video_restaurants vr2
|
||||||
|
WHERE vr2.restaurant_id = vr.restaurant_id
|
||||||
|
AND vr2.video_id != :vid
|
||||||
|
)
|
||||||
|
)
|
||||||
|
""", {"vid": video_db_id})
|
||||||
|
# Delete restaurants only linked to this video
|
||||||
|
cur.execute("""
|
||||||
|
DELETE FROM restaurants
|
||||||
|
WHERE id IN (
|
||||||
|
SELECT vr.restaurant_id FROM video_restaurants vr
|
||||||
|
WHERE vr.video_id = :vid
|
||||||
|
AND NOT EXISTS (
|
||||||
|
SELECT 1 FROM video_restaurants vr2
|
||||||
|
WHERE vr2.restaurant_id = vr.restaurant_id
|
||||||
|
AND vr2.video_id != :vid
|
||||||
|
)
|
||||||
|
)
|
||||||
|
""", {"vid": video_db_id})
|
||||||
|
# Delete video-restaurant links
|
||||||
|
cur.execute("DELETE FROM video_restaurants WHERE video_id = :vid", {"vid": video_db_id})
|
||||||
|
# Delete the video
|
||||||
|
cur.execute("DELETE FROM videos WHERE id = :vid", {"vid": video_db_id})
|
||||||
|
if cur.rowcount == 0:
|
||||||
|
from fastapi import HTTPException
|
||||||
|
raise HTTPException(404, "Video not found")
|
||||||
|
return {"ok": True}
|
||||||
|
|
||||||
|
|
||||||
|
@router.put("/{video_db_id}")
|
||||||
|
def update_video(video_db_id: str, body: dict):
|
||||||
|
"""Update video title."""
|
||||||
|
from fastapi import HTTPException
|
||||||
|
title = body.get("title")
|
||||||
|
if not title:
|
||||||
|
raise HTTPException(400, "title is required")
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(
|
||||||
|
"UPDATE videos SET title = :title WHERE id = :vid",
|
||||||
|
{"title": title, "vid": video_db_id},
|
||||||
|
)
|
||||||
|
if cur.rowcount == 0:
|
||||||
|
raise HTTPException(404, "Video not found")
|
||||||
|
return {"ok": True}
|
||||||
|
|
||||||
|
|
||||||
|
@router.delete("/{video_db_id}/restaurants/{restaurant_id}")
|
||||||
|
def delete_video_restaurant(video_db_id: str, restaurant_id: str):
|
||||||
|
"""Delete a video-restaurant mapping. Also cleans up orphaned restaurant."""
|
||||||
|
from fastapi import HTTPException
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(
|
||||||
|
"DELETE FROM video_restaurants WHERE video_id = :vid AND restaurant_id = :rid",
|
||||||
|
{"vid": video_db_id, "rid": restaurant_id},
|
||||||
|
)
|
||||||
|
if cur.rowcount == 0:
|
||||||
|
raise HTTPException(404, "Mapping not found")
|
||||||
|
# Clean up orphaned restaurant (no other video links)
|
||||||
|
cur.execute("""
|
||||||
|
DELETE FROM restaurant_vectors WHERE restaurant_id = :rid
|
||||||
|
AND NOT EXISTS (SELECT 1 FROM video_restaurants WHERE restaurant_id = :rid)
|
||||||
|
""", {"rid": restaurant_id})
|
||||||
|
cur.execute("""
|
||||||
|
DELETE FROM user_reviews WHERE restaurant_id = :rid
|
||||||
|
AND NOT EXISTS (SELECT 1 FROM video_restaurants WHERE restaurant_id = :rid)
|
||||||
|
""", {"rid": restaurant_id})
|
||||||
|
cur.execute("""
|
||||||
|
DELETE FROM restaurants WHERE id = :rid
|
||||||
|
AND NOT EXISTS (SELECT 1 FROM video_restaurants WHERE restaurant_id = :rid)
|
||||||
|
""", {"rid": restaurant_id})
|
||||||
|
return {"ok": True}
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/{video_db_id}/restaurants/manual")
|
||||||
|
def add_manual_restaurant(video_db_id: str, body: dict):
|
||||||
|
"""Manually add a restaurant and link it to a video."""
|
||||||
|
from fastapi import HTTPException
|
||||||
|
from core import restaurant as rest_mod
|
||||||
|
from core.geocoding import geocode_restaurant
|
||||||
|
|
||||||
|
name = body.get("name", "").strip()
|
||||||
|
if not name:
|
||||||
|
raise HTTPException(400, "Restaurant name is required")
|
||||||
|
|
||||||
|
address = body.get("address", "").strip() or None
|
||||||
|
region = body.get("region", "").strip() or None
|
||||||
|
cuisine_type = body.get("cuisine_type", "").strip() or None
|
||||||
|
price_range = body.get("price_range", "").strip() or None
|
||||||
|
foods = body.get("foods_mentioned", [])
|
||||||
|
evaluation = body.get("evaluation", "").strip() or None
|
||||||
|
guests = body.get("guests", [])
|
||||||
|
|
||||||
|
# Geocode to get lat/lng and Google data
|
||||||
|
geo = geocode_restaurant(name, address or region or "")
|
||||||
|
if not geo:
|
||||||
|
raise HTTPException(400, f"'{name}' 위치를 찾을 수 없습니다. 주소를 입력해주세요.")
|
||||||
|
|
||||||
|
rid = rest_mod.upsert(
|
||||||
|
name=name,
|
||||||
|
address=geo.get("formatted_address") or address,
|
||||||
|
region=region,
|
||||||
|
latitude=geo["latitude"],
|
||||||
|
longitude=geo["longitude"],
|
||||||
|
cuisine_type=cuisine_type,
|
||||||
|
price_range=price_range,
|
||||||
|
google_place_id=geo.get("google_place_id"),
|
||||||
|
phone=geo.get("phone"),
|
||||||
|
website=geo.get("website"),
|
||||||
|
business_status=geo.get("business_status"),
|
||||||
|
rating=geo.get("rating"),
|
||||||
|
rating_count=geo.get("rating_count"),
|
||||||
|
)
|
||||||
|
|
||||||
|
link_id = rest_mod.link_video_restaurant(
|
||||||
|
video_db_id=video_db_id,
|
||||||
|
restaurant_id=rid,
|
||||||
|
foods=foods if isinstance(foods, list) else [],
|
||||||
|
evaluation=evaluation,
|
||||||
|
guests=guests if isinstance(guests, list) else [],
|
||||||
|
)
|
||||||
|
|
||||||
|
return {"ok": True, "restaurant_id": rid, "link_id": link_id}
|
||||||
|
|
||||||
|
|
||||||
|
@router.put("/{video_db_id}/restaurants/{restaurant_id}")
|
||||||
|
def update_video_restaurant(video_db_id: str, restaurant_id: str, body: dict):
|
||||||
|
"""Update restaurant info linked to a video.
|
||||||
|
|
||||||
|
If name changed, re-geocode and remap to a new restaurant record.
|
||||||
|
"""
|
||||||
|
from fastapi import HTTPException
|
||||||
|
import json as _json
|
||||||
|
|
||||||
|
# Check if name changed — need to remap
|
||||||
|
new_name = body.get("name", "").strip() if "name" in body else None
|
||||||
|
if new_name:
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute("SELECT name FROM restaurants WHERE id = :rid", {"rid": restaurant_id})
|
||||||
|
row = cur.fetchone()
|
||||||
|
old_name = row[0] if row else ""
|
||||||
|
|
||||||
|
if old_name != new_name:
|
||||||
|
# Name changed: geocode new restaurant, remap
|
||||||
|
from core import restaurant as rest_mod
|
||||||
|
from core.geocoding import geocode_restaurant
|
||||||
|
|
||||||
|
address = body.get("address", "").strip() or body.get("region", "").strip() or ""
|
||||||
|
geo = geocode_restaurant(new_name, address)
|
||||||
|
if not geo:
|
||||||
|
raise HTTPException(400, f"'{new_name}' 위치를 찾을 수 없습니다.")
|
||||||
|
|
||||||
|
new_rid = rest_mod.upsert(
|
||||||
|
name=new_name,
|
||||||
|
address=geo.get("formatted_address") or body.get("address"),
|
||||||
|
region=body.get("region"),
|
||||||
|
latitude=geo["latitude"],
|
||||||
|
longitude=geo["longitude"],
|
||||||
|
cuisine_type=body.get("cuisine_type"),
|
||||||
|
price_range=body.get("price_range"),
|
||||||
|
google_place_id=geo.get("google_place_id"),
|
||||||
|
phone=geo.get("phone"),
|
||||||
|
website=geo.get("website"),
|
||||||
|
business_status=geo.get("business_status"),
|
||||||
|
rating=geo.get("rating"),
|
||||||
|
rating_count=geo.get("rating_count"),
|
||||||
|
)
|
||||||
|
|
||||||
|
# Read existing mapping data, delete old, create new
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(
|
||||||
|
"SELECT foods_mentioned, evaluation, guests FROM video_restaurants WHERE video_id = :vid AND restaurant_id = :rid",
|
||||||
|
{"vid": video_db_id, "rid": restaurant_id},
|
||||||
|
)
|
||||||
|
old_vr = cur.fetchone()
|
||||||
|
|
||||||
|
cur.execute(
|
||||||
|
"DELETE FROM video_restaurants WHERE video_id = :vid AND restaurant_id = :rid",
|
||||||
|
{"vid": video_db_id, "rid": restaurant_id},
|
||||||
|
)
|
||||||
|
|
||||||
|
# Build new mapping values from body or old data
|
||||||
|
def _parse(val, default):
|
||||||
|
if val is None:
|
||||||
|
return default
|
||||||
|
if hasattr(val, "read"):
|
||||||
|
val = val.read()
|
||||||
|
if isinstance(val, (list, dict)):
|
||||||
|
return val
|
||||||
|
try:
|
||||||
|
return _json.loads(val)
|
||||||
|
except Exception:
|
||||||
|
return default
|
||||||
|
|
||||||
|
old_foods = _parse(old_vr[0], []) if old_vr else []
|
||||||
|
old_eval = _parse(old_vr[1], {}) if old_vr else {}
|
||||||
|
old_guests = _parse(old_vr[2], []) if old_vr else []
|
||||||
|
|
||||||
|
foods = body.get("foods_mentioned", old_foods)
|
||||||
|
evaluation = body.get("evaluation", old_eval)
|
||||||
|
guests = body.get("guests", old_guests)
|
||||||
|
|
||||||
|
eval_text = evaluation.get("text", "") if isinstance(evaluation, dict) else str(evaluation or "")
|
||||||
|
|
||||||
|
rest_mod.link_video_restaurant(
|
||||||
|
video_db_id=video_db_id,
|
||||||
|
restaurant_id=new_rid,
|
||||||
|
foods=foods if isinstance(foods, list) else [],
|
||||||
|
evaluation=eval_text or None,
|
||||||
|
guests=guests if isinstance(guests, list) else [],
|
||||||
|
)
|
||||||
|
|
||||||
|
return {"ok": True, "remapped": True, "new_restaurant_id": new_rid}
|
||||||
|
|
||||||
|
# No name change — update in place
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
r_sets = []
|
||||||
|
r_params: dict = {"rid": restaurant_id}
|
||||||
|
for field in ("name", "address", "region", "cuisine_type", "price_range"):
|
||||||
|
if field in body:
|
||||||
|
r_sets.append(f"{field} = :{field}")
|
||||||
|
r_params[field] = body[field]
|
||||||
|
if r_sets:
|
||||||
|
r_sets.append("updated_at = SYSTIMESTAMP")
|
||||||
|
sql = f"UPDATE restaurants SET {', '.join(r_sets)} WHERE id = :rid"
|
||||||
|
cur.execute(sql, r_params)
|
||||||
|
|
||||||
|
vr_params: dict = {"vid": video_db_id, "rid": restaurant_id}
|
||||||
|
vr_sets = []
|
||||||
|
for field in ("foods_mentioned", "evaluation", "guests"):
|
||||||
|
if field in body:
|
||||||
|
vr_sets.append(f"{field} = :{field}")
|
||||||
|
val = body[field]
|
||||||
|
vr_params[field] = _json.dumps(val, ensure_ascii=False) if isinstance(val, (list, dict)) else val
|
||||||
|
if vr_sets:
|
||||||
|
sql = f"UPDATE video_restaurants SET {', '.join(vr_sets)} WHERE video_id = :vid AND restaurant_id = :rid"
|
||||||
|
cur.execute(sql, vr_params)
|
||||||
|
|
||||||
|
return {"ok": True}
|
||||||
|
|||||||
@@ -56,7 +56,7 @@ def _parse_json(raw: str) -> dict | list:
|
|||||||
return json.JSONDecoder(strict=False).decode(raw)
|
return json.JSONDecoder(strict=False).decode(raw)
|
||||||
except json.JSONDecodeError:
|
except json.JSONDecodeError:
|
||||||
pass
|
pass
|
||||||
# recover truncated array
|
# recover truncated array — extract complete objects one by one
|
||||||
if raw.lstrip().startswith("["):
|
if raw.lstrip().startswith("["):
|
||||||
decoder = json.JSONDecoder(strict=False)
|
decoder = json.JSONDecoder(strict=False)
|
||||||
items: list = []
|
items: list = []
|
||||||
@@ -71,8 +71,19 @@ def _parse_json(raw: str) -> dict | list:
|
|||||||
items.append(obj)
|
items.append(obj)
|
||||||
idx = end
|
idx = end
|
||||||
except json.JSONDecodeError:
|
except json.JSONDecodeError:
|
||||||
|
# Try to recover truncated last object by closing braces
|
||||||
|
remainder = raw[idx:]
|
||||||
|
for fix in ["}", "}]", '"}', '"}' , '"}]', "null}", "null}]"]:
|
||||||
|
try:
|
||||||
|
patched = remainder.rstrip().rstrip(",") + fix
|
||||||
|
obj = json.loads(patched)
|
||||||
|
if isinstance(obj, dict) and obj.get("name"):
|
||||||
|
items.append(obj)
|
||||||
|
except (json.JSONDecodeError, ValueError):
|
||||||
|
continue
|
||||||
break
|
break
|
||||||
if items:
|
if items:
|
||||||
|
logger.info("Recovered %d restaurants from truncated JSON", len(items))
|
||||||
return items
|
return items
|
||||||
raise ValueError(f"JSON parse failed: {raw[:80]!r}")
|
raise ValueError(f"JSON parse failed: {raw[:80]!r}")
|
||||||
|
|
||||||
@@ -104,7 +115,7 @@ _EXTRACT_PROMPT = """\
|
|||||||
JSON 배열:"""
|
JSON 배열:"""
|
||||||
|
|
||||||
|
|
||||||
def extract_restaurants(title: str, transcript: str) -> tuple[list[dict], str]:
|
def extract_restaurants(title: str, transcript: str, custom_prompt: str | None = None) -> tuple[list[dict], str]:
|
||||||
"""Extract restaurant info from a video transcript using LLM.
|
"""Extract restaurant info from a video transcript using LLM.
|
||||||
|
|
||||||
Returns (list of restaurant dicts, raw LLM response text).
|
Returns (list of restaurant dicts, raw LLM response text).
|
||||||
@@ -113,10 +124,11 @@ def extract_restaurants(title: str, transcript: str) -> tuple[list[dict], str]:
|
|||||||
if len(transcript) > 8000:
|
if len(transcript) > 8000:
|
||||||
transcript = transcript[:7000] + "\n...(중략)...\n" + transcript[-1000:]
|
transcript = transcript[:7000] + "\n...(중략)...\n" + transcript[-1000:]
|
||||||
|
|
||||||
prompt = _EXTRACT_PROMPT.format(title=title, transcript=transcript)
|
template = custom_prompt if custom_prompt else _EXTRACT_PROMPT
|
||||||
|
prompt = template.format(title=title, transcript=transcript)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
raw = _llm(prompt, max_tokens=4096)
|
raw = _llm(prompt, max_tokens=8192)
|
||||||
result = _parse_json(raw)
|
result = _parse_json(raw)
|
||||||
if isinstance(result, list):
|
if isinstance(result, list):
|
||||||
return result, raw
|
return result, raw
|
||||||
|
|||||||
@@ -57,17 +57,53 @@ def _places_text_search(query: str) -> dict | None:
|
|||||||
if data.get("status") == "OK" and data.get("results"):
|
if data.get("status") == "OK" and data.get("results"):
|
||||||
place = data["results"][0]
|
place = data["results"][0]
|
||||||
loc = place["geometry"]["location"]
|
loc = place["geometry"]["location"]
|
||||||
return {
|
result = {
|
||||||
"latitude": loc["lat"],
|
"latitude": loc["lat"],
|
||||||
"longitude": loc["lng"],
|
"longitude": loc["lng"],
|
||||||
"formatted_address": place.get("formatted_address", ""),
|
"formatted_address": place.get("formatted_address", ""),
|
||||||
"google_place_id": place.get("place_id", ""),
|
"google_place_id": place.get("place_id", ""),
|
||||||
|
"business_status": place.get("business_status"),
|
||||||
|
"rating": place.get("rating"),
|
||||||
|
"rating_count": place.get("user_ratings_total"),
|
||||||
}
|
}
|
||||||
|
# Fetch phone/website from Place Details
|
||||||
|
place_id = place.get("place_id")
|
||||||
|
if place_id:
|
||||||
|
details = _place_details(place_id)
|
||||||
|
if details:
|
||||||
|
result.update(details)
|
||||||
|
return result
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.warning("Places text search failed for '%s': %s", query, e)
|
logger.warning("Places text search failed for '%s': %s", query, e)
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def _place_details(place_id: str) -> dict | None:
|
||||||
|
"""Fetch phone and website from Google Place Details API."""
|
||||||
|
try:
|
||||||
|
r = httpx.get(
|
||||||
|
"https://maps.googleapis.com/maps/api/place/details/json",
|
||||||
|
params={
|
||||||
|
"place_id": place_id,
|
||||||
|
"key": _api_key(),
|
||||||
|
"language": "ko",
|
||||||
|
"fields": "formatted_phone_number,website",
|
||||||
|
},
|
||||||
|
timeout=10,
|
||||||
|
)
|
||||||
|
r.raise_for_status()
|
||||||
|
data = r.json()
|
||||||
|
if data.get("status") == "OK" and data.get("result"):
|
||||||
|
res = data["result"]
|
||||||
|
return {
|
||||||
|
"phone": res.get("formatted_phone_number"),
|
||||||
|
"website": res.get("website"),
|
||||||
|
}
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning("Place details failed for '%s': %s", place_id, e)
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
def _geocode(query: str) -> dict | None:
|
def _geocode(query: str) -> dict | None:
|
||||||
"""Geocode an address string."""
|
"""Geocode an address string."""
|
||||||
try:
|
try:
|
||||||
|
|||||||
@@ -28,7 +28,7 @@ def process_video(video: dict) -> int:
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
# 1. Transcript
|
# 1. Transcript
|
||||||
transcript = youtube.get_transcript(video_id)
|
transcript, _src = youtube.get_transcript(video_id)
|
||||||
if not transcript:
|
if not transcript:
|
||||||
logger.warning("No transcript for %s, marking done", video_id)
|
logger.warning("No transcript for %s, marking done", video_id)
|
||||||
youtube.update_video_status(video_db_id, "done")
|
youtube.update_video_status(video_db_id, "done")
|
||||||
@@ -72,6 +72,11 @@ def process_video(video: dict) -> int:
|
|||||||
cuisine_type=rest_data.get("cuisine_type"),
|
cuisine_type=rest_data.get("cuisine_type"),
|
||||||
price_range=rest_data.get("price_range"),
|
price_range=rest_data.get("price_range"),
|
||||||
google_place_id=place_id,
|
google_place_id=place_id,
|
||||||
|
phone=geo.get("phone") if geo else None,
|
||||||
|
website=geo.get("website") if geo else None,
|
||||||
|
business_status=geo.get("business_status") if geo else None,
|
||||||
|
rating=geo.get("rating") if geo else None,
|
||||||
|
rating_count=geo.get("rating_count") if geo else None,
|
||||||
)
|
)
|
||||||
|
|
||||||
# Link video <-> restaurant
|
# Link video <-> restaurant
|
||||||
@@ -101,6 +106,76 @@ def process_video(video: dict) -> int:
|
|||||||
return 0
|
return 0
|
||||||
|
|
||||||
|
|
||||||
|
def process_video_extract(video: dict, transcript: str, custom_prompt: str | None = None) -> int:
|
||||||
|
"""Run LLM extraction + geocode + save on an existing transcript.
|
||||||
|
Returns number of restaurants found."""
|
||||||
|
video_db_id = video["id"]
|
||||||
|
title = video["title"]
|
||||||
|
|
||||||
|
logger.info("Extracting restaurants from video: %s", title)
|
||||||
|
|
||||||
|
try:
|
||||||
|
restaurants, llm_raw = extractor.extract_restaurants(title, transcript, custom_prompt=custom_prompt)
|
||||||
|
if not restaurants:
|
||||||
|
youtube.update_video_status(video_db_id, "done", llm_raw=llm_raw)
|
||||||
|
return 0
|
||||||
|
|
||||||
|
count = 0
|
||||||
|
for rest_data in restaurants:
|
||||||
|
name = rest_data.get("name")
|
||||||
|
if not name:
|
||||||
|
continue
|
||||||
|
|
||||||
|
geo = geocoding.geocode_restaurant(
|
||||||
|
name,
|
||||||
|
address=rest_data.get("address"),
|
||||||
|
region=rest_data.get("region"),
|
||||||
|
)
|
||||||
|
|
||||||
|
lat = geo["latitude"] if geo else None
|
||||||
|
lng = geo["longitude"] if geo else None
|
||||||
|
addr = geo["formatted_address"] if geo else rest_data.get("address")
|
||||||
|
place_id = geo["google_place_id"] if geo else None
|
||||||
|
|
||||||
|
rest_id = restaurant.upsert(
|
||||||
|
name=name,
|
||||||
|
address=addr,
|
||||||
|
region=rest_data.get("region"),
|
||||||
|
latitude=lat,
|
||||||
|
longitude=lng,
|
||||||
|
cuisine_type=rest_data.get("cuisine_type"),
|
||||||
|
price_range=rest_data.get("price_range"),
|
||||||
|
google_place_id=place_id,
|
||||||
|
phone=geo.get("phone") if geo else None,
|
||||||
|
website=geo.get("website") if geo else None,
|
||||||
|
business_status=geo.get("business_status") if geo else None,
|
||||||
|
rating=geo.get("rating") if geo else None,
|
||||||
|
rating_count=geo.get("rating_count") if geo else None,
|
||||||
|
)
|
||||||
|
|
||||||
|
restaurant.link_video_restaurant(
|
||||||
|
video_db_id=video_db_id,
|
||||||
|
restaurant_id=rest_id,
|
||||||
|
foods=rest_data.get("foods_mentioned"),
|
||||||
|
evaluation=rest_data.get("evaluation"),
|
||||||
|
guests=rest_data.get("guests"),
|
||||||
|
)
|
||||||
|
|
||||||
|
chunks = _build_chunks(name, rest_data, title)
|
||||||
|
if chunks:
|
||||||
|
vector.save_restaurant_vectors(rest_id, chunks)
|
||||||
|
|
||||||
|
count += 1
|
||||||
|
logger.info("Saved restaurant: %s (geocoded=%s)", name, bool(geo))
|
||||||
|
|
||||||
|
youtube.update_video_status(video_db_id, "done", llm_raw=llm_raw)
|
||||||
|
return count
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("Extract error for %s: %s", video["video_id"], e, exc_info=True)
|
||||||
|
return 0
|
||||||
|
|
||||||
|
|
||||||
def _build_chunks(name: str, data: dict, video_title: str) -> list[str]:
|
def _build_chunks(name: str, data: dict, video_title: str) -> list[str]:
|
||||||
"""Build text chunks for vector embedding."""
|
"""Build text chunks for vector embedding."""
|
||||||
parts = [f"식당: {name}"]
|
parts = [f"식당: {name}"]
|
||||||
|
|||||||
@@ -9,6 +9,16 @@ import oracledb
|
|||||||
from core.db import conn
|
from core.db import conn
|
||||||
|
|
||||||
|
|
||||||
|
def _truncate_bytes(val: str | None, max_bytes: int) -> str | None:
|
||||||
|
"""Truncate a string to fit within max_bytes when encoded as UTF-8."""
|
||||||
|
if not val:
|
||||||
|
return val
|
||||||
|
encoded = val.encode("utf-8")
|
||||||
|
if len(encoded) <= max_bytes:
|
||||||
|
return val
|
||||||
|
return encoded[:max_bytes].decode("utf-8", errors="ignore").rstrip()
|
||||||
|
|
||||||
|
|
||||||
def find_by_name(name: str) -> dict | None:
|
def find_by_name(name: str) -> dict | None:
|
||||||
"""Find a restaurant by exact name match."""
|
"""Find a restaurant by exact name match."""
|
||||||
sql = "SELECT id, name, address, region, latitude, longitude FROM restaurants WHERE name = :n"
|
sql = "SELECT id, name, address, region, latitude, longitude FROM restaurants WHERE name = :n"
|
||||||
@@ -33,8 +43,19 @@ def upsert(
|
|||||||
cuisine_type: str | None = None,
|
cuisine_type: str | None = None,
|
||||||
price_range: str | None = None,
|
price_range: str | None = None,
|
||||||
google_place_id: str | None = None,
|
google_place_id: str | None = None,
|
||||||
|
phone: str | None = None,
|
||||||
|
website: str | None = None,
|
||||||
|
business_status: str | None = None,
|
||||||
|
rating: float | None = None,
|
||||||
|
rating_count: int | None = None,
|
||||||
) -> str:
|
) -> str:
|
||||||
"""Insert or update a restaurant. Returns row id."""
|
"""Insert or update a restaurant. Returns row id."""
|
||||||
|
# Truncate fields to fit DB column byte limits (VARCHAR2 is byte-based)
|
||||||
|
price_range = _truncate_bytes(price_range, 50)
|
||||||
|
cuisine_type = _truncate_bytes(cuisine_type, 100)
|
||||||
|
region = _truncate_bytes(region, 100)
|
||||||
|
website = _truncate_bytes(website, 500)
|
||||||
|
|
||||||
existing = find_by_name(name)
|
existing = find_by_name(name)
|
||||||
if existing:
|
if existing:
|
||||||
sql = """
|
sql = """
|
||||||
@@ -46,6 +67,11 @@ def upsert(
|
|||||||
cuisine_type = COALESCE(:cuisine, cuisine_type),
|
cuisine_type = COALESCE(:cuisine, cuisine_type),
|
||||||
price_range = COALESCE(:price, price_range),
|
price_range = COALESCE(:price, price_range),
|
||||||
google_place_id = COALESCE(:gid, google_place_id),
|
google_place_id = COALESCE(:gid, google_place_id),
|
||||||
|
phone = COALESCE(:phone, phone),
|
||||||
|
website = COALESCE(:web, website),
|
||||||
|
business_status = COALESCE(:bstatus, business_status),
|
||||||
|
rating = COALESCE(:rating, rating),
|
||||||
|
rating_count = COALESCE(:rcnt, rating_count),
|
||||||
updated_at = SYSTIMESTAMP
|
updated_at = SYSTIMESTAMP
|
||||||
WHERE id = :id
|
WHERE id = :id
|
||||||
"""
|
"""
|
||||||
@@ -54,14 +80,18 @@ def upsert(
|
|||||||
"addr": address, "reg": region,
|
"addr": address, "reg": region,
|
||||||
"lat": latitude, "lng": longitude,
|
"lat": latitude, "lng": longitude,
|
||||||
"cuisine": cuisine_type, "price": price_range,
|
"cuisine": cuisine_type, "price": price_range,
|
||||||
"gid": google_place_id, "id": existing["id"],
|
"gid": google_place_id, "phone": phone, "web": website,
|
||||||
|
"bstatus": business_status, "rating": rating, "rcnt": rating_count,
|
||||||
|
"id": existing["id"],
|
||||||
})
|
})
|
||||||
return existing["id"]
|
return existing["id"]
|
||||||
|
|
||||||
sql = """
|
sql = """
|
||||||
INSERT INTO restaurants (name, address, region, latitude, longitude,
|
INSERT INTO restaurants (name, address, region, latitude, longitude,
|
||||||
cuisine_type, price_range, google_place_id)
|
cuisine_type, price_range, google_place_id,
|
||||||
VALUES (:name, :addr, :reg, :lat, :lng, :cuisine, :price, :gid)
|
phone, website, business_status, rating, rating_count)
|
||||||
|
VALUES (:name, :addr, :reg, :lat, :lng, :cuisine, :price, :gid,
|
||||||
|
:phone, :web, :bstatus, :rating, :rcnt)
|
||||||
RETURNING id INTO :out_id
|
RETURNING id INTO :out_id
|
||||||
"""
|
"""
|
||||||
with conn() as c:
|
with conn() as c:
|
||||||
@@ -71,7 +101,9 @@ def upsert(
|
|||||||
"name": name, "addr": address, "reg": region,
|
"name": name, "addr": address, "reg": region,
|
||||||
"lat": latitude, "lng": longitude,
|
"lat": latitude, "lng": longitude,
|
||||||
"cuisine": cuisine_type, "price": price_range,
|
"cuisine": cuisine_type, "price": price_range,
|
||||||
"gid": google_place_id, "out_id": out_id,
|
"gid": google_place_id, "phone": phone, "web": website,
|
||||||
|
"bstatus": business_status, "rating": rating, "rcnt": rating_count,
|
||||||
|
"out_id": out_id,
|
||||||
})
|
})
|
||||||
return out_id.getvalue()[0]
|
return out_id.getvalue()[0]
|
||||||
|
|
||||||
@@ -116,38 +148,83 @@ def get_all(
|
|||||||
offset: int = 0,
|
offset: int = 0,
|
||||||
cuisine: str | None = None,
|
cuisine: str | None = None,
|
||||||
region: str | None = None,
|
region: str | None = None,
|
||||||
|
channel: str | None = None,
|
||||||
) -> list[dict]:
|
) -> list[dict]:
|
||||||
"""List restaurants with optional filters."""
|
"""List restaurants with optional filters."""
|
||||||
conditions = ["latitude IS NOT NULL"]
|
conditions = [
|
||||||
|
"r.latitude IS NOT NULL",
|
||||||
|
"EXISTS (SELECT 1 FROM video_restaurants vr0 WHERE vr0.restaurant_id = r.id)",
|
||||||
|
]
|
||||||
params: dict = {"lim": limit, "off": offset}
|
params: dict = {"lim": limit, "off": offset}
|
||||||
|
|
||||||
if cuisine:
|
if cuisine:
|
||||||
conditions.append("cuisine_type = :cuisine")
|
conditions.append("r.cuisine_type = :cuisine")
|
||||||
params["cuisine"] = cuisine
|
params["cuisine"] = cuisine
|
||||||
if region:
|
if region:
|
||||||
conditions.append("region LIKE :region")
|
conditions.append("r.region LIKE :region")
|
||||||
params["region"] = f"%{region}%"
|
params["region"] = f"%{region}%"
|
||||||
|
|
||||||
|
join_clause = ""
|
||||||
|
if channel:
|
||||||
|
join_clause = """
|
||||||
|
JOIN video_restaurants vr_f ON vr_f.restaurant_id = r.id
|
||||||
|
JOIN videos v_f ON v_f.id = vr_f.video_id
|
||||||
|
JOIN channels c_f ON c_f.id = v_f.channel_id
|
||||||
|
"""
|
||||||
|
conditions.append("c_f.channel_name = :channel")
|
||||||
|
params["channel"] = channel
|
||||||
|
|
||||||
where = " AND ".join(conditions)
|
where = " AND ".join(conditions)
|
||||||
sql = f"""
|
sql = f"""
|
||||||
SELECT id, name, address, region, latitude, longitude,
|
SELECT DISTINCT r.id, r.name, r.address, r.region, r.latitude, r.longitude,
|
||||||
cuisine_type, price_range, google_place_id
|
r.cuisine_type, r.price_range, r.google_place_id,
|
||||||
FROM restaurants
|
r.business_status, r.rating, r.rating_count, r.updated_at
|
||||||
|
FROM restaurants r
|
||||||
|
{join_clause}
|
||||||
WHERE {where}
|
WHERE {where}
|
||||||
ORDER BY updated_at DESC
|
ORDER BY r.updated_at DESC
|
||||||
OFFSET :off ROWS FETCH NEXT :lim ROWS ONLY
|
OFFSET :off ROWS FETCH NEXT :lim ROWS ONLY
|
||||||
"""
|
"""
|
||||||
with conn() as c:
|
with conn() as c:
|
||||||
cur = c.cursor()
|
cur = c.cursor()
|
||||||
cur.execute(sql, params)
|
cur.execute(sql, params)
|
||||||
cols = [d[0].lower() for d in cur.description]
|
cols = [d[0].lower() for d in cur.description]
|
||||||
return [dict(zip(cols, row)) for row in cur.fetchall()]
|
restaurants = [dict(zip(cols, row)) for row in cur.fetchall()]
|
||||||
|
for r in restaurants:
|
||||||
|
r.pop("updated_at", None)
|
||||||
|
|
||||||
|
if not restaurants:
|
||||||
|
return restaurants
|
||||||
|
|
||||||
|
# Attach channel names for each restaurant
|
||||||
|
ids = [r["id"] for r in restaurants]
|
||||||
|
placeholders = ", ".join(f":id{i}" for i in range(len(ids)))
|
||||||
|
ch_sql = f"""
|
||||||
|
SELECT DISTINCT vr.restaurant_id, c.channel_name
|
||||||
|
FROM video_restaurants vr
|
||||||
|
JOIN videos v ON v.id = vr.video_id
|
||||||
|
JOIN channels c ON c.id = v.channel_id
|
||||||
|
WHERE vr.restaurant_id IN ({placeholders})
|
||||||
|
"""
|
||||||
|
ch_params = {f"id{i}": rid for i, rid in enumerate(ids)}
|
||||||
|
ch_map: dict[str, list[str]] = {}
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(ch_sql, ch_params)
|
||||||
|
for row in cur.fetchall():
|
||||||
|
ch_map.setdefault(row[0], []).append(row[1])
|
||||||
|
|
||||||
|
for r in restaurants:
|
||||||
|
r["channels"] = ch_map.get(r["id"], [])
|
||||||
|
|
||||||
|
return restaurants
|
||||||
|
|
||||||
|
|
||||||
def get_by_id(restaurant_id: str) -> dict | None:
|
def get_by_id(restaurant_id: str) -> dict | None:
|
||||||
sql = """
|
sql = """
|
||||||
SELECT r.id, r.name, r.address, r.region, r.latitude, r.longitude,
|
SELECT r.id, r.name, r.address, r.region, r.latitude, r.longitude,
|
||||||
r.cuisine_type, r.price_range, r.phone, r.website, r.google_place_id
|
r.cuisine_type, r.price_range, r.phone, r.website, r.google_place_id,
|
||||||
|
r.business_status, r.rating, r.rating_count
|
||||||
FROM restaurants r
|
FROM restaurants r
|
||||||
WHERE r.id = :id
|
WHERE r.id = :id
|
||||||
"""
|
"""
|
||||||
@@ -165,9 +242,11 @@ def get_video_links(restaurant_id: str) -> list[dict]:
|
|||||||
"""Get all video appearances for a restaurant."""
|
"""Get all video appearances for a restaurant."""
|
||||||
sql = """
|
sql = """
|
||||||
SELECT v.video_id, v.title, v.url, v.published_at,
|
SELECT v.video_id, v.title, v.url, v.published_at,
|
||||||
vr.foods_mentioned, vr.evaluation, vr.guests
|
vr.foods_mentioned, vr.evaluation, vr.guests,
|
||||||
|
c.channel_name, c.channel_id
|
||||||
FROM video_restaurants vr
|
FROM video_restaurants vr
|
||||||
JOIN videos v ON v.id = vr.video_id
|
JOIN videos v ON v.id = vr.video_id
|
||||||
|
JOIN channels c ON c.id = v.channel_id
|
||||||
WHERE vr.restaurant_id = :rid
|
WHERE vr.restaurant_id = :rid
|
||||||
ORDER BY v.published_at DESC
|
ORDER BY v.published_at DESC
|
||||||
"""
|
"""
|
||||||
@@ -187,6 +266,8 @@ def get_video_links(restaurant_id: str) -> list[dict]:
|
|||||||
"foods_mentioned": _parse_json_field(foods_raw, []),
|
"foods_mentioned": _parse_json_field(foods_raw, []),
|
||||||
"evaluation": _parse_json_field(eval_raw, {}),
|
"evaluation": _parse_json_field(eval_raw, {}),
|
||||||
"guests": _parse_json_field(guests_raw, []),
|
"guests": _parse_json_field(guests_raw, []),
|
||||||
|
"channel_name": r[7],
|
||||||
|
"channel_id": r[8],
|
||||||
})
|
})
|
||||||
return results
|
return results
|
||||||
|
|
||||||
|
|||||||
@@ -131,13 +131,15 @@ def get_user_reviews(
|
|||||||
limit: int = 20,
|
limit: int = 20,
|
||||||
offset: int = 0,
|
offset: int = 0,
|
||||||
) -> list[dict]:
|
) -> list[dict]:
|
||||||
"""List reviews by a specific user."""
|
"""List reviews by a specific user, including restaurant name."""
|
||||||
sql = """
|
sql = """
|
||||||
SELECT r.id, r.user_id, r.restaurant_id, r.rating, r.review_text,
|
SELECT r.id, r.user_id, r.restaurant_id, r.rating, r.review_text,
|
||||||
r.visited_at, r.created_at, r.updated_at,
|
r.visited_at, r.created_at, r.updated_at,
|
||||||
u.nickname, u.avatar_url
|
u.nickname, u.avatar_url,
|
||||||
|
rest.name AS restaurant_name
|
||||||
FROM user_reviews r
|
FROM user_reviews r
|
||||||
JOIN tasteby_users u ON u.id = r.user_id
|
JOIN tasteby_users u ON u.id = r.user_id
|
||||||
|
LEFT JOIN restaurants rest ON rest.id = r.restaurant_id
|
||||||
WHERE r.user_id = :user_id
|
WHERE r.user_id = :user_id
|
||||||
ORDER BY r.created_at DESC
|
ORDER BY r.created_at DESC
|
||||||
OFFSET :off ROWS FETCH NEXT :lim ROWS ONLY
|
OFFSET :off ROWS FETCH NEXT :lim ROWS ONLY
|
||||||
@@ -149,7 +151,12 @@ def get_user_reviews(
|
|||||||
"off": offset,
|
"off": offset,
|
||||||
"lim": limit,
|
"lim": limit,
|
||||||
})
|
})
|
||||||
return [_row_to_dict(row) for row in cur.fetchall()]
|
rows = []
|
||||||
|
for row in cur.fetchall():
|
||||||
|
d = _row_to_dict(row)
|
||||||
|
d["restaurant_name"] = row[10]
|
||||||
|
rows.append(d)
|
||||||
|
return rows
|
||||||
|
|
||||||
|
|
||||||
def get_restaurant_avg_rating(restaurant_id: str) -> dict:
|
def get_restaurant_avg_rating(restaurant_id: str) -> dict:
|
||||||
|
|||||||
@@ -32,11 +32,11 @@ def extract_video_id(url: str) -> str:
|
|||||||
|
|
||||||
# -- Channel operations -------------------------------------------------------
|
# -- Channel operations -------------------------------------------------------
|
||||||
|
|
||||||
def add_channel(channel_id: str, channel_name: str) -> str:
|
def add_channel(channel_id: str, channel_name: str, title_filter: str | None = None) -> str:
|
||||||
"""Register a YouTube channel. Returns DB row id."""
|
"""Register a YouTube channel. Returns DB row id."""
|
||||||
sql = """
|
sql = """
|
||||||
INSERT INTO channels (channel_id, channel_name, channel_url)
|
INSERT INTO channels (channel_id, channel_name, channel_url, title_filter)
|
||||||
VALUES (:cid, :cname, :curl)
|
VALUES (:cid, :cname, :curl, :tf)
|
||||||
RETURNING id INTO :out_id
|
RETURNING id INTO :out_id
|
||||||
"""
|
"""
|
||||||
with conn() as c:
|
with conn() as c:
|
||||||
@@ -47,45 +47,77 @@ def add_channel(channel_id: str, channel_name: str) -> str:
|
|||||||
"cid": channel_id,
|
"cid": channel_id,
|
||||||
"cname": channel_name,
|
"cname": channel_name,
|
||||||
"curl": f"https://www.youtube.com/channel/{channel_id}",
|
"curl": f"https://www.youtube.com/channel/{channel_id}",
|
||||||
|
"tf": title_filter,
|
||||||
"out_id": out_id,
|
"out_id": out_id,
|
||||||
})
|
})
|
||||||
return out_id.getvalue()[0]
|
return out_id.getvalue()[0]
|
||||||
|
|
||||||
|
|
||||||
|
def deactivate_channel(channel_id: str) -> bool:
|
||||||
|
"""Deactivate a channel by channel_id. Returns True if found."""
|
||||||
|
sql = "UPDATE channels SET is_active = 0 WHERE channel_id = :cid AND is_active = 1"
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(sql, {"cid": channel_id})
|
||||||
|
return cur.rowcount > 0
|
||||||
|
|
||||||
|
|
||||||
|
def deactivate_channel_by_db_id(db_id: str) -> bool:
|
||||||
|
"""Deactivate a channel by DB id. Returns True if found."""
|
||||||
|
sql = "UPDATE channels SET is_active = 0 WHERE id = :did AND is_active = 1"
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(sql, {"did": db_id})
|
||||||
|
return cur.rowcount > 0
|
||||||
|
|
||||||
|
|
||||||
def get_active_channels() -> list[dict]:
|
def get_active_channels() -> list[dict]:
|
||||||
sql = "SELECT id, channel_id, channel_name FROM channels WHERE is_active = 1"
|
sql = "SELECT id, channel_id, channel_name, title_filter FROM channels WHERE is_active = 1"
|
||||||
with conn() as c:
|
with conn() as c:
|
||||||
cur = c.cursor()
|
cur = c.cursor()
|
||||||
cur.execute(sql)
|
cur.execute(sql)
|
||||||
return [
|
return [
|
||||||
{"id": r[0], "channel_id": r[1], "channel_name": r[2]}
|
{"id": r[0], "channel_id": r[1], "channel_name": r[2], "title_filter": r[3]}
|
||||||
for r in cur.fetchall()
|
for r in cur.fetchall()
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
# -- Video listing via YouTube Data API v3 ------------------------------------
|
# -- Video listing via YouTube Data API v3 ------------------------------------
|
||||||
|
|
||||||
def fetch_channel_videos(
|
def get_latest_video_date(channel_db_id: str) -> str | None:
|
||||||
channel_id: str,
|
"""Get the latest published_at for a channel's videos in ISO 8601 format."""
|
||||||
max_results: int = 50,
|
sql = """
|
||||||
published_after: str | None = None,
|
SELECT MAX(published_at) FROM videos
|
||||||
) -> list[dict]:
|
WHERE channel_id = :ch_id AND published_at IS NOT NULL
|
||||||
"""Fetch video list from a YouTube channel via Data API v3.
|
"""
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute(sql, {"ch_id": channel_db_id})
|
||||||
|
row = cur.fetchone()
|
||||||
|
if row and row[0]:
|
||||||
|
return row[0].strftime("%Y-%m-%dT%H:%M:%SZ")
|
||||||
|
return None
|
||||||
|
|
||||||
Returns list of dicts: video_id, title, published_at, url.
|
|
||||||
|
def fetch_channel_videos_iter(
|
||||||
|
channel_id: str,
|
||||||
|
published_after: str | None = None,
|
||||||
|
):
|
||||||
|
"""Yield pages of videos from a YouTube channel via Data API v3.
|
||||||
|
|
||||||
|
Each yield is a list of dicts for one API page (up to 50).
|
||||||
"""
|
"""
|
||||||
params: dict = {
|
params: dict = {
|
||||||
"key": _api_key(),
|
"key": _api_key(),
|
||||||
"channelId": channel_id,
|
"channelId": channel_id,
|
||||||
"part": "snippet",
|
"part": "snippet",
|
||||||
"order": "date",
|
"order": "date",
|
||||||
"maxResults": min(max_results, 50),
|
"maxResults": 50,
|
||||||
"type": "video",
|
"type": "video",
|
||||||
}
|
}
|
||||||
if published_after:
|
if published_after:
|
||||||
params["publishedAfter"] = published_after
|
params["publishedAfter"] = published_after
|
||||||
|
|
||||||
videos: list[dict] = []
|
|
||||||
next_page = None
|
next_page = None
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
@@ -100,33 +132,337 @@ def fetch_channel_videos(
|
|||||||
r.raise_for_status()
|
r.raise_for_status()
|
||||||
data = r.json()
|
data = r.json()
|
||||||
|
|
||||||
|
page_videos = []
|
||||||
for item in data.get("items", []):
|
for item in data.get("items", []):
|
||||||
snippet = item["snippet"]
|
snippet = item["snippet"]
|
||||||
vid = item["id"]["videoId"]
|
vid = item["id"]["videoId"]
|
||||||
videos.append({
|
page_videos.append({
|
||||||
"video_id": vid,
|
"video_id": vid,
|
||||||
"title": snippet["title"],
|
"title": snippet["title"],
|
||||||
"published_at": snippet["publishedAt"],
|
"published_at": snippet["publishedAt"],
|
||||||
"url": f"https://www.youtube.com/watch?v={vid}",
|
"url": f"https://www.youtube.com/watch?v={vid}",
|
||||||
})
|
})
|
||||||
|
|
||||||
|
if page_videos:
|
||||||
|
yield page_videos
|
||||||
|
|
||||||
next_page = data.get("nextPageToken")
|
next_page = data.get("nextPageToken")
|
||||||
if not next_page or len(videos) >= max_results:
|
if not next_page:
|
||||||
break
|
break
|
||||||
|
|
||||||
return videos[:max_results]
|
|
||||||
|
def fetch_channel_videos(
|
||||||
|
channel_id: str,
|
||||||
|
max_results: int = 0,
|
||||||
|
published_after: str | None = None,
|
||||||
|
) -> list[dict]:
|
||||||
|
"""Fetch video list from a YouTube channel via Data API v3.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
max_results: 0 means fetch all available videos.
|
||||||
|
"""
|
||||||
|
videos: list[dict] = []
|
||||||
|
for page in fetch_channel_videos_iter(channel_id, published_after=published_after):
|
||||||
|
videos.extend(page)
|
||||||
|
if max_results > 0 and len(videos) >= max_results:
|
||||||
|
break
|
||||||
|
return videos[:max_results] if max_results > 0 else videos
|
||||||
|
|
||||||
|
|
||||||
# -- Transcript extraction ----------------------------------------------------
|
# -- Transcript extraction ----------------------------------------------------
|
||||||
|
|
||||||
def get_transcript(video_id: str) -> str | None:
|
def get_transcript(video_id: str, mode: str = "auto") -> tuple[str | None, str | None]:
|
||||||
"""Fetch transcript text for a video. Returns None if unavailable."""
|
"""Fetch transcript using Playwright (headless browser).
|
||||||
|
|
||||||
|
Args:
|
||||||
|
mode: "manual" = manual only, "generated" = auto-generated only,
|
||||||
|
"auto" = try API first, fallback to browser transcript panel.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
(transcript_text, source) where source describes origin, or (None, None).
|
||||||
|
"""
|
||||||
|
# Try youtube-transcript-api first (fast path)
|
||||||
|
text, source = _get_transcript_api(video_id, mode)
|
||||||
|
if text:
|
||||||
|
return text, source
|
||||||
|
|
||||||
|
# Fallback: Playwright browser
|
||||||
|
logger.warning("API failed for %s, trying Playwright browser", video_id)
|
||||||
|
print(f"[TRANSCRIPT] API failed for {video_id}, trying Playwright browser", flush=True)
|
||||||
|
return _get_transcript_browser(video_id)
|
||||||
|
|
||||||
|
|
||||||
|
def _make_ytt() -> YouTubeTranscriptApi:
|
||||||
|
"""Create YouTubeTranscriptApi with cookies if available."""
|
||||||
|
cookie_file = os.path.join(os.path.dirname(__file__), "..", "cookies.txt")
|
||||||
|
if os.path.exists(cookie_file):
|
||||||
|
import http.cookiejar
|
||||||
|
import requests
|
||||||
|
jar = http.cookiejar.MozillaCookieJar(cookie_file)
|
||||||
|
jar.load(ignore_discard=True, ignore_expires=True)
|
||||||
|
session = requests.Session()
|
||||||
|
session.cookies = jar
|
||||||
|
return YouTubeTranscriptApi(http_client=session)
|
||||||
|
return YouTubeTranscriptApi()
|
||||||
|
|
||||||
|
|
||||||
|
def _get_transcript_api(video_id: str, mode: str = "auto") -> tuple[str | None, str | None]:
|
||||||
|
"""Try youtube-transcript-api (fast but may be IP-blocked)."""
|
||||||
|
ytt = _make_ytt()
|
||||||
|
prefer = ["ko", "en"]
|
||||||
|
|
||||||
try:
|
try:
|
||||||
fetched = YouTubeTranscriptApi().fetch(video_id, languages=["ko", "en"])
|
transcript_list = ytt.list(video_id)
|
||||||
return " ".join(seg.text for seg in fetched)
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.warning("Transcript unavailable for %s: %s", video_id, e)
|
logger.warning("Cannot list transcripts for %s: %s", video_id, e)
|
||||||
return None
|
return None, None
|
||||||
|
|
||||||
|
all_transcripts = list(transcript_list)
|
||||||
|
manual = [t for t in all_transcripts if not t.is_generated]
|
||||||
|
generated = [t for t in all_transcripts if t.is_generated]
|
||||||
|
|
||||||
|
def _pick(candidates):
|
||||||
|
for lang in prefer:
|
||||||
|
for t in candidates:
|
||||||
|
if t.language_code == lang:
|
||||||
|
return t
|
||||||
|
return candidates[0] if candidates else None
|
||||||
|
|
||||||
|
def _fetch(t):
|
||||||
|
try:
|
||||||
|
return " ".join(seg.text for seg in t.fetch()), t.language_code
|
||||||
|
except Exception:
|
||||||
|
return None, None
|
||||||
|
|
||||||
|
if mode == "manual":
|
||||||
|
t = _pick(manual)
|
||||||
|
if t:
|
||||||
|
text, lang = _fetch(t)
|
||||||
|
return (text, f"manual ({lang})") if text else (None, None)
|
||||||
|
return None, None
|
||||||
|
elif mode == "generated":
|
||||||
|
t = _pick(generated)
|
||||||
|
if t:
|
||||||
|
text, lang = _fetch(t)
|
||||||
|
return (text, f"generated ({lang})") if text else (None, None)
|
||||||
|
return None, None
|
||||||
|
else:
|
||||||
|
t = _pick(manual)
|
||||||
|
if t:
|
||||||
|
text, lang = _fetch(t)
|
||||||
|
if text:
|
||||||
|
return text, f"manual ({lang})"
|
||||||
|
t = _pick(generated)
|
||||||
|
if t:
|
||||||
|
text, lang = _fetch(t)
|
||||||
|
if text:
|
||||||
|
return text, f"generated ({lang})"
|
||||||
|
return None, None
|
||||||
|
|
||||||
|
|
||||||
|
def _get_transcript_browser(video_id: str) -> tuple[str | None, str | None]:
|
||||||
|
"""Fetch transcript via Playwright browser (bypasses IP blocks)."""
|
||||||
|
try:
|
||||||
|
from playwright.sync_api import sync_playwright
|
||||||
|
except ImportError:
|
||||||
|
logger.error("playwright not installed")
|
||||||
|
return None, None
|
||||||
|
|
||||||
|
try:
|
||||||
|
with sync_playwright() as p:
|
||||||
|
browser = p.chromium.launch(
|
||||||
|
headless=False,
|
||||||
|
args=["--disable-blink-features=AutomationControlled"],
|
||||||
|
)
|
||||||
|
ctx = browser.new_context(locale="ko-KR", viewport={"width": 1280, "height": 900})
|
||||||
|
|
||||||
|
# Load YouTube cookies if available
|
||||||
|
cookie_file = os.path.join(os.path.dirname(__file__), "..", "cookies.txt")
|
||||||
|
if os.path.exists(cookie_file):
|
||||||
|
import http.cookiejar
|
||||||
|
jar = http.cookiejar.MozillaCookieJar(cookie_file)
|
||||||
|
jar.load(ignore_discard=True, ignore_expires=True)
|
||||||
|
pw_cookies = []
|
||||||
|
for c in jar:
|
||||||
|
if "youtube" in c.domain or "google" in c.domain:
|
||||||
|
pw_cookies.append({
|
||||||
|
"name": c.name, "value": c.value,
|
||||||
|
"domain": c.domain, "path": c.path,
|
||||||
|
"secure": c.secure, "httpOnly": False,
|
||||||
|
})
|
||||||
|
if pw_cookies:
|
||||||
|
ctx.add_cookies(pw_cookies)
|
||||||
|
print(f"[TRANSCRIPT] Loaded {len(pw_cookies)} cookies", flush=True)
|
||||||
|
|
||||||
|
page = ctx.new_page()
|
||||||
|
page.add_init_script("Object.defineProperty(navigator, 'webdriver', {get: () => false})")
|
||||||
|
|
||||||
|
print(f"[TRANSCRIPT] Opening YouTube page for {video_id}", flush=True)
|
||||||
|
page.goto(
|
||||||
|
f"https://www.youtube.com/watch?v={video_id}",
|
||||||
|
wait_until="domcontentloaded",
|
||||||
|
timeout=30000,
|
||||||
|
)
|
||||||
|
page.wait_for_timeout(5000)
|
||||||
|
|
||||||
|
# Skip ads if present
|
||||||
|
for ad_wait in range(12): # up to ~60s for ads
|
||||||
|
ad_status = page.evaluate("""() => {
|
||||||
|
const skipBtn = document.querySelector('.ytp-skip-ad-button, .ytp-ad-skip-button, .ytp-ad-skip-button-modern, button.ytp-ad-skip-button-modern');
|
||||||
|
if (skipBtn) { skipBtn.click(); return 'skipped'; }
|
||||||
|
const adOverlay = document.querySelector('.ytp-ad-player-overlay, .ad-showing');
|
||||||
|
if (adOverlay) return 'playing';
|
||||||
|
const adBadge = document.querySelector('.ytp-ad-text');
|
||||||
|
if (adBadge && adBadge.textContent) return 'badge';
|
||||||
|
return 'none';
|
||||||
|
}""")
|
||||||
|
if ad_status == "none":
|
||||||
|
break
|
||||||
|
print(f"[TRANSCRIPT] Ad detected: {ad_status}, waiting...", flush=True)
|
||||||
|
if ad_status == "skipped":
|
||||||
|
page.wait_for_timeout(2000)
|
||||||
|
break
|
||||||
|
page.wait_for_timeout(5000)
|
||||||
|
|
||||||
|
page.wait_for_timeout(2000)
|
||||||
|
print(f"[TRANSCRIPT] Page loaded, looking for transcript button", flush=True)
|
||||||
|
|
||||||
|
# Click "더보기" (more actions) button first to reveal transcript option
|
||||||
|
page.evaluate("""
|
||||||
|
() => {
|
||||||
|
// Try clicking the "...더보기" button in description area
|
||||||
|
const moreBtn = document.querySelector('tp-yt-paper-button#expand');
|
||||||
|
if (moreBtn) moreBtn.click();
|
||||||
|
}
|
||||||
|
""")
|
||||||
|
page.wait_for_timeout(2000)
|
||||||
|
|
||||||
|
# Click "스크립트 표시" button via JS
|
||||||
|
clicked = page.evaluate("""
|
||||||
|
() => {
|
||||||
|
// Method 1: aria-label
|
||||||
|
for (const label of ['스크립트 표시', 'Show transcript']) {
|
||||||
|
const btns = document.querySelectorAll(`button[aria-label="${label}"]`);
|
||||||
|
for (const b of btns) { b.click(); return 'aria-label: ' + label; }
|
||||||
|
}
|
||||||
|
// Method 2: search all buttons by text content
|
||||||
|
const allBtns = document.querySelectorAll('button');
|
||||||
|
for (const b of allBtns) {
|
||||||
|
const text = b.textContent.trim();
|
||||||
|
if (text === '스크립트 표시' || text === 'Show transcript') {
|
||||||
|
b.click();
|
||||||
|
return 'text: ' + text;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Method 3: look for transcript button in engagement panel
|
||||||
|
const engBtns = document.querySelectorAll('ytd-button-renderer button, ytd-button-renderer a');
|
||||||
|
for (const b of engBtns) {
|
||||||
|
const text = b.textContent.trim().toLowerCase();
|
||||||
|
if (text.includes('transcript') || text.includes('스크립트')) {
|
||||||
|
b.click();
|
||||||
|
return 'engagement: ' + text;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
""")
|
||||||
|
print(f"[TRANSCRIPT] Clicked transcript button: {clicked}", flush=True)
|
||||||
|
if not clicked:
|
||||||
|
# Dump available buttons for debugging
|
||||||
|
btn_labels = page.evaluate("""
|
||||||
|
() => {
|
||||||
|
const btns = document.querySelectorAll('button[aria-label]');
|
||||||
|
return Array.from(btns).map(b => b.getAttribute('aria-label')).slice(0, 30);
|
||||||
|
}
|
||||||
|
""")
|
||||||
|
print(f"[TRANSCRIPT] Available buttons: {btn_labels}", flush=True)
|
||||||
|
browser.close()
|
||||||
|
return None, None
|
||||||
|
|
||||||
|
# Wait for transcript panel segments to appear (max ~40s)
|
||||||
|
page.wait_for_timeout(3000) # initial wait for panel to render
|
||||||
|
for attempt in range(12):
|
||||||
|
page.wait_for_timeout(3000)
|
||||||
|
count = page.evaluate(
|
||||||
|
"() => document.querySelectorAll('ytd-transcript-segment-renderer').length"
|
||||||
|
)
|
||||||
|
print(f"[TRANSCRIPT] Wait {(attempt+1)*3+3}s: {count} segments", flush=True)
|
||||||
|
if count > 0:
|
||||||
|
break
|
||||||
|
|
||||||
|
# Select Korean if available (language selector in transcript panel)
|
||||||
|
page.evaluate("""
|
||||||
|
() => {
|
||||||
|
// Open language dropdown and pick Korean if available
|
||||||
|
const menu = document.querySelector('ytd-transcript-renderer ytd-menu-renderer yt-dropdown-menu');
|
||||||
|
if (!menu) return;
|
||||||
|
const trigger = menu.querySelector('button, tp-yt-paper-button');
|
||||||
|
if (trigger) trigger.click();
|
||||||
|
}
|
||||||
|
""")
|
||||||
|
page.wait_for_timeout(1000)
|
||||||
|
page.evaluate("""
|
||||||
|
() => {
|
||||||
|
const items = document.querySelectorAll('tp-yt-paper-listbox a, tp-yt-paper-listbox tp-yt-paper-item');
|
||||||
|
for (const item of items) {
|
||||||
|
const text = item.textContent.trim();
|
||||||
|
if (text.includes('한국어') || text.includes('Korean')) {
|
||||||
|
item.click();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
""")
|
||||||
|
page.wait_for_timeout(2000)
|
||||||
|
|
||||||
|
# Scroll transcript panel to load all segments
|
||||||
|
segments = page.evaluate("""
|
||||||
|
async () => {
|
||||||
|
const container = document.querySelector(
|
||||||
|
'ytd-transcript-segment-list-renderer #segments-container, ' +
|
||||||
|
'ytd-transcript-renderer #body'
|
||||||
|
);
|
||||||
|
if (!container) {
|
||||||
|
// Fallback: just grab what's there
|
||||||
|
const segs = document.querySelectorAll('ytd-transcript-segment-renderer');
|
||||||
|
return Array.from(segs).map(s => {
|
||||||
|
const txt = s.querySelector('.segment-text, yt-formatted-string.segment-text');
|
||||||
|
return txt ? txt.textContent.trim() : '';
|
||||||
|
}).filter(t => t);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Scroll to bottom repeatedly to load all virtual segments
|
||||||
|
let prevCount = 0;
|
||||||
|
for (let i = 0; i < 50; i++) {
|
||||||
|
container.scrollTop = container.scrollHeight;
|
||||||
|
await new Promise(r => setTimeout(r, 300));
|
||||||
|
const segs = document.querySelectorAll('ytd-transcript-segment-renderer');
|
||||||
|
if (segs.length === prevCount && i > 3) break;
|
||||||
|
prevCount = segs.length;
|
||||||
|
}
|
||||||
|
|
||||||
|
const segs = document.querySelectorAll('ytd-transcript-segment-renderer');
|
||||||
|
return Array.from(segs).map(s => {
|
||||||
|
const txt = s.querySelector('.segment-text, yt-formatted-string.segment-text');
|
||||||
|
return txt ? txt.textContent.trim() : '';
|
||||||
|
}).filter(t => t);
|
||||||
|
}
|
||||||
|
""")
|
||||||
|
|
||||||
|
browser.close()
|
||||||
|
|
||||||
|
print(f"[TRANSCRIPT] Got {len(segments) if segments else 0} segments for {video_id}", flush=True)
|
||||||
|
if segments:
|
||||||
|
text = " ".join(segments)
|
||||||
|
print(f"[TRANSCRIPT] Success: {len(text)} chars from {len(segments)} segments", flush=True)
|
||||||
|
return text, "browser"
|
||||||
|
return None, None
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("Playwright transcript failed for %s: %s", video_id, e)
|
||||||
|
print(f"[TRANSCRIPT] Playwright FAILED for {video_id}: {e}", flush=True)
|
||||||
|
import traceback
|
||||||
|
traceback.print_exc()
|
||||||
|
return None, None
|
||||||
|
|
||||||
|
|
||||||
# -- DB operations for videos -------------------------------------------------
|
# -- DB operations for videos -------------------------------------------------
|
||||||
@@ -163,6 +499,48 @@ def save_video(channel_db_id: str, video: dict) -> str | None:
|
|||||||
raise
|
raise
|
||||||
|
|
||||||
|
|
||||||
|
def get_existing_video_ids(channel_db_id: str) -> set[str]:
|
||||||
|
"""Get all video_ids already in DB for a channel."""
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
cur.execute("SELECT video_id FROM videos WHERE channel_id = :cid", {"cid": channel_db_id})
|
||||||
|
return {r[0] for r in cur.fetchall()}
|
||||||
|
|
||||||
|
|
||||||
|
def save_videos_batch(channel_db_id: str, videos: list[dict]) -> int:
|
||||||
|
"""Insert multiple videos in a single DB connection. Returns count of new videos."""
|
||||||
|
if not videos:
|
||||||
|
return 0
|
||||||
|
import oracledb
|
||||||
|
sql = """
|
||||||
|
INSERT INTO videos (channel_id, video_id, title, url, published_at, status)
|
||||||
|
VALUES (:ch_id, :vid, :title, :url, :pub_at, 'pending')
|
||||||
|
"""
|
||||||
|
new_count = 0
|
||||||
|
with conn() as c:
|
||||||
|
cur = c.cursor()
|
||||||
|
for video in videos:
|
||||||
|
try:
|
||||||
|
pub_at = None
|
||||||
|
if video.get("published_at"):
|
||||||
|
pub_at = datetime.fromisoformat(
|
||||||
|
video["published_at"].replace("Z", "+00:00")
|
||||||
|
)
|
||||||
|
cur.execute(sql, {
|
||||||
|
"ch_id": channel_db_id,
|
||||||
|
"vid": video["video_id"],
|
||||||
|
"title": video["title"],
|
||||||
|
"url": video["url"],
|
||||||
|
"pub_at": pub_at,
|
||||||
|
})
|
||||||
|
new_count += 1
|
||||||
|
except Exception as e:
|
||||||
|
if "UQ_VIDEOS_VID" in str(e).upper():
|
||||||
|
continue
|
||||||
|
raise
|
||||||
|
return new_count
|
||||||
|
|
||||||
|
|
||||||
def get_pending_videos(limit: int = 10) -> list[dict]:
|
def get_pending_videos(limit: int = 10) -> list[dict]:
|
||||||
sql = """
|
sql = """
|
||||||
SELECT id, video_id, title, url
|
SELECT id, video_id, title, url
|
||||||
@@ -201,20 +579,28 @@ def update_video_status(
|
|||||||
|
|
||||||
# -- Scan: fetch new videos for all active channels ---------------------------
|
# -- Scan: fetch new videos for all active channels ---------------------------
|
||||||
|
|
||||||
def scan_all_channels(max_per_channel: int = 50) -> int:
|
def scan_all_channels() -> int:
|
||||||
"""Scan all active channels for new videos. Returns count of new videos."""
|
"""Scan all active channels for new videos. Returns count of new videos."""
|
||||||
channels = get_active_channels()
|
channels = get_active_channels()
|
||||||
total_new = 0
|
total_new = 0
|
||||||
for ch in channels:
|
for ch in channels:
|
||||||
try:
|
try:
|
||||||
videos = fetch_channel_videos(ch["channel_id"], max_per_channel)
|
after = get_latest_video_date(ch["id"])
|
||||||
for v in videos:
|
title_filter = ch.get("title_filter")
|
||||||
row_id = save_video(ch["id"], v)
|
new_count = 0
|
||||||
if row_id:
|
fetched = 0
|
||||||
total_new += 1
|
for page in fetch_channel_videos_iter(ch["channel_id"], published_after=after):
|
||||||
|
fetched += len(page)
|
||||||
|
for v in page:
|
||||||
|
if title_filter and title_filter not in v["title"]:
|
||||||
|
continue
|
||||||
|
row_id = save_video(ch["id"], v)
|
||||||
|
if row_id:
|
||||||
|
new_count += 1
|
||||||
|
total_new += new_count
|
||||||
logger.info(
|
logger.info(
|
||||||
"Channel %s: fetched %d videos, %d new",
|
"Channel %s: fetched %d videos (after=%s), %d new (filter=%s)",
|
||||||
ch["channel_name"], len(videos), total_new,
|
ch["channel_name"], fetched, after or "all", new_count, title_filter or "none",
|
||||||
)
|
)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error("Failed to scan channel %s: %s", ch["channel_name"], e)
|
logger.error("Failed to scan channel %s: %s", ch["channel_name"], e)
|
||||||
|
|||||||
@@ -6,4 +6,11 @@ load_dotenv()
|
|||||||
import uvicorn
|
import uvicorn
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
uvicorn.run("api.main:app", host="0.0.0.0", port=8000, reload=True)
|
uvicorn.run(
|
||||||
|
"api.main:app",
|
||||||
|
host="0.0.0.0",
|
||||||
|
port=8000,
|
||||||
|
reload=True,
|
||||||
|
workers=1,
|
||||||
|
limit_concurrency=20,
|
||||||
|
)
|
||||||
|
|||||||
@@ -2,25 +2,25 @@ module.exports = {
|
|||||||
apps: [
|
apps: [
|
||||||
{
|
{
|
||||||
name: "tasteby-api",
|
name: "tasteby-api",
|
||||||
cwd: "./backend",
|
cwd: "/Users/joungmin/workspaces/tasteby/backend",
|
||||||
script: "run_api.py",
|
script: "run_api.py",
|
||||||
interpreter: ".venv/bin/python",
|
interpreter: "/Users/joungmin/workspaces/tasteby/backend/.venv/bin/python",
|
||||||
env: {
|
env: {
|
||||||
PYTHONPATH: ".",
|
PYTHONPATH: ".",
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "tasteby-daemon",
|
name: "tasteby-daemon",
|
||||||
cwd: "./backend",
|
cwd: "/Users/joungmin/workspaces/tasteby/backend",
|
||||||
script: "run_daemon.py",
|
script: "run_daemon.py",
|
||||||
interpreter: ".venv/bin/python",
|
interpreter: "/Users/joungmin/workspaces/tasteby/backend/.venv/bin/python",
|
||||||
env: {
|
env: {
|
||||||
PYTHONPATH: ".",
|
PYTHONPATH: ".",
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "tasteby-web",
|
name: "tasteby-web",
|
||||||
cwd: "./frontend",
|
cwd: "/Users/joungmin/workspaces/tasteby/frontend",
|
||||||
script: "npm",
|
script: "npm",
|
||||||
args: "run dev",
|
args: "run dev",
|
||||||
},
|
},
|
||||||
|
|||||||
1798
frontend/src/app/admin/page.tsx
Normal file
1798
frontend/src/app/admin/page.tsx
Normal file
File diff suppressed because it is too large
Load Diff
@@ -3,6 +3,7 @@
|
|||||||
:root {
|
:root {
|
||||||
--background: #ffffff;
|
--background: #ffffff;
|
||||||
--foreground: #171717;
|
--foreground: #171717;
|
||||||
|
color-scheme: light only;
|
||||||
}
|
}
|
||||||
|
|
||||||
@theme inline {
|
@theme inline {
|
||||||
@@ -11,6 +12,18 @@
|
|||||||
--font-sans: var(--font-geist);
|
--font-sans: var(--font-geist);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@media (prefers-color-scheme: dark) {
|
||||||
|
:root {
|
||||||
|
--background: #ffffff;
|
||||||
|
--foreground: #171717;
|
||||||
|
color-scheme: light only;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
* {
|
||||||
|
color-scheme: light only;
|
||||||
|
}
|
||||||
|
|
||||||
body {
|
body {
|
||||||
background: var(--background);
|
background: var(--background);
|
||||||
color: var(--foreground);
|
color: var(--foreground);
|
||||||
@@ -20,3 +33,21 @@ html, body, #__next {
|
|||||||
height: 100%;
|
height: 100%;
|
||||||
margin: 0;
|
margin: 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
input, select, textarea {
|
||||||
|
color-scheme: light;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Force Google Maps InfoWindow to light mode */
|
||||||
|
.gm-style .gm-style-iw,
|
||||||
|
.gm-style .gm-style-iw-c,
|
||||||
|
.gm-style .gm-style-iw-d,
|
||||||
|
.gm-style .gm-style-iw-t {
|
||||||
|
background-color: #ffffff !important;
|
||||||
|
color: #171717 !important;
|
||||||
|
color-scheme: light !important;
|
||||||
|
}
|
||||||
|
|
||||||
|
.gm-style .gm-style-iw-d {
|
||||||
|
overflow: auto !important;
|
||||||
|
}
|
||||||
|
|||||||
@@ -19,7 +19,7 @@ export default function RootLayout({
|
|||||||
children: React.ReactNode;
|
children: React.ReactNode;
|
||||||
}>) {
|
}>) {
|
||||||
return (
|
return (
|
||||||
<html lang="ko">
|
<html lang="ko" style={{ colorScheme: "light" }}>
|
||||||
<body className={`${geist.variable} font-sans antialiased`}>
|
<body className={`${geist.variable} font-sans antialiased`}>
|
||||||
<Providers>{children}</Providers>
|
<Providers>{children}</Providers>
|
||||||
</body>
|
</body>
|
||||||
|
|||||||
@@ -3,12 +3,13 @@
|
|||||||
import { useCallback, useEffect, useState } from "react";
|
import { useCallback, useEffect, useState } from "react";
|
||||||
import { GoogleLogin } from "@react-oauth/google";
|
import { GoogleLogin } from "@react-oauth/google";
|
||||||
import { api } from "@/lib/api";
|
import { api } from "@/lib/api";
|
||||||
import type { Restaurant } from "@/lib/api";
|
import type { Restaurant, Channel, Review } from "@/lib/api";
|
||||||
import { useAuth } from "@/lib/auth-context";
|
import { useAuth } from "@/lib/auth-context";
|
||||||
import MapView from "@/components/MapView";
|
import MapView from "@/components/MapView";
|
||||||
import SearchBar from "@/components/SearchBar";
|
import SearchBar from "@/components/SearchBar";
|
||||||
import RestaurantList from "@/components/RestaurantList";
|
import RestaurantList from "@/components/RestaurantList";
|
||||||
import RestaurantDetail from "@/components/RestaurantDetail";
|
import RestaurantDetail from "@/components/RestaurantDetail";
|
||||||
|
import MyReviewsList from "@/components/MyReviewsList";
|
||||||
|
|
||||||
export default function Home() {
|
export default function Home() {
|
||||||
const { user, login, logout, isLoading: authLoading } = useAuth();
|
const { user, login, logout, isLoading: authLoading } = useAuth();
|
||||||
@@ -16,14 +17,26 @@ export default function Home() {
|
|||||||
const [selected, setSelected] = useState<Restaurant | null>(null);
|
const [selected, setSelected] = useState<Restaurant | null>(null);
|
||||||
const [loading, setLoading] = useState(false);
|
const [loading, setLoading] = useState(false);
|
||||||
const [showDetail, setShowDetail] = useState(false);
|
const [showDetail, setShowDetail] = useState(false);
|
||||||
|
const [channels, setChannels] = useState<Channel[]>([]);
|
||||||
|
const [channelFilter, setChannelFilter] = useState("");
|
||||||
|
const [showFavorites, setShowFavorites] = useState(false);
|
||||||
|
const [showMyReviews, setShowMyReviews] = useState(false);
|
||||||
|
const [myReviews, setMyReviews] = useState<(Review & { restaurant_id: string; restaurant_name: string | null })[]>([]);
|
||||||
|
const [visits, setVisits] = useState<{ today: number; total: number } | null>(null);
|
||||||
|
|
||||||
// Load all restaurants on mount
|
// Load channels + record visit on mount
|
||||||
|
useEffect(() => {
|
||||||
|
api.getChannels().then(setChannels).catch(console.error);
|
||||||
|
api.recordVisit().then(() => api.getVisits().then(setVisits)).catch(console.error);
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
// Load restaurants on mount and when channel filter changes
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
api
|
api
|
||||||
.getRestaurants({ limit: 200 })
|
.getRestaurants({ limit: 200, channel: channelFilter || undefined })
|
||||||
.then(setRestaurants)
|
.then(setRestaurants)
|
||||||
.catch(console.error);
|
.catch(console.error);
|
||||||
}, []);
|
}, [channelFilter]);
|
||||||
|
|
||||||
const handleSearch = useCallback(
|
const handleSearch = useCallback(
|
||||||
async (query: string, mode: "keyword" | "semantic" | "hybrid") => {
|
async (query: string, mode: "keyword" | "semantic" | "hybrid") => {
|
||||||
@@ -53,6 +66,9 @@ export default function Home() {
|
|||||||
|
|
||||||
const handleReset = useCallback(() => {
|
const handleReset = useCallback(() => {
|
||||||
setLoading(true);
|
setLoading(true);
|
||||||
|
setChannelFilter("");
|
||||||
|
setShowFavorites(false);
|
||||||
|
setShowMyReviews(false);
|
||||||
api
|
api
|
||||||
.getRestaurants({ limit: 200 })
|
.getRestaurants({ limit: 200 })
|
||||||
.then((data) => {
|
.then((data) => {
|
||||||
@@ -64,77 +80,260 @@ export default function Home() {
|
|||||||
.finally(() => setLoading(false));
|
.finally(() => setLoading(false));
|
||||||
}, []);
|
}, []);
|
||||||
|
|
||||||
|
const handleToggleFavorites = async () => {
|
||||||
|
if (showFavorites) {
|
||||||
|
setShowFavorites(false);
|
||||||
|
const data = await api.getRestaurants({ limit: 200, channel: channelFilter || undefined });
|
||||||
|
setRestaurants(data);
|
||||||
|
} else {
|
||||||
|
try {
|
||||||
|
const favs = await api.getMyFavorites();
|
||||||
|
setRestaurants(favs);
|
||||||
|
setShowFavorites(true);
|
||||||
|
setShowMyReviews(false);
|
||||||
|
setMyReviews([]);
|
||||||
|
setSelected(null);
|
||||||
|
setShowDetail(false);
|
||||||
|
} catch { /* ignore */ }
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleToggleMyReviews = async () => {
|
||||||
|
if (showMyReviews) {
|
||||||
|
setShowMyReviews(false);
|
||||||
|
setMyReviews([]);
|
||||||
|
} else {
|
||||||
|
try {
|
||||||
|
const reviews = await api.getMyReviews();
|
||||||
|
setMyReviews(reviews);
|
||||||
|
setShowMyReviews(true);
|
||||||
|
setShowFavorites(false);
|
||||||
|
setSelected(null);
|
||||||
|
setShowDetail(false);
|
||||||
|
} catch { /* ignore */ }
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const sidebarContent = showMyReviews ? (
|
||||||
|
<MyReviewsList
|
||||||
|
reviews={myReviews}
|
||||||
|
onClose={() => { setShowMyReviews(false); setMyReviews([]); }}
|
||||||
|
onSelectRestaurant={async (restaurantId) => {
|
||||||
|
try {
|
||||||
|
const r = await api.getRestaurant(restaurantId);
|
||||||
|
handleSelectRestaurant(r);
|
||||||
|
setShowMyReviews(false);
|
||||||
|
setMyReviews([]);
|
||||||
|
} catch { /* ignore */ }
|
||||||
|
}}
|
||||||
|
/>
|
||||||
|
) : showDetail && selected ? (
|
||||||
|
<RestaurantDetail
|
||||||
|
restaurant={selected}
|
||||||
|
onClose={handleCloseDetail}
|
||||||
|
/>
|
||||||
|
) : (
|
||||||
|
<RestaurantList
|
||||||
|
restaurants={restaurants}
|
||||||
|
selectedId={selected?.id}
|
||||||
|
onSelect={handleSelectRestaurant}
|
||||||
|
/>
|
||||||
|
);
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="h-screen flex flex-col">
|
<div className="h-screen flex flex-col">
|
||||||
{/* Header */}
|
{/* ── Header row 1: Logo + User ── */}
|
||||||
<header className="bg-white border-b px-4 py-3 flex items-center gap-4 shrink-0">
|
<header className="bg-white border-b shrink-0">
|
||||||
<button onClick={handleReset} className="text-lg font-bold whitespace-nowrap">
|
<div className="px-4 py-2 flex items-center justify-between">
|
||||||
Tasteby
|
<button onClick={handleReset} className="text-lg font-bold whitespace-nowrap">
|
||||||
</button>
|
Tasteby
|
||||||
<div className="flex-1 max-w-xl">
|
</button>
|
||||||
<SearchBar onSearch={handleSearch} isLoading={loading} />
|
|
||||||
</div>
|
{/* Desktop: search inline */}
|
||||||
<span className="text-sm text-gray-500">
|
<div className="hidden md:block flex-1 max-w-xl mx-4">
|
||||||
{restaurants.length}개 식당
|
<SearchBar onSearch={handleSearch} isLoading={loading} />
|
||||||
</span>
|
</div>
|
||||||
<div className="shrink-0">
|
|
||||||
{authLoading ? null : user ? (
|
{/* Desktop: filters inline */}
|
||||||
<div className="flex items-center gap-2">
|
<div className="hidden md:flex items-center gap-2">
|
||||||
{user.avatar_url && (
|
<select
|
||||||
<img
|
value={channelFilter}
|
||||||
src={user.avatar_url}
|
onChange={(e) => {
|
||||||
alt=""
|
setChannelFilter(e.target.value);
|
||||||
className="w-7 h-7 rounded-full"
|
setSelected(null);
|
||||||
/>
|
setShowDetail(false);
|
||||||
)}
|
|
||||||
<span className="text-sm">{user.nickname || user.email}</span>
|
|
||||||
<button
|
|
||||||
onClick={logout}
|
|
||||||
className="text-xs text-gray-500 hover:text-gray-700"
|
|
||||||
>
|
|
||||||
로그아웃
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
) : (
|
|
||||||
<GoogleLogin
|
|
||||||
onSuccess={(credentialResponse) => {
|
|
||||||
if (credentialResponse.credential) {
|
|
||||||
login(credentialResponse.credential).catch(console.error);
|
|
||||||
}
|
|
||||||
}}
|
}}
|
||||||
onError={() => console.error("Google login failed")}
|
className="border rounded px-2 py-1.5 text-sm text-gray-600"
|
||||||
size="small"
|
>
|
||||||
/>
|
<option value="">전체 채널</option>
|
||||||
)}
|
{channels.map((ch) => (
|
||||||
|
<option key={ch.id} value={ch.channel_name}>
|
||||||
|
{ch.channel_name}
|
||||||
|
</option>
|
||||||
|
))}
|
||||||
|
</select>
|
||||||
|
{user && (
|
||||||
|
<>
|
||||||
|
<button
|
||||||
|
onClick={handleToggleFavorites}
|
||||||
|
className={`px-3 py-1.5 text-sm rounded-full border transition-colors ${
|
||||||
|
showFavorites
|
||||||
|
? "bg-red-50 border-red-300 text-red-600"
|
||||||
|
: "border-gray-300 text-gray-600 hover:bg-gray-100"
|
||||||
|
}`}
|
||||||
|
>
|
||||||
|
{showFavorites ? "♥ 내 찜" : "♡ 찜"}
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
onClick={handleToggleMyReviews}
|
||||||
|
className={`px-3 py-1.5 text-sm rounded-full border transition-colors ${
|
||||||
|
showMyReviews
|
||||||
|
? "bg-blue-50 border-blue-300 text-blue-600"
|
||||||
|
: "border-gray-300 text-gray-600 hover:bg-gray-100"
|
||||||
|
}`}
|
||||||
|
>
|
||||||
|
{showMyReviews ? "✎ 내 리뷰" : "✎ 리뷰"}
|
||||||
|
</button>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
<span className="text-sm text-gray-500 whitespace-nowrap">
|
||||||
|
{restaurants.length}개
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="w-4 shrink-0 hidden md:block" />
|
||||||
|
|
||||||
|
{/* User area */}
|
||||||
|
<div className="shrink-0">
|
||||||
|
{authLoading ? null : user ? (
|
||||||
|
<div className="flex items-center gap-2">
|
||||||
|
{user.avatar_url ? (
|
||||||
|
<img
|
||||||
|
src={user.avatar_url}
|
||||||
|
alt=""
|
||||||
|
className="w-8 h-8 rounded-full border border-gray-200"
|
||||||
|
/>
|
||||||
|
) : (
|
||||||
|
<div className="w-8 h-8 rounded-full bg-amber-100 text-amber-700 flex items-center justify-center text-sm font-semibold border border-amber-200">
|
||||||
|
{(user.nickname || user.email || "?").charAt(0).toUpperCase()}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
<span className="hidden sm:inline text-sm font-medium text-gray-700">
|
||||||
|
{user.nickname || user.email}
|
||||||
|
</span>
|
||||||
|
<button
|
||||||
|
onClick={logout}
|
||||||
|
className="ml-1 px-2.5 py-1 text-xs text-gray-500 border border-gray-300 rounded-full hover:bg-gray-100 hover:text-gray-700 transition-colors"
|
||||||
|
>
|
||||||
|
로그아웃
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<GoogleLogin
|
||||||
|
onSuccess={(credentialResponse) => {
|
||||||
|
if (credentialResponse.credential) {
|
||||||
|
login(credentialResponse.credential).catch(console.error);
|
||||||
|
}
|
||||||
|
}}
|
||||||
|
onError={() => console.error("Google login failed")}
|
||||||
|
size="small"
|
||||||
|
/>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* ── Header row 2 (mobile only): search + filters ── */}
|
||||||
|
<div className="md:hidden px-4 pb-2 space-y-2">
|
||||||
|
<SearchBar onSearch={handleSearch} isLoading={loading} />
|
||||||
|
<div className="flex items-center gap-2 overflow-x-auto">
|
||||||
|
<select
|
||||||
|
value={channelFilter}
|
||||||
|
onChange={(e) => {
|
||||||
|
setChannelFilter(e.target.value);
|
||||||
|
setSelected(null);
|
||||||
|
setShowDetail(false);
|
||||||
|
}}
|
||||||
|
className="border rounded px-2 py-1 text-xs text-gray-600 shrink-0"
|
||||||
|
>
|
||||||
|
<option value="">전체 채널</option>
|
||||||
|
{channels.map((ch) => (
|
||||||
|
<option key={ch.id} value={ch.channel_name}>
|
||||||
|
{ch.channel_name}
|
||||||
|
</option>
|
||||||
|
))}
|
||||||
|
</select>
|
||||||
|
{user && (
|
||||||
|
<>
|
||||||
|
<button
|
||||||
|
onClick={handleToggleFavorites}
|
||||||
|
className={`px-2.5 py-1 text-xs rounded-full border transition-colors shrink-0 ${
|
||||||
|
showFavorites
|
||||||
|
? "bg-red-50 border-red-300 text-red-600"
|
||||||
|
: "border-gray-300 text-gray-600"
|
||||||
|
}`}
|
||||||
|
>
|
||||||
|
{showFavorites ? "♥ 내 찜" : "♡ 찜"}
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
onClick={handleToggleMyReviews}
|
||||||
|
className={`px-2.5 py-1 text-xs rounded-full border transition-colors shrink-0 ${
|
||||||
|
showMyReviews
|
||||||
|
? "bg-blue-50 border-blue-300 text-blue-600"
|
||||||
|
: "border-gray-300 text-gray-600"
|
||||||
|
}`}
|
||||||
|
>
|
||||||
|
{showMyReviews ? "✎ 내 리뷰" : "✎ 리뷰"}
|
||||||
|
</button>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
<span className="text-xs text-gray-400 shrink-0 ml-1">
|
||||||
|
{restaurants.length}개
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</header>
|
</header>
|
||||||
|
|
||||||
{/* Body */}
|
{/* ── Body: Desktop = side-by-side, Mobile = stacked ── */}
|
||||||
<div className="flex-1 flex overflow-hidden">
|
|
||||||
{/* Sidebar */}
|
|
||||||
<aside className="w-80 bg-white border-r overflow-y-auto shrink-0">
|
|
||||||
{showDetail && selected ? (
|
|
||||||
<RestaurantDetail
|
|
||||||
restaurant={selected}
|
|
||||||
onClose={handleCloseDetail}
|
|
||||||
/>
|
|
||||||
) : (
|
|
||||||
<RestaurantList
|
|
||||||
restaurants={restaurants}
|
|
||||||
selectedId={selected?.id}
|
|
||||||
onSelect={handleSelectRestaurant}
|
|
||||||
/>
|
|
||||||
)}
|
|
||||||
</aside>
|
|
||||||
|
|
||||||
{/* Map */}
|
{/* Desktop layout */}
|
||||||
<main className="flex-1">
|
<div className="hidden md:flex flex-1 overflow-hidden">
|
||||||
|
<aside className="w-80 bg-white border-r overflow-y-auto shrink-0">
|
||||||
|
{sidebarContent}
|
||||||
|
</aside>
|
||||||
|
<main className="flex-1 relative">
|
||||||
<MapView
|
<MapView
|
||||||
restaurants={restaurants}
|
restaurants={restaurants}
|
||||||
|
selected={selected}
|
||||||
onSelectRestaurant={handleSelectRestaurant}
|
onSelectRestaurant={handleSelectRestaurant}
|
||||||
/>
|
/>
|
||||||
|
{visits && (
|
||||||
|
<div className="absolute bottom-1 right-2 bg-black/40 text-white text-[10px] px-2 py-0.5 rounded">
|
||||||
|
오늘 {visits.today} · 전체 {visits.total.toLocaleString()}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
</main>
|
</main>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
{/* Mobile layout */}
|
||||||
|
<div className="md:hidden flex-1 flex flex-col overflow-hidden">
|
||||||
|
{/* Map: fixed height */}
|
||||||
|
<div className="h-[40vh] shrink-0 relative">
|
||||||
|
<MapView
|
||||||
|
restaurants={restaurants}
|
||||||
|
selected={selected}
|
||||||
|
onSelectRestaurant={handleSelectRestaurant}
|
||||||
|
/>
|
||||||
|
{visits && (
|
||||||
|
<div className="absolute bottom-1 right-2 bg-black/40 text-white text-[10px] px-2 py-0.5 rounded z-10">
|
||||||
|
오늘 {visits.today} · 전체 {visits.total.toLocaleString()}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
{/* List/Detail: scrollable below */}
|
||||||
|
<div className="flex-1 bg-white border-t overflow-y-auto">
|
||||||
|
{sidebarContent}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,33 +1,172 @@
|
|||||||
"use client";
|
"use client";
|
||||||
|
|
||||||
import { useCallback, useState } from "react";
|
import { useCallback, useEffect, useMemo, useRef, useState } from "react";
|
||||||
import {
|
import {
|
||||||
APIProvider,
|
APIProvider,
|
||||||
Map,
|
Map,
|
||||||
AdvancedMarker,
|
AdvancedMarker,
|
||||||
InfoWindow,
|
InfoWindow,
|
||||||
|
useMap,
|
||||||
} from "@vis.gl/react-google-maps";
|
} from "@vis.gl/react-google-maps";
|
||||||
import type { Restaurant } from "@/lib/api";
|
import type { Restaurant } from "@/lib/api";
|
||||||
|
|
||||||
const SEOUL_CENTER = { lat: 37.5665, lng: 126.978 };
|
const SEOUL_CENTER = { lat: 37.5665, lng: 126.978 };
|
||||||
const API_KEY = process.env.NEXT_PUBLIC_GOOGLE_MAPS_API_KEY || "";
|
const API_KEY = process.env.NEXT_PUBLIC_GOOGLE_MAPS_API_KEY || "";
|
||||||
|
|
||||||
|
// Channel color palette
|
||||||
|
const CHANNEL_COLORS = [
|
||||||
|
{ bg: "#fff7ed", text: "#78350f", border: "#f59e0b", arrow: "#f59e0b" }, // amber (default)
|
||||||
|
{ bg: "#eff6ff", text: "#1e3a5f", border: "#3b82f6", arrow: "#3b82f6" }, // blue
|
||||||
|
{ bg: "#f0fdf4", text: "#14532d", border: "#22c55e", arrow: "#22c55e" }, // green
|
||||||
|
{ bg: "#fdf2f8", text: "#831843", border: "#ec4899", arrow: "#ec4899" }, // pink
|
||||||
|
{ bg: "#faf5ff", text: "#581c87", border: "#a855f7", arrow: "#a855f7" }, // purple
|
||||||
|
{ bg: "#fff1f2", text: "#7f1d1d", border: "#ef4444", arrow: "#ef4444" }, // red
|
||||||
|
{ bg: "#f0fdfa", text: "#134e4a", border: "#14b8a6", arrow: "#14b8a6" }, // teal
|
||||||
|
{ bg: "#fefce8", text: "#713f12", border: "#eab308", arrow: "#eab308" }, // yellow
|
||||||
|
];
|
||||||
|
|
||||||
|
function getChannelColorMap(restaurants: Restaurant[]) {
|
||||||
|
const channels = new Set<string>();
|
||||||
|
restaurants.forEach((r) => r.channels?.forEach((ch) => channels.add(ch)));
|
||||||
|
const map: Record<string, typeof CHANNEL_COLORS[0]> = {};
|
||||||
|
let i = 0;
|
||||||
|
for (const ch of channels) {
|
||||||
|
map[ch] = CHANNEL_COLORS[i % CHANNEL_COLORS.length];
|
||||||
|
i++;
|
||||||
|
}
|
||||||
|
return map;
|
||||||
|
}
|
||||||
|
|
||||||
interface MapViewProps {
|
interface MapViewProps {
|
||||||
restaurants: Restaurant[];
|
restaurants: Restaurant[];
|
||||||
|
selected?: Restaurant | null;
|
||||||
onSelectRestaurant?: (r: Restaurant) => void;
|
onSelectRestaurant?: (r: Restaurant) => void;
|
||||||
}
|
}
|
||||||
|
|
||||||
export default function MapView({ restaurants, onSelectRestaurant }: MapViewProps) {
|
function MapContent({ restaurants, selected, onSelectRestaurant }: MapViewProps) {
|
||||||
const [selected, setSelected] = useState<Restaurant | null>(null);
|
const map = useMap();
|
||||||
|
const [infoTarget, setInfoTarget] = useState<Restaurant | null>(null);
|
||||||
|
const channelColors = useMemo(() => getChannelColorMap(restaurants), [restaurants]);
|
||||||
|
|
||||||
const handleMarkerClick = useCallback(
|
const handleMarkerClick = useCallback(
|
||||||
(r: Restaurant) => {
|
(r: Restaurant) => {
|
||||||
setSelected(r);
|
setInfoTarget(r);
|
||||||
onSelectRestaurant?.(r);
|
onSelectRestaurant?.(r);
|
||||||
},
|
},
|
||||||
[onSelectRestaurant]
|
[onSelectRestaurant]
|
||||||
);
|
);
|
||||||
|
|
||||||
|
// Pan and zoom to selected restaurant
|
||||||
|
useEffect(() => {
|
||||||
|
if (!map || !selected) return;
|
||||||
|
map.panTo({ lat: selected.latitude, lng: selected.longitude });
|
||||||
|
map.setZoom(16);
|
||||||
|
setInfoTarget(selected);
|
||||||
|
}, [map, selected]);
|
||||||
|
|
||||||
|
return (
|
||||||
|
<>
|
||||||
|
{restaurants.map((r) => {
|
||||||
|
const isSelected = selected?.id === r.id;
|
||||||
|
const isClosed = r.business_status === "CLOSED_PERMANENTLY";
|
||||||
|
const chColor = r.channels?.[0] ? channelColors[r.channels[0]] : CHANNEL_COLORS[0];
|
||||||
|
const c = chColor || CHANNEL_COLORS[0];
|
||||||
|
return (
|
||||||
|
<AdvancedMarker
|
||||||
|
key={r.id}
|
||||||
|
position={{ lat: r.latitude, lng: r.longitude }}
|
||||||
|
onClick={() => handleMarkerClick(r)}
|
||||||
|
zIndex={isSelected ? 1000 : 1}
|
||||||
|
>
|
||||||
|
<div style={{ display: "flex", flexDirection: "column", alignItems: "center", transition: "transform 0.2s ease", transform: isSelected ? "scale(1.15)" : "scale(1)", opacity: isClosed ? 0.5 : 1 }}>
|
||||||
|
<div
|
||||||
|
style={{
|
||||||
|
padding: "4px 8px",
|
||||||
|
backgroundColor: isSelected ? "#2563eb" : isClosed ? "#f3f4f6" : c.bg,
|
||||||
|
color: isSelected ? "#fff" : isClosed ? "#9ca3af" : c.text,
|
||||||
|
fontSize: 12,
|
||||||
|
fontWeight: 600,
|
||||||
|
borderRadius: 6,
|
||||||
|
border: isSelected ? "2px solid #1d4ed8" : `1.5px solid ${c.border}`,
|
||||||
|
boxShadow: isSelected
|
||||||
|
? "0 2px 8px rgba(37,99,235,0.4)"
|
||||||
|
: `0 1px 4px ${c.border}40`,
|
||||||
|
whiteSpace: "nowrap",
|
||||||
|
maxWidth: 120,
|
||||||
|
overflow: "hidden",
|
||||||
|
textOverflow: "ellipsis",
|
||||||
|
textDecoration: isClosed ? "line-through" : "none",
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
{r.name}
|
||||||
|
</div>
|
||||||
|
<div
|
||||||
|
style={{
|
||||||
|
width: 0,
|
||||||
|
height: 0,
|
||||||
|
borderLeft: "6px solid transparent",
|
||||||
|
borderRight: "6px solid transparent",
|
||||||
|
borderTop: isSelected ? "6px solid #1d4ed8" : `6px solid ${c.arrow}`,
|
||||||
|
marginTop: -1,
|
||||||
|
}}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
</AdvancedMarker>
|
||||||
|
);
|
||||||
|
})}
|
||||||
|
|
||||||
|
{infoTarget && (
|
||||||
|
<InfoWindow
|
||||||
|
position={{ lat: infoTarget.latitude, lng: infoTarget.longitude }}
|
||||||
|
onCloseClick={() => setInfoTarget(null)}
|
||||||
|
>
|
||||||
|
<div style={{ backgroundColor: "#ffffff", color: "#171717", colorScheme: "light" }} className="max-w-xs p-1">
|
||||||
|
<div className="flex items-center gap-2">
|
||||||
|
<h3 className="font-bold text-base" style={{ color: "#171717" }}>{infoTarget.name}</h3>
|
||||||
|
{infoTarget.business_status === "CLOSED_PERMANENTLY" && (
|
||||||
|
<span className="px-1.5 py-0.5 bg-red-100 text-red-700 rounded text-[10px] font-semibold">폐업</span>
|
||||||
|
)}
|
||||||
|
{infoTarget.business_status === "CLOSED_TEMPORARILY" && (
|
||||||
|
<span className="px-1.5 py-0.5 bg-yellow-100 text-yellow-700 rounded text-[10px] font-semibold">임시휴업</span>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
{infoTarget.rating && (
|
||||||
|
<p className="text-xs mt-0.5">
|
||||||
|
<span className="text-yellow-500">★</span> {infoTarget.rating}
|
||||||
|
{infoTarget.rating_count && (
|
||||||
|
<span className="text-gray-400 ml-1">({infoTarget.rating_count.toLocaleString()})</span>
|
||||||
|
)}
|
||||||
|
</p>
|
||||||
|
)}
|
||||||
|
{infoTarget.cuisine_type && (
|
||||||
|
<p className="text-sm text-gray-600">{infoTarget.cuisine_type}</p>
|
||||||
|
)}
|
||||||
|
{infoTarget.address && (
|
||||||
|
<p className="text-xs text-gray-500 mt-1">{infoTarget.address}</p>
|
||||||
|
)}
|
||||||
|
{infoTarget.price_range && (
|
||||||
|
<p className="text-xs text-gray-500">{infoTarget.price_range}</p>
|
||||||
|
)}
|
||||||
|
{infoTarget.phone && (
|
||||||
|
<p className="text-xs text-gray-500">{infoTarget.phone}</p>
|
||||||
|
)}
|
||||||
|
<button
|
||||||
|
onClick={() => onSelectRestaurant?.(infoTarget)}
|
||||||
|
className="mt-2 text-sm text-blue-600 hover:underline"
|
||||||
|
>
|
||||||
|
상세 보기
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</InfoWindow>
|
||||||
|
)}
|
||||||
|
</>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
export default function MapView({ restaurants, selected, onSelectRestaurant }: MapViewProps) {
|
||||||
|
const channelColors = useMemo(() => getChannelColorMap(restaurants), [restaurants]);
|
||||||
|
const channelNames = useMemo(() => Object.keys(channelColors), [channelColors]);
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<APIProvider apiKey={API_KEY}>
|
<APIProvider apiKey={API_KEY}>
|
||||||
<Map
|
<Map
|
||||||
@@ -35,41 +174,27 @@ export default function MapView({ restaurants, onSelectRestaurant }: MapViewProp
|
|||||||
defaultZoom={12}
|
defaultZoom={12}
|
||||||
mapId="tasteby-map"
|
mapId="tasteby-map"
|
||||||
className="h-full w-full"
|
className="h-full w-full"
|
||||||
|
colorScheme="LIGHT"
|
||||||
>
|
>
|
||||||
{restaurants.map((r) => (
|
<MapContent
|
||||||
<AdvancedMarker
|
restaurants={restaurants}
|
||||||
key={r.id}
|
selected={selected}
|
||||||
position={{ lat: r.latitude, lng: r.longitude }}
|
onSelectRestaurant={onSelectRestaurant}
|
||||||
onClick={() => handleMarkerClick(r)}
|
/>
|
||||||
/>
|
|
||||||
))}
|
|
||||||
|
|
||||||
{selected && (
|
|
||||||
<InfoWindow
|
|
||||||
position={{ lat: selected.latitude, lng: selected.longitude }}
|
|
||||||
onCloseClick={() => setSelected(null)}
|
|
||||||
>
|
|
||||||
<div className="max-w-xs p-1">
|
|
||||||
<h3 className="font-bold text-base">{selected.name}</h3>
|
|
||||||
{selected.cuisine_type && (
|
|
||||||
<p className="text-sm text-gray-600">{selected.cuisine_type}</p>
|
|
||||||
)}
|
|
||||||
{selected.address && (
|
|
||||||
<p className="text-xs text-gray-500 mt-1">{selected.address}</p>
|
|
||||||
)}
|
|
||||||
{selected.price_range && (
|
|
||||||
<p className="text-xs text-gray-500">{selected.price_range}</p>
|
|
||||||
)}
|
|
||||||
<button
|
|
||||||
onClick={() => onSelectRestaurant?.(selected)}
|
|
||||||
className="mt-2 text-sm text-blue-600 hover:underline"
|
|
||||||
>
|
|
||||||
상세 보기
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
</InfoWindow>
|
|
||||||
)}
|
|
||||||
</Map>
|
</Map>
|
||||||
|
{channelNames.length > 1 && (
|
||||||
|
<div className="absolute top-2 left-2 bg-white/90 rounded-lg shadow px-2.5 py-1.5 flex flex-wrap gap-x-3 gap-y-1 text-[11px] z-10">
|
||||||
|
{channelNames.map((ch) => (
|
||||||
|
<div key={ch} className="flex items-center gap-1">
|
||||||
|
<span
|
||||||
|
className="inline-block w-2.5 h-2.5 rounded-full border"
|
||||||
|
style={{ backgroundColor: channelColors[ch].border, borderColor: channelColors[ch].border }}
|
||||||
|
/>
|
||||||
|
<span className="text-gray-700">{ch}</span>
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
</APIProvider>
|
</APIProvider>
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|||||||
69
frontend/src/components/MyReviewsList.tsx
Normal file
69
frontend/src/components/MyReviewsList.tsx
Normal file
@@ -0,0 +1,69 @@
|
|||||||
|
"use client";
|
||||||
|
|
||||||
|
import type { Review } from "@/lib/api";
|
||||||
|
|
||||||
|
interface MyReview extends Review {
|
||||||
|
restaurant_id: string;
|
||||||
|
restaurant_name: string | null;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface MyReviewsListProps {
|
||||||
|
reviews: MyReview[];
|
||||||
|
onClose: () => void;
|
||||||
|
onSelectRestaurant: (restaurantId: string) => void;
|
||||||
|
}
|
||||||
|
|
||||||
|
export default function MyReviewsList({
|
||||||
|
reviews,
|
||||||
|
onClose,
|
||||||
|
onSelectRestaurant,
|
||||||
|
}: MyReviewsListProps) {
|
||||||
|
return (
|
||||||
|
<div className="p-4 space-y-3">
|
||||||
|
<div className="flex justify-between items-center">
|
||||||
|
<h2 className="font-bold text-lg">내 리뷰 ({reviews.length})</h2>
|
||||||
|
<button
|
||||||
|
onClick={onClose}
|
||||||
|
className="text-gray-400 hover:text-gray-600 text-xl leading-none"
|
||||||
|
>
|
||||||
|
x
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{reviews.length === 0 ? (
|
||||||
|
<p className="text-sm text-gray-500 py-8 text-center">
|
||||||
|
아직 작성한 리뷰가 없습니다.
|
||||||
|
</p>
|
||||||
|
) : (
|
||||||
|
<div className="space-y-2">
|
||||||
|
{reviews.map((r) => (
|
||||||
|
<button
|
||||||
|
key={r.id}
|
||||||
|
onClick={() => onSelectRestaurant(r.restaurant_id)}
|
||||||
|
className="w-full text-left border rounded-lg p-3 hover:bg-gray-50 transition-colors"
|
||||||
|
>
|
||||||
|
<div className="flex items-center justify-between mb-1">
|
||||||
|
<span className="font-semibold text-sm truncate">
|
||||||
|
{r.restaurant_name || "알 수 없는 식당"}
|
||||||
|
</span>
|
||||||
|
<span className="text-yellow-500 text-sm shrink-0 ml-2">
|
||||||
|
{"★".repeat(Math.round(r.rating))}
|
||||||
|
<span className="text-gray-500 ml-1">{r.rating}</span>
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
{r.review_text && (
|
||||||
|
<p className="text-xs text-gray-600 line-clamp-2">
|
||||||
|
{r.review_text}
|
||||||
|
</p>
|
||||||
|
)}
|
||||||
|
<div className="flex items-center gap-2 mt-1.5 text-[10px] text-gray-400">
|
||||||
|
{r.visited_at && <span>방문: {r.visited_at}</span>}
|
||||||
|
{r.created_at && <span>{r.created_at.slice(0, 10)}</span>}
|
||||||
|
</div>
|
||||||
|
</button>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
"use client";
|
"use client";
|
||||||
|
|
||||||
import { useEffect, useState } from "react";
|
import { useEffect, useState } from "react";
|
||||||
import { api } from "@/lib/api";
|
import { api, getToken } from "@/lib/api";
|
||||||
import type { Restaurant, VideoLink } from "@/lib/api";
|
import type { Restaurant, VideoLink } from "@/lib/api";
|
||||||
import ReviewSection from "@/components/ReviewSection";
|
import ReviewSection from "@/components/ReviewSection";
|
||||||
|
|
||||||
@@ -16,6 +16,8 @@ export default function RestaurantDetail({
|
|||||||
}: RestaurantDetailProps) {
|
}: RestaurantDetailProps) {
|
||||||
const [videos, setVideos] = useState<VideoLink[]>([]);
|
const [videos, setVideos] = useState<VideoLink[]>([]);
|
||||||
const [loading, setLoading] = useState(true);
|
const [loading, setLoading] = useState(true);
|
||||||
|
const [favorited, setFavorited] = useState(false);
|
||||||
|
const [favLoading, setFavLoading] = useState(false);
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
setLoading(true);
|
setLoading(true);
|
||||||
@@ -24,12 +26,53 @@ export default function RestaurantDetail({
|
|||||||
.then(setVideos)
|
.then(setVideos)
|
||||||
.catch(() => setVideos([]))
|
.catch(() => setVideos([]))
|
||||||
.finally(() => setLoading(false));
|
.finally(() => setLoading(false));
|
||||||
|
|
||||||
|
// Load favorite status if logged in
|
||||||
|
if (getToken()) {
|
||||||
|
api.getFavoriteStatus(restaurant.id)
|
||||||
|
.then((r) => setFavorited(r.favorited))
|
||||||
|
.catch(() => {});
|
||||||
|
}
|
||||||
}, [restaurant.id]);
|
}, [restaurant.id]);
|
||||||
|
|
||||||
|
const handleToggleFavorite = async () => {
|
||||||
|
if (!getToken()) return;
|
||||||
|
setFavLoading(true);
|
||||||
|
try {
|
||||||
|
const res = await api.toggleFavorite(restaurant.id);
|
||||||
|
setFavorited(res.favorited);
|
||||||
|
} catch { /* ignore */ }
|
||||||
|
finally { setFavLoading(false); }
|
||||||
|
};
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="p-4 space-y-4">
|
<div className="p-4 space-y-4">
|
||||||
<div className="flex justify-between items-start">
|
<div className="flex justify-between items-start">
|
||||||
<h2 className="text-lg font-bold">{restaurant.name}</h2>
|
<div className="flex items-center gap-2">
|
||||||
|
<h2 className="text-lg font-bold">{restaurant.name}</h2>
|
||||||
|
{getToken() && (
|
||||||
|
<button
|
||||||
|
onClick={handleToggleFavorite}
|
||||||
|
disabled={favLoading}
|
||||||
|
className={`text-xl leading-none transition-colors ${
|
||||||
|
favorited ? "text-red-500" : "text-gray-300 hover:text-red-400"
|
||||||
|
}`}
|
||||||
|
title={favorited ? "찜 해제" : "찜하기"}
|
||||||
|
>
|
||||||
|
{favorited ? "♥" : "♡"}
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
|
{restaurant.business_status === "CLOSED_PERMANENTLY" && (
|
||||||
|
<span className="px-2 py-0.5 bg-red-100 text-red-700 rounded text-xs font-semibold">
|
||||||
|
폐업
|
||||||
|
</span>
|
||||||
|
)}
|
||||||
|
{restaurant.business_status === "CLOSED_TEMPORARILY" && (
|
||||||
|
<span className="px-2 py-0.5 bg-yellow-100 text-yellow-700 rounded text-xs font-semibold">
|
||||||
|
임시휴업
|
||||||
|
</span>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
<button
|
<button
|
||||||
onClick={onClose}
|
onClick={onClose}
|
||||||
className="text-gray-400 hover:text-gray-600 text-xl leading-none"
|
className="text-gray-400 hover:text-gray-600 text-xl leading-none"
|
||||||
@@ -38,6 +81,16 @@ export default function RestaurantDetail({
|
|||||||
</button>
|
</button>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
{restaurant.rating && (
|
||||||
|
<div className="flex items-center gap-2 text-sm">
|
||||||
|
<span className="text-yellow-500">{"★".repeat(Math.round(restaurant.rating))}</span>
|
||||||
|
<span className="font-medium">{restaurant.rating}</span>
|
||||||
|
{restaurant.rating_count && (
|
||||||
|
<span className="text-gray-400 text-xs">({restaurant.rating_count.toLocaleString()})</span>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
<div className="space-y-1 text-sm">
|
<div className="space-y-1 text-sm">
|
||||||
{restaurant.cuisine_type && (
|
{restaurant.cuisine_type && (
|
||||||
<p>
|
<p>
|
||||||
@@ -59,6 +112,26 @@ export default function RestaurantDetail({
|
|||||||
<span className="text-gray-500">가격대:</span> {restaurant.price_range}
|
<span className="text-gray-500">가격대:</span> {restaurant.price_range}
|
||||||
</p>
|
</p>
|
||||||
)}
|
)}
|
||||||
|
{restaurant.phone && (
|
||||||
|
<p>
|
||||||
|
<span className="text-gray-500">전화:</span>{" "}
|
||||||
|
<a href={`tel:${restaurant.phone}`} className="text-blue-600 hover:underline">
|
||||||
|
{restaurant.phone}
|
||||||
|
</a>
|
||||||
|
</p>
|
||||||
|
)}
|
||||||
|
{restaurant.google_place_id && (
|
||||||
|
<p>
|
||||||
|
<a
|
||||||
|
href={`https://www.google.com/maps/place/?q=place_id:${restaurant.google_place_id}`}
|
||||||
|
target="_blank"
|
||||||
|
rel="noopener noreferrer"
|
||||||
|
className="text-blue-600 hover:underline text-xs"
|
||||||
|
>
|
||||||
|
Google Maps에서 보기
|
||||||
|
</a>
|
||||||
|
</p>
|
||||||
|
)}
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<div>
|
<div>
|
||||||
@@ -71,11 +144,23 @@ export default function RestaurantDetail({
|
|||||||
<div className="space-y-3">
|
<div className="space-y-3">
|
||||||
{videos.map((v) => (
|
{videos.map((v) => (
|
||||||
<div key={v.video_id} className="border rounded-lg p-3">
|
<div key={v.video_id} className="border rounded-lg p-3">
|
||||||
|
<div className="flex items-center gap-2 mb-1">
|
||||||
|
{v.channel_name && (
|
||||||
|
<span className="inline-block px-1.5 py-0.5 bg-red-50 text-red-600 rounded text-[10px] font-medium">
|
||||||
|
{v.channel_name}
|
||||||
|
</span>
|
||||||
|
)}
|
||||||
|
{v.published_at && (
|
||||||
|
<span className="text-[10px] text-gray-400">
|
||||||
|
{v.published_at.slice(0, 10)}
|
||||||
|
</span>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
<a
|
<a
|
||||||
href={v.url}
|
href={v.url}
|
||||||
target="_blank"
|
target="_blank"
|
||||||
rel="noopener noreferrer"
|
rel="noopener noreferrer"
|
||||||
className="text-sm font-medium text-blue-600 hover:underline"
|
className="block text-sm font-medium text-blue-600 hover:underline"
|
||||||
>
|
>
|
||||||
{v.title}
|
{v.title}
|
||||||
</a>
|
</a>
|
||||||
@@ -107,6 +192,17 @@ export default function RestaurantDetail({
|
|||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
{videos.length > 0 && (
|
||||||
|
<div className="bg-gray-50 rounded-lg px-4 py-3 text-center space-y-1">
|
||||||
|
<p className="text-xs text-gray-500">
|
||||||
|
더 자세한 내용은 영상에서 확인하실 수 있습니다.
|
||||||
|
</p>
|
||||||
|
<p className="text-xs text-gray-400">
|
||||||
|
맛집을 소개해주신 크리에이터를 구독과 좋아요로 응원해주세요!
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
<ReviewSection restaurantId={restaurant.id} />
|
<ReviewSection restaurantId={restaurant.id} />
|
||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
|
|||||||
@@ -37,6 +37,18 @@ export default function RestaurantList({
|
|||||||
{r.region && <span>{r.region}</span>}
|
{r.region && <span>{r.region}</span>}
|
||||||
{r.price_range && <span>{r.price_range}</span>}
|
{r.price_range && <span>{r.price_range}</span>}
|
||||||
</div>
|
</div>
|
||||||
|
{r.channels && r.channels.length > 0 && (
|
||||||
|
<div className="flex flex-wrap gap-1 mt-1">
|
||||||
|
{r.channels.map((ch) => (
|
||||||
|
<span
|
||||||
|
key={ch}
|
||||||
|
className="px-1.5 py-0.5 bg-red-50 text-red-600 rounded text-[10px] font-medium"
|
||||||
|
>
|
||||||
|
{ch}
|
||||||
|
</span>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
</button>
|
</button>
|
||||||
))}
|
))}
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
@@ -19,18 +19,18 @@ export default function SearchBar({ onSearch, isLoading }: SearchBarProps) {
|
|||||||
};
|
};
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<form onSubmit={handleSubmit} className="flex gap-2 items-center">
|
<form onSubmit={handleSubmit} className="flex gap-1.5 items-center">
|
||||||
<input
|
<input
|
||||||
type="text"
|
type="text"
|
||||||
value={query}
|
value={query}
|
||||||
onChange={(e) => setQuery(e.target.value)}
|
onChange={(e) => setQuery(e.target.value)}
|
||||||
placeholder="식당, 지역, 음식 종류..."
|
placeholder="식당, 지역, 음식..."
|
||||||
className="flex-1 px-4 py-2 border border-gray-300 rounded-lg focus:outline-none focus:ring-2 focus:ring-blue-500 text-sm"
|
className="flex-1 min-w-0 px-3 py-2 border border-gray-300 rounded-lg focus:outline-none focus:ring-2 focus:ring-blue-500 text-sm"
|
||||||
/>
|
/>
|
||||||
<select
|
<select
|
||||||
value={mode}
|
value={mode}
|
||||||
onChange={(e) => setMode(e.target.value as typeof mode)}
|
onChange={(e) => setMode(e.target.value as typeof mode)}
|
||||||
className="px-3 py-2 border border-gray-300 rounded-lg text-sm bg-white"
|
className="shrink-0 px-2 py-2 border border-gray-300 rounded-lg text-sm bg-white"
|
||||||
>
|
>
|
||||||
<option value="hybrid">통합</option>
|
<option value="hybrid">통합</option>
|
||||||
<option value="keyword">키워드</option>
|
<option value="keyword">키워드</option>
|
||||||
@@ -39,7 +39,7 @@ export default function SearchBar({ onSearch, isLoading }: SearchBarProps) {
|
|||||||
<button
|
<button
|
||||||
type="submit"
|
type="submit"
|
||||||
disabled={isLoading || !query.trim()}
|
disabled={isLoading || !query.trim()}
|
||||||
className="px-4 py-2 bg-blue-600 text-white rounded-lg hover:bg-blue-700 disabled:opacity-50 text-sm"
|
className="shrink-0 px-3 py-2 bg-blue-600 text-white rounded-lg hover:bg-blue-700 disabled:opacity-50 text-sm"
|
||||||
>
|
>
|
||||||
{isLoading ? "..." : "검색"}
|
{isLoading ? "..." : "검색"}
|
||||||
</button>
|
</button>
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
const API_BASE = process.env.NEXT_PUBLIC_API_URL || "http://localhost:8000";
|
const API_BASE = process.env.NEXT_PUBLIC_API_URL || "";
|
||||||
|
|
||||||
const TOKEN_KEY = "tasteby_token";
|
const TOKEN_KEY = "tasteby_token";
|
||||||
|
|
||||||
@@ -42,6 +42,12 @@ export interface Restaurant {
|
|||||||
cuisine_type: string | null;
|
cuisine_type: string | null;
|
||||||
price_range: string | null;
|
price_range: string | null;
|
||||||
google_place_id: string | null;
|
google_place_id: string | null;
|
||||||
|
business_status: string | null;
|
||||||
|
rating: number | null;
|
||||||
|
rating_count: number | null;
|
||||||
|
phone: string | null;
|
||||||
|
website: string | null;
|
||||||
|
channels?: string[];
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface VideoLink {
|
export interface VideoLink {
|
||||||
@@ -52,12 +58,48 @@ export interface VideoLink {
|
|||||||
foods_mentioned: string[];
|
foods_mentioned: string[];
|
||||||
evaluation: Record<string, string>;
|
evaluation: Record<string, string>;
|
||||||
guests: string[];
|
guests: string[];
|
||||||
|
channel_name: string | null;
|
||||||
|
channel_id: string | null;
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface Channel {
|
export interface Channel {
|
||||||
id: string;
|
id: string;
|
||||||
channel_id: string;
|
channel_id: string;
|
||||||
channel_name: string;
|
channel_name: string;
|
||||||
|
title_filter: string | null;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface Video {
|
||||||
|
id: string;
|
||||||
|
video_id: string;
|
||||||
|
title: string;
|
||||||
|
url: string;
|
||||||
|
status: string;
|
||||||
|
published_at: string | null;
|
||||||
|
channel_name: string;
|
||||||
|
has_transcript: boolean;
|
||||||
|
has_llm: boolean;
|
||||||
|
restaurant_count: number;
|
||||||
|
matched_count: number;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface VideoRestaurant {
|
||||||
|
restaurant_id: string;
|
||||||
|
name: string;
|
||||||
|
address: string | null;
|
||||||
|
cuisine_type: string | null;
|
||||||
|
price_range: string | null;
|
||||||
|
region: string | null;
|
||||||
|
foods_mentioned: string[];
|
||||||
|
evaluation: Record<string, string>;
|
||||||
|
guests: string[];
|
||||||
|
google_place_id: string | null;
|
||||||
|
has_location: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface VideoDetail extends Video {
|
||||||
|
transcript: string | null;
|
||||||
|
restaurants: VideoRestaurant[];
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface User {
|
export interface User {
|
||||||
@@ -88,12 +130,14 @@ export const api = {
|
|||||||
getRestaurants(params?: {
|
getRestaurants(params?: {
|
||||||
cuisine?: string;
|
cuisine?: string;
|
||||||
region?: string;
|
region?: string;
|
||||||
|
channel?: string;
|
||||||
limit?: number;
|
limit?: number;
|
||||||
offset?: number;
|
offset?: number;
|
||||||
}) {
|
}) {
|
||||||
const sp = new URLSearchParams();
|
const sp = new URLSearchParams();
|
||||||
if (params?.cuisine) sp.set("cuisine", params.cuisine);
|
if (params?.cuisine) sp.set("cuisine", params.cuisine);
|
||||||
if (params?.region) sp.set("region", params.region);
|
if (params?.region) sp.set("region", params.region);
|
||||||
|
if (params?.channel) sp.set("channel", params.channel);
|
||||||
if (params?.limit) sp.set("limit", String(params.limit));
|
if (params?.limit) sp.set("limit", String(params.limit));
|
||||||
if (params?.offset) sp.set("offset", String(params.offset));
|
if (params?.offset) sp.set("offset", String(params.offset));
|
||||||
const qs = sp.toString();
|
const qs = sp.toString();
|
||||||
@@ -104,6 +148,19 @@ export const api = {
|
|||||||
return fetchApi<Restaurant>(`/api/restaurants/${id}`);
|
return fetchApi<Restaurant>(`/api/restaurants/${id}`);
|
||||||
},
|
},
|
||||||
|
|
||||||
|
updateRestaurant(id: string, data: Partial<Restaurant>) {
|
||||||
|
return fetchApi<{ ok: boolean }>(`/api/restaurants/${id}`, {
|
||||||
|
method: "PUT",
|
||||||
|
body: JSON.stringify(data),
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
|
deleteRestaurant(id: string) {
|
||||||
|
return fetchApi<{ ok: boolean }>(`/api/restaurants/${id}`, {
|
||||||
|
method: "DELETE",
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
getRestaurantVideos(id: string) {
|
getRestaurantVideos(id: string) {
|
||||||
return fetchApi<VideoLink[]>(`/api/restaurants/${id}/videos`);
|
return fetchApi<VideoLink[]>(`/api/restaurants/${id}/videos`);
|
||||||
},
|
},
|
||||||
@@ -129,6 +186,20 @@ export const api = {
|
|||||||
return fetchApi<User>("/api/auth/me");
|
return fetchApi<User>("/api/auth/me");
|
||||||
},
|
},
|
||||||
|
|
||||||
|
getFavoriteStatus(restaurantId: string) {
|
||||||
|
return fetchApi<{ favorited: boolean }>(`/api/restaurants/${restaurantId}/favorite`);
|
||||||
|
},
|
||||||
|
|
||||||
|
toggleFavorite(restaurantId: string) {
|
||||||
|
return fetchApi<{ favorited: boolean }>(`/api/restaurants/${restaurantId}/favorite`, {
|
||||||
|
method: "POST",
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
|
getMyFavorites() {
|
||||||
|
return fetchApi<Restaurant[]>("/api/users/me/favorites");
|
||||||
|
},
|
||||||
|
|
||||||
getReviews(restaurantId: string) {
|
getReviews(restaurantId: string) {
|
||||||
return fetchApi<ReviewsResponse>(`/api/restaurants/${restaurantId}/reviews`);
|
return fetchApi<ReviewsResponse>(`/api/restaurants/${restaurantId}/reviews`);
|
||||||
},
|
},
|
||||||
@@ -158,4 +229,203 @@ export const api = {
|
|||||||
method: "DELETE",
|
method: "DELETE",
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
|
|
||||||
|
getMyReviews() {
|
||||||
|
return fetchApi<(Review & { restaurant_id: string; restaurant_name: string | null })[]>(
|
||||||
|
"/api/users/me/reviews?limit=100"
|
||||||
|
);
|
||||||
|
},
|
||||||
|
|
||||||
|
// Stats
|
||||||
|
recordVisit() {
|
||||||
|
return fetchApi<{ ok: boolean }>("/api/stats/visit", { method: "POST" });
|
||||||
|
},
|
||||||
|
|
||||||
|
getVisits() {
|
||||||
|
return fetchApi<{ today: number; total: number }>("/api/stats/visits");
|
||||||
|
},
|
||||||
|
|
||||||
|
// Admin - Users
|
||||||
|
getAdminUsers(params?: { limit?: number; offset?: number }) {
|
||||||
|
const sp = new URLSearchParams();
|
||||||
|
if (params?.limit) sp.set("limit", String(params.limit));
|
||||||
|
if (params?.offset) sp.set("offset", String(params.offset));
|
||||||
|
const qs = sp.toString();
|
||||||
|
return fetchApi<{
|
||||||
|
users: {
|
||||||
|
id: string;
|
||||||
|
email: string | null;
|
||||||
|
nickname: string | null;
|
||||||
|
avatar_url: string | null;
|
||||||
|
provider: string | null;
|
||||||
|
created_at: string | null;
|
||||||
|
favorite_count: number;
|
||||||
|
review_count: number;
|
||||||
|
}[];
|
||||||
|
total: number;
|
||||||
|
}>(`/api/admin/users${qs ? `?${qs}` : ""}`);
|
||||||
|
},
|
||||||
|
|
||||||
|
getAdminUserFavorites(userId: string) {
|
||||||
|
return fetchApi<
|
||||||
|
{
|
||||||
|
id: string;
|
||||||
|
name: string;
|
||||||
|
address: string | null;
|
||||||
|
region: string | null;
|
||||||
|
cuisine_type: string | null;
|
||||||
|
rating: number | null;
|
||||||
|
business_status: string | null;
|
||||||
|
created_at: string | null;
|
||||||
|
}[]
|
||||||
|
>(`/api/admin/users/${userId}/favorites`);
|
||||||
|
},
|
||||||
|
|
||||||
|
getAdminUserReviews(userId: string) {
|
||||||
|
return fetchApi<
|
||||||
|
{
|
||||||
|
id: string;
|
||||||
|
restaurant_id: string;
|
||||||
|
rating: number;
|
||||||
|
review_text: string | null;
|
||||||
|
visited_at: string | null;
|
||||||
|
created_at: string | null;
|
||||||
|
restaurant_name: string | null;
|
||||||
|
}[]
|
||||||
|
>(`/api/admin/users/${userId}/reviews`);
|
||||||
|
},
|
||||||
|
|
||||||
|
// Admin
|
||||||
|
addChannel(channelId: string, channelName: string, titleFilter?: string) {
|
||||||
|
return fetchApi<{ id: string; channel_id: string }>("/api/channels", {
|
||||||
|
method: "POST",
|
||||||
|
body: JSON.stringify({
|
||||||
|
channel_id: channelId,
|
||||||
|
channel_name: channelName,
|
||||||
|
title_filter: titleFilter || null,
|
||||||
|
}),
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
|
deleteChannel(channelId: string) {
|
||||||
|
return fetchApi<{ ok: boolean }>(`/api/channels/${channelId}`, {
|
||||||
|
method: "DELETE",
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
|
scanChannel(channelId: string, full: boolean = false) {
|
||||||
|
return fetchApi<{ total_fetched: number; new_videos: number }>(
|
||||||
|
`/api/channels/${channelId}/scan${full ? "?full=true" : ""}`,
|
||||||
|
{ method: "POST" }
|
||||||
|
);
|
||||||
|
},
|
||||||
|
|
||||||
|
getVideos(params?: { status?: string; limit?: number; offset?: number }) {
|
||||||
|
const sp = new URLSearchParams();
|
||||||
|
if (params?.status) sp.set("status", params.status);
|
||||||
|
if (params?.limit) sp.set("limit", String(params.limit));
|
||||||
|
if (params?.offset) sp.set("offset", String(params.offset));
|
||||||
|
const qs = sp.toString();
|
||||||
|
return fetchApi<Video[]>(`/api/videos${qs ? `?${qs}` : ""}`);
|
||||||
|
},
|
||||||
|
|
||||||
|
getVideoDetail(videoDbId: string) {
|
||||||
|
return fetchApi<VideoDetail>(`/api/videos/${videoDbId}`);
|
||||||
|
},
|
||||||
|
|
||||||
|
skipVideo(videoDbId: string) {
|
||||||
|
return fetchApi<{ ok: boolean }>(`/api/videos/${videoDbId}/skip`, {
|
||||||
|
method: "POST",
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
|
deleteVideo(videoDbId: string) {
|
||||||
|
return fetchApi<{ ok: boolean }>(`/api/videos/${videoDbId}`, {
|
||||||
|
method: "DELETE",
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
|
getExtractPrompt() {
|
||||||
|
return fetchApi<{ prompt: string }>("/api/videos/extract/prompt");
|
||||||
|
},
|
||||||
|
|
||||||
|
extractRestaurants(videoDbId: string, customPrompt?: string) {
|
||||||
|
return fetchApi<{ ok: boolean; restaurants_extracted: number }>(
|
||||||
|
`/api/videos/${videoDbId}/extract`,
|
||||||
|
{
|
||||||
|
method: "POST",
|
||||||
|
body: JSON.stringify(customPrompt ? { prompt: customPrompt } : {}),
|
||||||
|
}
|
||||||
|
);
|
||||||
|
},
|
||||||
|
|
||||||
|
fetchTranscript(videoDbId: string, mode: "auto" | "manual" | "generated" = "auto") {
|
||||||
|
return fetchApi<{ ok: boolean; length: number; source: string }>(
|
||||||
|
`/api/videos/${videoDbId}/fetch-transcript?mode=${mode}`,
|
||||||
|
{ method: "POST" }
|
||||||
|
);
|
||||||
|
},
|
||||||
|
|
||||||
|
triggerProcessing(limit: number = 5) {
|
||||||
|
return fetchApi<{ restaurants_extracted: number }>(
|
||||||
|
`/api/videos/process?limit=${limit}`,
|
||||||
|
{ method: "POST" }
|
||||||
|
);
|
||||||
|
},
|
||||||
|
|
||||||
|
getBulkExtractPending() {
|
||||||
|
return fetchApi<{ count: number; videos: { id: string; title: string }[] }>(
|
||||||
|
"/api/videos/bulk-extract/pending"
|
||||||
|
);
|
||||||
|
},
|
||||||
|
|
||||||
|
getBulkTranscriptPending() {
|
||||||
|
return fetchApi<{ count: number; videos: { id: string; title: string }[] }>(
|
||||||
|
"/api/videos/bulk-transcript/pending"
|
||||||
|
);
|
||||||
|
},
|
||||||
|
|
||||||
|
updateVideo(videoDbId: string, data: { title: string }) {
|
||||||
|
return fetchApi<{ ok: boolean }>(`/api/videos/${videoDbId}`, {
|
||||||
|
method: "PUT",
|
||||||
|
body: JSON.stringify(data),
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
|
addManualRestaurant(
|
||||||
|
videoDbId: string,
|
||||||
|
data: {
|
||||||
|
name: string;
|
||||||
|
address?: string;
|
||||||
|
region?: string;
|
||||||
|
cuisine_type?: string;
|
||||||
|
price_range?: string;
|
||||||
|
foods_mentioned?: string[];
|
||||||
|
evaluation?: string;
|
||||||
|
guests?: string[];
|
||||||
|
}
|
||||||
|
) {
|
||||||
|
return fetchApi<{ ok: boolean; restaurant_id: string; link_id: string }>(
|
||||||
|
`/api/videos/${videoDbId}/restaurants/manual`,
|
||||||
|
{ method: "POST", body: JSON.stringify(data) }
|
||||||
|
);
|
||||||
|
},
|
||||||
|
|
||||||
|
deleteVideoRestaurant(videoDbId: string, restaurantId: string) {
|
||||||
|
return fetchApi<{ ok: boolean }>(
|
||||||
|
`/api/videos/${videoDbId}/restaurants/${restaurantId}`,
|
||||||
|
{ method: "DELETE" }
|
||||||
|
);
|
||||||
|
},
|
||||||
|
|
||||||
|
updateVideoRestaurant(
|
||||||
|
videoDbId: string,
|
||||||
|
restaurantId: string,
|
||||||
|
data: Partial<VideoRestaurant>
|
||||||
|
) {
|
||||||
|
return fetchApi<{ ok: boolean }>(
|
||||||
|
`/api/videos/${videoDbId}/restaurants/${restaurantId}`,
|
||||||
|
{ method: "PUT", body: JSON.stringify(data) }
|
||||||
|
);
|
||||||
|
},
|
||||||
};
|
};
|
||||||
|
|||||||
47
nginx.conf
Normal file
47
nginx.conf
Normal file
@@ -0,0 +1,47 @@
|
|||||||
|
server {
|
||||||
|
listen 80;
|
||||||
|
server_name www.tasteby.net tasteby.net;
|
||||||
|
return 301 https://www.tasteby.net$request_uri;
|
||||||
|
}
|
||||||
|
|
||||||
|
server {
|
||||||
|
listen 443 ssl http2;
|
||||||
|
server_name tasteby.net;
|
||||||
|
|
||||||
|
ssl_certificate /etc/letsencrypt/live/www.tasteby.net/fullchain.pem;
|
||||||
|
ssl_certificate_key /etc/letsencrypt/live/www.tasteby.net/privkey.pem;
|
||||||
|
|
||||||
|
return 301 https://www.tasteby.net$request_uri;
|
||||||
|
}
|
||||||
|
|
||||||
|
server {
|
||||||
|
listen 443 ssl http2;
|
||||||
|
server_name www.tasteby.net;
|
||||||
|
|
||||||
|
ssl_certificate /etc/letsencrypt/live/www.tasteby.net/fullchain.pem;
|
||||||
|
ssl_certificate_key /etc/letsencrypt/live/www.tasteby.net/privkey.pem;
|
||||||
|
ssl_protocols TLSv1.2 TLSv1.3;
|
||||||
|
ssl_ciphers HIGH:!aNULL:!MD5;
|
||||||
|
|
||||||
|
# Frontend (Next.js)
|
||||||
|
location / {
|
||||||
|
proxy_pass http://127.0.0.1:3001;
|
||||||
|
proxy_http_version 1.1;
|
||||||
|
proxy_set_header Host $host;
|
||||||
|
proxy_set_header X-Real-IP $remote_addr;
|
||||||
|
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
|
||||||
|
proxy_set_header X-Forwarded-Proto $scheme;
|
||||||
|
proxy_set_header Upgrade $http_upgrade;
|
||||||
|
proxy_set_header Connection "upgrade";
|
||||||
|
}
|
||||||
|
|
||||||
|
# Backend API (FastAPI)
|
||||||
|
location /api/ {
|
||||||
|
proxy_pass http://127.0.0.1:8000;
|
||||||
|
proxy_http_version 1.1;
|
||||||
|
proxy_set_header Host $host;
|
||||||
|
proxy_set_header X-Real-IP $remote_addr;
|
||||||
|
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
|
||||||
|
proxy_set_header X-Forwarded-Proto $scheme;
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user