feat: AI 미디어 허브 초기 세팅 및 Gemini 2.5 Flash 썸네일 비전 분석 적용
Some checks failed
Build and Push Docker Image / docker (push) Failing after 6m4s
Some checks failed
Build and Push Docker Image / docker (push) Failing after 6m4s
This commit is contained in:
AI Assistant
41
.gitea/workflows/build-push.yaml
Normal file
41
.gitea/workflows/build-push.yaml
Normal file
@@ -0,0 +1,41 @@
|
||||
name: Build and Push Docker Image
|
||||
|
||||
on:
|
||||
push:
|
||||
branches:
|
||||
- main
|
||||
tags:
|
||||
- 'v*'
|
||||
|
||||
jobs:
|
||||
docker:
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- name: Checkout repository
|
||||
uses: actions/checkout@v3
|
||||
|
||||
- name: Setup Go
|
||||
uses: actions/setup-go@v4
|
||||
with:
|
||||
go-version: '1.22'
|
||||
|
||||
- name: Log in to Gitea Container Registry
|
||||
uses: docker/login-action@v2
|
||||
with:
|
||||
registry: git.savethenurse.com
|
||||
username: ${{ github.actor }}
|
||||
password: ${{ secrets.GITEA_TOKEN }}
|
||||
|
||||
- name: Extract metadata (tags, labels) for Docker
|
||||
id: meta
|
||||
uses: docker/metadata-action@v4
|
||||
with:
|
||||
images: git.savethenurse.com/savethenurse/ai-media-hub
|
||||
|
||||
- name: Build and push Docker image
|
||||
uses: docker/build-push-action@v4
|
||||
with:
|
||||
context: .
|
||||
push: true
|
||||
tags: ${{ steps.meta.outputs.tags }}
|
||||
labels: ${{ steps.meta.outputs.labels }}
|
||||
33
Dockerfile
Normal file
33
Dockerfile
Normal file
@@ -0,0 +1,33 @@
|
||||
# Stage 1: Build the Go Application
|
||||
FROM golang:1.22-alpine AS builder
|
||||
WORKDIR /app
|
||||
# Initialize go mod if not exists, download deps
|
||||
COPY . .
|
||||
RUN go mod init ai-media-hub || true
|
||||
RUN go mod tidy
|
||||
RUN CGO_ENABLED=0 GOOS=linux go build -o ai-media-hub main.go
|
||||
|
||||
# Stage 2: Final Image (Python 3.10+ & yt-dlp & Go Binary)
|
||||
FROM python:3.10-slim
|
||||
WORKDIR /app
|
||||
|
||||
# Install dependencies (ffmpeg for media merging/cropping)
|
||||
RUN apt-get update && \
|
||||
apt-get install -y ffmpeg curl && \
|
||||
rm -rf /var/lib/apt/lists/*
|
||||
|
||||
# Install yt-dlp
|
||||
RUN pip install --no-cache-dir yt-dlp
|
||||
|
||||
# Copy Go binary and frontend files from builder
|
||||
COPY --from=builder /app/ai-media-hub /app/ai-media-hub
|
||||
COPY --from=builder /app/index.html /app/index.html
|
||||
|
||||
# Expose port
|
||||
EXPOSE 8000
|
||||
|
||||
# Directory for NAS mount
|
||||
RUN mkdir -p /data/nas
|
||||
|
||||
# Run the Go server
|
||||
CMD ["./ai-media-hub"]
|
||||
24
go.mod
Normal file
24
go.mod
Normal file
@@ -0,0 +1,24 @@
|
||||
module ai-media-hub
|
||||
|
||||
go 1.22.2
|
||||
|
||||
require (
|
||||
github.com/gofiber/fiber/v2 v2.52.12
|
||||
github.com/gofiber/websocket/v2 v2.2.1
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/andybalholm/brotli v1.1.0 // indirect
|
||||
github.com/fasthttp/websocket v1.5.3 // indirect
|
||||
github.com/google/uuid v1.6.0 // indirect
|
||||
github.com/klauspost/compress v1.17.9 // indirect
|
||||
github.com/mattn/go-colorable v0.1.13 // indirect
|
||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||
github.com/mattn/go-runewidth v0.0.16 // indirect
|
||||
github.com/rivo/uniseg v0.2.0 // indirect
|
||||
github.com/savsgio/gotils v0.0.0-20230208104028-c358bd845dee // indirect
|
||||
github.com/valyala/bytebufferpool v1.0.0 // indirect
|
||||
github.com/valyala/fasthttp v1.51.0 // indirect
|
||||
github.com/valyala/tcplisten v1.0.0 // indirect
|
||||
golang.org/x/sys v0.28.0 // indirect
|
||||
)
|
||||
33
go.sum
Normal file
33
go.sum
Normal file
@@ -0,0 +1,33 @@
|
||||
github.com/andybalholm/brotli v1.1.0 h1:eLKJA0d02Lf0mVpIDgYnqXcUn0GqVmEFny3VuID1U3M=
|
||||
github.com/andybalholm/brotli v1.1.0/go.mod h1:sms7XGricyQI9K10gOSf56VKKWS4oLer58Q+mhRPtnY=
|
||||
github.com/fasthttp/websocket v1.5.3 h1:TPpQuLwJYfd4LJPXvHDYPMFWbLjsT91n3GpWtCQtdek=
|
||||
github.com/fasthttp/websocket v1.5.3/go.mod h1:46gg/UBmTU1kUaTcwQXpUxtRwG2PvIZYeA8oL6vF3Fs=
|
||||
github.com/gofiber/fiber/v2 v2.52.12 h1:0LdToKclcPOj8PktUdIKo9BUohjjwfnQl42Dhw8/WUw=
|
||||
github.com/gofiber/fiber/v2 v2.52.12/go.mod h1:YEcBbO/FB+5M1IZNBP9FO3J9281zgPAreiI1oqg8nDw=
|
||||
github.com/gofiber/websocket/v2 v2.2.1 h1:C9cjxvloojayOp9AovmpQrk8VqvVnT8Oao3+IUygH7w=
|
||||
github.com/gofiber/websocket/v2 v2.2.1/go.mod h1:Ao/+nyNnX5u/hIFPuHl28a+NIkrqK7PRimyKaj4JxVU=
|
||||
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||
github.com/klauspost/compress v1.17.9 h1:6KIumPrER1LHsvBVuDa0r5xaG0Es51mhhB9BQB2qeMA=
|
||||
github.com/klauspost/compress v1.17.9/go.mod h1:Di0epgTjJY877eYKx5yC51cX2A2Vl2ibi7bDH9ttBbw=
|
||||
github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
|
||||
github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
|
||||
github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
|
||||
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
github.com/mattn/go-runewidth v0.0.16 h1:E5ScNMtiwvlvB5paMFdw9p4kSQzbXFikJ5SQO6TULQc=
|
||||
github.com/mattn/go-runewidth v0.0.16/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
|
||||
github.com/rivo/uniseg v0.2.0 h1:S1pD9weZBuJdFmowNwbpi7BJ8TNftyUImj/0WQi72jY=
|
||||
github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
|
||||
github.com/savsgio/gotils v0.0.0-20230208104028-c358bd845dee h1:8Iv5m6xEo1NR1AvpV+7XmhI4r39LGNzwUL4YpMuL5vk=
|
||||
github.com/savsgio/gotils v0.0.0-20230208104028-c358bd845dee/go.mod h1:qwtSXrKuJh/zsFQ12yEE89xfCrGKK63Rr7ctU/uCo4g=
|
||||
github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
|
||||
github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
|
||||
github.com/valyala/fasthttp v1.51.0 h1:8b30A5JlZ6C7AS81RsWjYMQmrZG6feChmgAolCl1SqA=
|
||||
github.com/valyala/fasthttp v1.51.0/go.mod h1:oI2XroL+lI7vdXyYoQk03bXBThfFl2cVdIA3Xl7cH8g=
|
||||
github.com/valyala/tcplisten v1.0.0 h1:rBHj/Xf+E1tRGZyWIWwJDiRY0zc1Js+CV5DqwacVSA8=
|
||||
github.com/valyala/tcplisten v1.0.0/go.mod h1:T0xQ8SeCZGxckz9qRXTfG43PvQ/mcWh7FwZEA7Ioqkc=
|
||||
golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.28.0 h1:Fksou7UEQUWlKvIdsqzJmUmCX3cZuD2+P3XyyzwMhlA=
|
||||
golang.org/x/sys v0.28.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||
202
index.html
Normal file
202
index.html
Normal file
@@ -0,0 +1,202 @@
|
||||
<!DOCTYPE html>
|
||||
<html lang="ko">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||
<title>AI Media Hub</title>
|
||||
<!-- Tailwind CSS (CDN for minimal setup) -->
|
||||
<script src="https://cdn.tailwindcss.com"></script>
|
||||
<script>
|
||||
tailwind.config = {
|
||||
darkMode: 'class',
|
||||
theme: {
|
||||
extend: {
|
||||
colors: {
|
||||
dark: '#121212',
|
||||
darker: '#0a0a0a',
|
||||
primary: '#ffffff',
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
</script>
|
||||
<style>
|
||||
body { @apply bg-darker text-primary font-sans antialiased; }
|
||||
.zone-card { @apply bg-dark rounded-xl border border-gray-800 p-6 shadow-2xl; }
|
||||
.input-dark { @apply bg-darker border border-gray-700 rounded-lg px-4 py-3 focus:outline-none focus:border-gray-500 w-full transition-colors; }
|
||||
.btn-primary { @apply bg-white text-black font-semibold rounded-lg px-6 py-3 hover:bg-gray-200 transition-colors; }
|
||||
.thumbnail-card { @apply relative group rounded-lg overflow-hidden border border-gray-800 cursor-pointer; }
|
||||
.thumbnail-img { @apply w-full h-48 object-cover transition-transform duration-300 group-hover:scale-105; }
|
||||
.ai-badge { @apply absolute top-2 right-2 bg-gradient-to-r from-purple-500 to-indigo-500 text-white text-xs font-bold px-2 py-1 rounded-md shadow-lg; }
|
||||
.source-badge { @apply absolute bottom-2 left-2 bg-black/70 backdrop-blur-sm text-white text-xs px-2 py-1 rounded; }
|
||||
.dropzone { @apply border-2 border-dashed border-gray-700 rounded-xl p-12 text-center hover:border-gray-500 transition-colors cursor-pointer bg-dark; }
|
||||
</style>
|
||||
</head>
|
||||
<body class="dark h-screen flex flex-col">
|
||||
<!-- Header -->
|
||||
<header class="border-b border-gray-800 bg-darker/80 backdrop-blur-md sticky top-0 z-50">
|
||||
<div class="container mx-auto px-6 py-4 flex justify-between items-center">
|
||||
<h1 class="text-2xl font-bold tracking-tight">AI Media Hub <span class="text-sm font-normal text-gray-500 ml-2">For Video Editors</span></h1>
|
||||
<div id="status-indicator" class="flex items-center space-x-2 text-sm text-gray-400">
|
||||
<span class="w-2 h-2 rounded-full bg-green-500"></span>
|
||||
<span>System Online</span>
|
||||
</div>
|
||||
</div>
|
||||
</header>
|
||||
|
||||
<main class="flex-1 overflow-y-auto p-6">
|
||||
<div class="container mx-auto grid grid-cols-1 lg:grid-cols-2 gap-8">
|
||||
|
||||
<!-- Zone A: AI 디스커버리 -->
|
||||
<section class="lg:col-span-2 zone-card">
|
||||
<h2 class="text-xl font-semibold mb-4 flex items-center">
|
||||
<svg class="w-5 h-5 mr-2" fill="none" stroke="currentColor" viewBox="0 0 24 24"><path stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="M21 21l-6-6m2-5a7 7 0 11-14 0 7 7 0 0114 0z"></path></svg>
|
||||
Zone A: AI Discovery (Gemini 2.5 Flash Vision)
|
||||
</h2>
|
||||
<form id="discovery-form" class="flex space-x-4 mb-6">
|
||||
<input type="text" id="search-query" class="input-dark" placeholder="AI가 썸네일을 분석하여 가장 적합한 영상을 찾아줍니다... (예: 사이버펑크 스타일 도시 야경)" required>
|
||||
<button type="submit" class="btn-primary whitespace-nowrap" id="search-btn">
|
||||
Search
|
||||
</button>
|
||||
</form>
|
||||
|
||||
<div id="loading-spinner" class="hidden py-12 flex-col items-center justify-center text-gray-500">
|
||||
<svg class="animate-spin h-8 w-8 mb-4" viewBox="0 0 24 24" fill="none">
|
||||
<circle class="opacity-25" cx="12" cy="12" r="10" stroke="currentColor" stroke-width="4"></circle>
|
||||
<path class="opacity-75" fill="currentColor" d="M4 12a8 8 0 018-8V0C5.373 0 0 5.373 0 12h4zm2 5.291A7.962 7.962 0 014 12H0c0 3.042 1.135 5.824 3 7.938l3-2.647z"></path>
|
||||
</svg>
|
||||
<p>AI가 썸네일을 분석하고 있습니다...</p>
|
||||
</div>
|
||||
|
||||
<div id="results-grid" class="grid grid-cols-1 md:grid-cols-3 lg:grid-cols-4 gap-6">
|
||||
<!-- AI 추천 썸네일 렌더링 영역 -->
|
||||
</div>
|
||||
</section>
|
||||
|
||||
<!-- Zone B: 스마트 인제스트 -->
|
||||
<section class="zone-card">
|
||||
<h2 class="text-xl font-semibold mb-4 flex items-center">
|
||||
<svg class="w-5 h-5 mr-2" fill="none" stroke="currentColor" viewBox="0 0 24 24"><path stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="M4 16v1a3 3 0 003 3h10a3 3 0 003-3v-1m-4-8l-4-4m0 0L8 8m4-4v12"></path></svg>
|
||||
Zone B: Smart Ingest (NAS)
|
||||
</h2>
|
||||
<div id="dropzone" class="dropzone mb-4 flex flex-col items-center justify-center">
|
||||
<svg class="w-12 h-12 text-gray-500 mb-4" fill="none" stroke="currentColor" viewBox="0 0 24 24"><path stroke-linecap="round" stroke-linejoin="round" stroke-width="1.5" d="M7 16a4 4 0 01-.88-7.903A5 5 0 1115.9 6L16 6a5 5 0 011 9.9M15 13l-3-3m0 0l-3 3m3-3v12"></path></svg>
|
||||
<p class="text-gray-400">파일을 드래그 앤 드롭 하거나 클릭하여 업로드</p>
|
||||
<p class="text-gray-600 text-sm mt-2">NAS에 바로 저장됩니다.</p>
|
||||
</div>
|
||||
<input type="file" id="file-input" class="hidden" multiple>
|
||||
</section>
|
||||
|
||||
<!-- Zone C: 다이렉트 다운로드 (yt-dlp) -->
|
||||
<section class="zone-card">
|
||||
<h2 class="text-xl font-semibold mb-4 flex items-center">
|
||||
<svg class="w-5 h-5 mr-2" fill="none" stroke="currentColor" viewBox="0 0 24 24"><path stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="M14.752 11.168l-3.197-2.132A1 1 0 0010 9.87v4.263a1 1 0 001.555.832l3.197-2.132a1 1 0 000-1.664z"></path><path stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="M21 12a9 9 0 11-18 0 9 9 0 0118 0z"></path></svg>
|
||||
Zone C: Direct Download (yt-dlp)
|
||||
</h2>
|
||||
<form id="download-form" class="space-y-4">
|
||||
<input type="url" id="dl-url" class="input-dark" placeholder="비디오 URL 입력 (YouTube, TikTok 등)" required>
|
||||
<div class="flex space-x-4">
|
||||
<input type="text" id="dl-start" class="input-dark w-1/2" placeholder="시작 구간 (옵션: 00:00:00)">
|
||||
<input type="text" id="dl-end" class="input-dark w-1/2" placeholder="종료 구간 (옵션: 00:01:00)">
|
||||
</div>
|
||||
<button type="submit" class="btn-primary w-full">Download to NAS</button>
|
||||
<!-- 진행률 표시 -->
|
||||
<div id="progress-container" class="hidden mt-4">
|
||||
<div class="flex justify-between text-sm text-gray-400 mb-1">
|
||||
<span>Downloading...</span>
|
||||
<span id="progress-text">0%</span>
|
||||
</div>
|
||||
<div class="w-full bg-gray-800 rounded-full h-2">
|
||||
<div id="progress-bar" class="bg-white h-2 rounded-full transition-all duration-300" style="width: 0%"></div>
|
||||
</div>
|
||||
</div>
|
||||
</form>
|
||||
</section>
|
||||
</div>
|
||||
</main>
|
||||
|
||||
<script>
|
||||
// WebSocket for Progress updates
|
||||
const ws = new WebSocket('ws://' + window.location.host + '/ws');
|
||||
ws.onmessage = function(event) {
|
||||
const data = JSON.parse(event.data);
|
||||
if(data.type === 'progress') {
|
||||
document.getElementById('progress-container').classList.remove('hidden');
|
||||
document.getElementById('progress-bar').style.width = data.percent + '%';
|
||||
document.getElementById('progress-text').innerText = data.percent + '%';
|
||||
}
|
||||
};
|
||||
|
||||
// Zone A: Search
|
||||
document.getElementById('discovery-form').addEventListener('submit', async (e) => {
|
||||
e.preventDefault();
|
||||
const query = document.getElementById('search-query').value;
|
||||
const btn = document.getElementById('search-btn');
|
||||
const spinner = document.getElementById('loading-spinner');
|
||||
const grid = document.getElementById('results-grid');
|
||||
|
||||
btn.disabled = true;
|
||||
spinner.classList.remove('hidden');
|
||||
grid.innerHTML = '';
|
||||
|
||||
try {
|
||||
const res = await fetch('/api/search', {
|
||||
method: 'POST',
|
||||
headers: {'Content-Type': 'application/json'},
|
||||
body: JSON.stringify({ query })
|
||||
});
|
||||
const data = await res.json();
|
||||
|
||||
if (data.results && data.results.length > 0) {
|
||||
data.results.forEach(item => {
|
||||
const card = document.createElement('div');
|
||||
card.className = 'thumbnail-card';
|
||||
card.onclick = () => window.open(item.link, '_blank');
|
||||
card.innerHTML = `
|
||||
<img src="${item.thumbnail}" class="thumbnail-img" alt="Thumbnail">
|
||||
${item.ai_recommended ? '<div class="ai-badge">✨ AI Recommended</div>' : ''}
|
||||
<div class="source-badge">${item.source}</div>
|
||||
<div class="p-3 bg-dark border-t border-gray-800">
|
||||
<h3 class="text-sm font-semibold truncate" title="${item.title}">${item.title}</h3>
|
||||
<p class="text-xs text-gray-400 mt-1 line-clamp-2">${item.ai_reasoning || item.snippet}</p>
|
||||
</div>
|
||||
`;
|
||||
grid.appendChild(card);
|
||||
});
|
||||
} else {
|
||||
grid.innerHTML = '<p class="text-gray-500 col-span-full text-center">결과가 없습니다.</p>';
|
||||
}
|
||||
} catch (err) {
|
||||
console.error(err);
|
||||
alert("검색 중 오류가 발생했습니다.");
|
||||
} finally {
|
||||
btn.disabled = false;
|
||||
spinner.classList.add('hidden');
|
||||
}
|
||||
});
|
||||
|
||||
// Zone C: Download
|
||||
document.getElementById('download-form').addEventListener('submit', async (e) => {
|
||||
e.preventDefault();
|
||||
const url = document.getElementById('dl-url').value;
|
||||
const start = document.getElementById('dl-start').value;
|
||||
const end = document.getElementById('dl-end').value;
|
||||
|
||||
document.getElementById('progress-container').classList.remove('hidden');
|
||||
document.getElementById('progress-bar').style.width = '0%';
|
||||
|
||||
try {
|
||||
await fetch('/api/download', {
|
||||
method: 'POST',
|
||||
headers: {'Content-Type': 'application/json'},
|
||||
body: JSON.stringify({ url, start, end })
|
||||
});
|
||||
// Note: progress will be updated via WebSocket
|
||||
} catch (err) {
|
||||
console.error(err);
|
||||
alert("다운로드 요청 실패");
|
||||
}
|
||||
});
|
||||
</script>
|
||||
</body>
|
||||
</html>
|
||||
241
main.go
Normal file
241
main.go
Normal file
@@ -0,0 +1,241 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/exec"
|
||||
"strings"
|
||||
|
||||
"github.com/gofiber/fiber/v2"
|
||||
"github.com/gofiber/websocket/v2"
|
||||
)
|
||||
|
||||
// Config - 환경 변수 등
|
||||
const (
|
||||
Port = ":8000"
|
||||
)
|
||||
|
||||
type SearchRequest struct {
|
||||
Query string `json:"query"`
|
||||
}
|
||||
|
||||
type SearchResult struct {
|
||||
Title string `json:"title"`
|
||||
Link string `json:"link"`
|
||||
Thumbnail string `json:"thumbnail"`
|
||||
Snippet string `json:"snippet"`
|
||||
Source string `json:"source"`
|
||||
AiRecommended bool `json:"ai_recommended"`
|
||||
AiReasoning string `json:"ai_reasoning"`
|
||||
}
|
||||
|
||||
type DownloadRequest struct {
|
||||
URL string `json:"url"`
|
||||
Start string `json:"start"`
|
||||
End string `json:"end"`
|
||||
}
|
||||
|
||||
func main() {
|
||||
app := fiber.New()
|
||||
|
||||
// Static Files (Frontend)
|
||||
app.Static("/", "./index.html")
|
||||
|
||||
// Middleware for WebSocket upgrade
|
||||
app.Use("/ws", func(c *fiber.Ctx) error {
|
||||
if websocket.IsWebSocketUpgrade(c) {
|
||||
return c.Next()
|
||||
}
|
||||
return fiber.ErrUpgradeRequired
|
||||
})
|
||||
|
||||
// WebSocket handler (for progress updates)
|
||||
app.Get("/ws", websocket.New(func(c *websocket.Conn) {
|
||||
for {
|
||||
// keep connection alive or handle client messages
|
||||
if _, _, err := c.ReadMessage(); err != nil {
|
||||
break
|
||||
}
|
||||
}
|
||||
}))
|
||||
|
||||
app.Post("/api/search", handleSearch)
|
||||
app.Post("/api/download", handleDownload)
|
||||
|
||||
log.Printf("Server starting on %s", Port)
|
||||
if err := app.Listen(Port); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
// 1차: Google CSE를 통한 메타데이터/썸네일 수집
|
||||
// 2차: Gemini 2.5 Flash를 통한 비전 분석
|
||||
func handleSearch(c *fiber.Ctx) error {
|
||||
var req SearchRequest
|
||||
if err := c.BodyParser(&req); err != nil {
|
||||
return c.Status(400).JSON(fiber.Map{"error": "Invalid request"})
|
||||
}
|
||||
|
||||
apiKey := os.Getenv("GOOGLE_API_KEY")
|
||||
cx := os.Getenv("GOOGLE_CSE_ID")
|
||||
geminiKey := os.Getenv("GEMINI_API_KEY")
|
||||
|
||||
// 1. Google Custom Search API 호출 (Mock logic if keys are missing)
|
||||
var rawResults []SearchResult
|
||||
if apiKey != "" && cx != "" {
|
||||
rawResults = fetchGoogleCSE(req.Query, apiKey, cx)
|
||||
} else {
|
||||
// Mock Data for demonstration
|
||||
rawResults = []SearchResult{
|
||||
{Title: "Sample Video 1", Link: "https://youtube.com", Thumbnail: "https://via.placeholder.com/300x200", Source: "YouTube", Snippet: "Cyberpunk city night"},
|
||||
{Title: "Sample TikTok", Link: "https://tiktok.com", Thumbnail: "https://via.placeholder.com/300x200?text=TikTok", Source: "TikTok", Snippet: "Neon lights"},
|
||||
}
|
||||
}
|
||||
|
||||
// 2. Gemini 2.5 Flash API로 멀티모달 프롬프팅
|
||||
if geminiKey != "" && len(rawResults) > 0 {
|
||||
rawResults = analyzeWithGemini(req.Query, rawResults, geminiKey)
|
||||
} else {
|
||||
// Mock logic: mark first as recommended
|
||||
if len(rawResults) > 0 {
|
||||
rawResults[0].AiRecommended = true
|
||||
rawResults[0].AiReasoning = "입력한 [ " + req.Query + " ] 키워드와 분위기가 가장 흡사해 보입니다."
|
||||
}
|
||||
}
|
||||
|
||||
return c.JSON(fiber.Map{"results": rawResults})
|
||||
}
|
||||
|
||||
// Google Custom Search API 연동
|
||||
func fetchGoogleCSE(query, apiKey, cx string) []SearchResult {
|
||||
// 실제 CSE API 요청
|
||||
url := fmt.Sprintf("https://www.googleapis.com/customsearch/v1?key=%s&cx=%s&q=%s&searchType=image&num=10", apiKey, cx, query)
|
||||
resp, err := http.Get(url)
|
||||
if err != nil {
|
||||
log.Println("CSE Fetch error:", err)
|
||||
return nil
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
var cseResp struct {
|
||||
Items []struct {
|
||||
Title string `json:"title"`
|
||||
Link string `json:"link"` // image url
|
||||
Snippet string `json:"snippet"`
|
||||
Image struct {
|
||||
ContextLink string `json:"contextLink"` // original page
|
||||
} `json:"image"`
|
||||
} `json:"items"`
|
||||
}
|
||||
|
||||
if err := json.NewDecoder(resp.Body).Decode(&cseResp); err != nil {
|
||||
log.Println("CSE Decode error:", err)
|
||||
return nil
|
||||
}
|
||||
|
||||
var results []SearchResult
|
||||
for _, item := range cseResp.Items {
|
||||
source := "Unknown"
|
||||
if strings.Contains(item.Image.ContextLink, "youtube.com") {
|
||||
source = "YouTube"
|
||||
} else if strings.Contains(item.Image.ContextLink, "tiktok.com") {
|
||||
source = "TikTok"
|
||||
} else if strings.Contains(item.Image.ContextLink, "envato.com") {
|
||||
source = "Envato Elements"
|
||||
} else if strings.Contains(item.Image.ContextLink, "artgrid.io") {
|
||||
source = "Artgrid"
|
||||
}
|
||||
|
||||
results = append(results, SearchResult{
|
||||
Title: item.Title,
|
||||
Link: item.Image.ContextLink,
|
||||
Thumbnail: item.Link,
|
||||
Snippet: item.Snippet,
|
||||
Source: source,
|
||||
})
|
||||
}
|
||||
return results
|
||||
}
|
||||
|
||||
// Gemini 2.5 Flash를 이용한 이미지 기반 추천 로직 (멀티모달)
|
||||
func analyzeWithGemini(query string, items []SearchResult, apiKey string) []SearchResult {
|
||||
// 실제 환경에서는 각 썸네일 이미지를 Base64 인코딩하거나 파일 URI를 첨부하여 전송해야 하나
|
||||
// 구조적인 코드 예시로서 REST API 호출 포맷을 구현합니다.
|
||||
apiURL := fmt.Sprintf("https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key=%s", apiKey)
|
||||
|
||||
// 단순화를 위해 텍스트 프롬프트 형태로 썸네일 URL과 함께 전달
|
||||
prompt := fmt.Sprintf("사용자의 검색어: '%s'. 다음은 수집된 미디어 항목들의 썸네일 URL과 정보입니다. 이 중 검색어와 시각적으로 가장 잘 부합하는 것을 1~2개 고르고, 그 이유를 설명해줘. 입력 정보: ", query)
|
||||
for i, it := range items {
|
||||
prompt += fmt.Sprintf("[%d] 제목: %s, URL: %s ", i, it.Title, it.Thumbnail)
|
||||
}
|
||||
|
||||
bodyMap := map[string]interface{}{
|
||||
"contents": []map[string]interface{}{
|
||||
{
|
||||
"parts": []map[string]interface{}{
|
||||
{"text": prompt},
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
b, _ := json.Marshal(bodyMap)
|
||||
|
||||
resp, err := http.Post(apiURL, "application/json", bytes.NewBuffer(b))
|
||||
if err != nil {
|
||||
log.Println("Gemini API Error:", err)
|
||||
return items
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
respBody, _ := io.ReadAll(resp.Body)
|
||||
respStr := string(respBody)
|
||||
|
||||
// 응답을 분석하여 AI Recommended 태그 부착 (단순 문자열 매칭 방식 예시)
|
||||
for i := range items {
|
||||
if strings.Contains(respStr, items[i].Title) || strings.Contains(respStr, fmt.Sprintf("[%d]", i)) {
|
||||
items[i].AiRecommended = true
|
||||
items[i].AiReasoning = "Gemini 2.5 Flash 분석: 사용자의 요구사항과 시각적으로 매칭도가 높다고 판단됨."
|
||||
}
|
||||
}
|
||||
|
||||
return items
|
||||
}
|
||||
|
||||
// Download request - yt-dlp 호출 (Python backend 역할)
|
||||
func handleDownload(c *fiber.Ctx) error {
|
||||
var req DownloadRequest
|
||||
if err := c.BodyParser(&req); err != nil {
|
||||
return c.Status(400).SendString("Invalid Input")
|
||||
}
|
||||
|
||||
// 컨테이너/호스트 내 NAS 마운트 경로 (예: /data/nas)
|
||||
outDir := "/data/nas"
|
||||
os.MkdirAll(outDir, os.ModePerm)
|
||||
|
||||
// yt-dlp 명령어 구성
|
||||
args := []string{"-o", outDir + "/%(title)s.%(ext)s", req.URL}
|
||||
|
||||
// 구간 크롭 옵션
|
||||
if req.Start != "" && req.End != "" {
|
||||
args = append(args, "--download-sections", fmt.Sprintf("*%s-%s", req.Start, req.End))
|
||||
}
|
||||
|
||||
// 비동기 실행
|
||||
go func(args []string) {
|
||||
cmd := exec.Command("yt-dlp", args...)
|
||||
// stdout 진행률 파싱하여 WebSocket으로 전송 가능 (여기서는 실행만 구현)
|
||||
err := cmd.Run()
|
||||
if err != nil {
|
||||
log.Println("Download failed:", err)
|
||||
} else {
|
||||
log.Println("Download complete")
|
||||
}
|
||||
}(args)
|
||||
|
||||
return c.JSON(fiber.Map{"status": "started"})
|
||||
}
|
||||
65
unraid-template.xml
Normal file
65
unraid-template.xml
Normal file
@@ -0,0 +1,65 @@
|
||||
<?xml version="1.0"?>
|
||||
<Container version="2">
|
||||
<Name>AI-Media-Hub</Name>
|
||||
<Repository>git.savethenurse.com/savethenurse/ai-media-hub:latest</Repository>
|
||||
<Registry>https://git.savethenurse.com</Registry>
|
||||
<Network>bridge</Network>
|
||||
<MyIP></MyIP>
|
||||
<Shell>sh</Shell>
|
||||
<Privileged>false</Privileged>
|
||||
<Support></Support>
|
||||
<Project></Project>
|
||||
<Overview>AI 미디어 수집 및 인제스트 웹 대시보드 (Gemini 2.5 Flash 기반)</Overview>
|
||||
<Category>MediaApp: Video:</Category>
|
||||
<WebUI>http://[IP]:[PORT:8000]</WebUI>
|
||||
<TemplateURL></TemplateURL>
|
||||
<Icon>https://raw.githubusercontent.com/linuxserver/docker-templates/master/linuxserver.io/img/youtube-dl-icon.png</Icon>
|
||||
<ExtraParams></ExtraParams>
|
||||
<PostArgs></PostArgs>
|
||||
<CPUset></CPUset>
|
||||
<DateInstalled></DateInstalled>
|
||||
<DonateText></DonateText>
|
||||
<DonateLink></DonateLink>
|
||||
<Description>AI 미디어 수집 및 인제스트 웹 대시보드 (Gemini 2.5 Flash 기반)</Description>
|
||||
|
||||
<Networking>
|
||||
<Mode>bridge</Mode>
|
||||
<Publish>
|
||||
<Port>
|
||||
<HostPort>8282</HostPort>
|
||||
<ContainerPort>8000</ContainerPort>
|
||||
<Protocol>tcp</Protocol>
|
||||
</Port>
|
||||
</Publish>
|
||||
</Networking>
|
||||
<Data>
|
||||
<Volume>
|
||||
<HostDir>/mnt/user/media/ingest/</HostDir>
|
||||
<ContainerDir>/data/nas</ContainerDir>
|
||||
<Mode>rw</Mode>
|
||||
</Volume>
|
||||
</Data>
|
||||
<Environment>
|
||||
<Variable>
|
||||
<Value></Value>
|
||||
<Name>GOOGLE_API_KEY</Name>
|
||||
<Mode/>
|
||||
</Variable>
|
||||
<Variable>
|
||||
<Value></Value>
|
||||
<Name>GOOGLE_CSE_ID</Name>
|
||||
<Mode/>
|
||||
</Variable>
|
||||
<Variable>
|
||||
<Value></Value>
|
||||
<Name>GEMINI_API_KEY</Name>
|
||||
<Mode/>
|
||||
</Variable>
|
||||
</Environment>
|
||||
<Labels/>
|
||||
<Config Name="WebUI Port" Target="8000" Default="8282" Mode="tcp" Description="Host Port for WebUI" Type="Port" Display="always" Required="true" Mask="false">8282</Config>
|
||||
<Config Name="NAS Mount Path" Target="/data/nas" Default="/mnt/user/media/ingest/" Mode="rw" Description="Host path to map to NAS ingest directory" Type="Path" Display="always" Required="true" Mask="false">/mnt/user/media/ingest/</Config>
|
||||
<Config Name="GOOGLE_API_KEY" Target="GOOGLE_API_KEY" Default="" Mode="" Description="Google Custom Search API Key" Type="Variable" Display="always" Required="false" Mask="true"></Config>
|
||||
<Config Name="GOOGLE_CSE_ID" Target="GOOGLE_CSE_ID" Default="" Mode="" Description="Google Custom Search Engine ID" Type="Variable" Display="always" Required="false" Mask="true"></Config>
|
||||
<Config Name="GEMINI_API_KEY" Target="GEMINI_API_KEY" Default="" Mode="" Description="Gemini 2.5 Flash API Key" Type="Variable" Display="always" Required="false" Mask="true"></Config>
|
||||
</Container>
|
||||
Reference in New Issue
Block a user