Create main.py

main.py

@@ -0,0 +1,79 @@
+# main.py
+# FastAPI API dla tomasz-svd (Etap 5/6)
+
+from fastapi import FastAPI
+from pydantic import BaseModel
+from typing import Optional, List
+
+from core import (
+    analyze_domain,
+    generate_script,
+    generate_music,
+    convert_to_mp3,
+    generate_video_from_b64,
+)
+
+app = FastAPI(title="tomasz-svd API")
+
+class GenerateAdRequest(BaseModel):
+    domain: str
+    style: str = "energetyczny TikTok"
+    length: int = 15
+
+class GenerateAdResponse(BaseModel):
+    status: str
+    message: Optional[str] = None
+    video_path: Optional[str] = None
+
+@app.get("/health")
+def health():
+    return {"status": "ok"}
+
+@app.post("/analyze")
+def api_analyze(domain: str):
+    return analyze_domain(domain)
+
+@app.post("/script")
+def api_script(domain: str, brand_prompt: str, brand_text: str, length: int = 15, style: str = "energetyczny TikTok"):
+    return generate_script(brand_prompt, domain, brand_text, length, style)
+
+@app.post("/audio")
+def api_audio(prompt: str = "energetic modern ad music", duration: int = 15):
+    wav = generate_music(prompt, duration)
+    mp3 = convert_to_mp3(wav)
+    return {"status": "ok", "audio_path": mp3}
+
+class VideoRequest(BaseModel):
+    images_b64: List[str]
+    script_json: str
+    audio_path: str
+
+@app.post("/video")
+def api_video(req: VideoRequest):
+    video = generate_video_from_b64(req.images_b64, req.script_json, req.audio_path)
+    if not video:
+        return {"status": "error", "message": "Nie udało się wygenerować wideo."}
+    return {"status": "ok", "video_path": video}
+
+@app.post("/generate_ad", response_model=GenerateAdResponse)
+def api_generate_ad(req: GenerateAdRequest):
+    analyzed = analyze_domain(req.domain)
+    if "error" in analyzed:
+        return GenerateAdResponse(status="error", message=analyzed["error"])
+
+    script = generate_script(
+        analyzed["prompt"],
+        analyzed["domain"],
+        analyzed["text_snippet"],
+        req.length,
+        req.style
+    )
+
+    wav = generate_music("energetic modern ad music", req.length)
+    mp3 = convert_to_mp3(wav)
+
+    video = generate_video_from_b64(analyzed["images"], script, mp3)
+    if not video:
+        return GenerateAdResponse(status="error", message="Nie udało się wygenerować wideo.")
+
+    return GenerateAdResponse(status="ok", video_path=video)