app.py

import os
import base64
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
import uvicorn
from io import BytesIO

from models.semantic_analyzer import SemanticAnalyzer
from models.wordcloud_gen import WordCloudGenerator

app = FastAPI()

analyzer = SemanticAnalyzer()
wordcloud_gen = WordCloudGenerator()

class AnalyzeRequest(BaseModel):
    text: str
    topics: int = 5
    max_words: int = 100
    style: str = "light"

@app.post("/analyze")
async def analyze_text(request: AnalyzeRequest):
    try:
        topics, embeddings = analyzer.analyze(request.text, num_topics=request.topics)

        wordcloud_image = wordcloud_gen.generate_wordcloud(
            topics,
            max_words=request.max_words,
            style=request.style
        )

        buffered = BytesIO()
        wordcloud_image.save(buffered, format="PNG")
        img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")

        return {
            "topics": topics,
            "wordcloud": img_str,
            "embeddings": embeddings.tolist()
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))

@app.get("/health")
async def health_check():
    return {"status": "ok"}

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=8000)