Delete main.py

main.py

@@ -1,239 +0,0 @@
-import os
-import threading
-from pathlib import Path
-from typing import Optional
-from fastapi import FastAPI, Request
-
-from fastapi import FastAPI, Query, Response
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import FileResponse, HTMLResponse, RedirectResponse
-from pydantic import BaseModel
-
-from .agent import ShoppingAgent
-from .feedback import caminho_feedback, google_sheets_habilitado, salvar_feedback
-from .google_oauth import build_flow, get_authorization_url, load_credentials, save_credentials
-from .logger import salvar_log_busca
-from .memory import caminho_memoria_negativa
-
-EMBEDDING_PROVIDER = os.getenv("EMBEDDING_PROVIDER", "transformers").strip().lower()
-HF_MODEL_REPO = os.getenv("HF_MODEL_REPO", "Ana2012/bertimbau-buscador").strip()
-
-
-def _env_flag(name, default="true"):
-    return os.getenv(name, default).strip().lower() in {"1", "true", "yes", "on"}
-
-
-PRELOAD_AGENT = _env_flag("PRELOAD_AGENT", "true")
-LOGS_DIR = os.getenv("LOGS_DIR", "/data/logs")
-DATA_DIR = "/data"
-
-app = FastAPI(title="TCC2 Agent API")
-
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=False,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-agent = None
-agent_lock = threading.Lock()
-
-
-def get_agent():
-    global agent
-    if agent is None:
-        with agent_lock:
-            if agent is None:
-                agent = ShoppingAgent()
-    return agent
-
-
-@app.on_event("startup")
-def preload_agent():
-    if PRELOAD_AGENT:
-        get_agent()
-
-
-class ChatRequest(BaseModel):
-    query: Optional[str] = None
-    message: Optional[str] = None
-    top_k: int = 5
-
-
-class FeedbackRequest(BaseModel):
-    query: str
-    product_id: str
-    product_name: str
-    rating: Optional[int] = None
-    is_helpful: Optional[bool] = None
-    note: Optional[str] = None
-    categoria_inferida: Optional[str] = None
-    feedback: Optional[str] = None
-    motivo: Optional[str] = None
-    score_final: Optional[float] = None
-    score_semantico: Optional[float] = None
-    bonus_lexical: Optional[float] = None
-    penalidade_feedback: Optional[float] = None
-    user_message: Optional[str] = None
-
-
-@app.get("/health")
-def health():
-    runtime = get_agent().runtime_info() if agent is not None else None
-    return {
-        "status": "ok",
-        "agent_ready": agent is not None,
-        "embedding_provider": EMBEDDING_PROVIDER,
-        "model_repo": HF_MODEL_REPO,
-        "preload_agent": PRELOAD_AGENT,
-        "runtime": runtime,
-        "feedback_storage": "google_sheets" if google_sheets_habilitado() else "csv",
-    }
-
-
-@app.get("/", include_in_schema=False)
-def root():
-    return RedirectResponse(url="/docs")
-
-
-@app.get("/favicon.ico", include_in_schema=False)
-def favicon():
-    return Response(status_code=204)
-
-
-@app.get("/auth/google")
-def auth_google():
-    authorization_url, _state = get_authorization_url()
-    return RedirectResponse(url=authorization_url)
-
-
-@app.get("/oauth2callback")
-def oauth2callback(code: str = Query(...)):
-    flow = build_flow()
-    flow.fetch_token(code=code)
-    save_credentials(flow.credentials)
-    return HTMLResponse(
-        "<h3>Autorizacao concluida. O backend ja pode salvar feedbacks no Google Sheets.</h3>"
-    )
-
-
-@app.get("/auth/status")
-def auth_status():
-    credentials = load_credentials()
-    connected = credentials is not None and credentials.valid
-    return {
-        "google_sheets_connected": connected,
-        "message": (
-            "Google Sheets autorizado e pronto para uso."
-            if connected
-            else "Google Sheets ainda nao autorizado. Acesse /auth/google para conectar."
-        ),
-    }
-
-
-@app.get("/debug/files")
-def debug_files():
-    data_path = Path(DATA_DIR)
-    logs_path = Path(LOGS_DIR)
-    feedback_path = Path(caminho_feedback())
-    memory_path = Path(caminho_memoria_negativa())
-
-    return {
-        "data_exists": data_path.exists(),
-        "logs_exists": logs_path.exists(),
-        "feedback_exists": feedback_path.exists(),
-        "negative_memory_exists": memory_path.exists(),
-        "data_files": sorted(p.name for p in data_path.iterdir()) if data_path.exists() else [],
-        "logs_files": sorted(p.name for p in logs_path.iterdir()) if logs_path.exists() else [],
-        "feedback_file": str(feedback_path),
-        "negative_memory_file": str(memory_path),
-        "feedback_storage": "google_sheets" if google_sheets_habilitado() else "csv",
-    }
-
-
-@app.get("/debug/feedback")
-def debug_feedback():
-    feedback_path = Path(caminho_feedback())
-    if not feedback_path.exists():
-        return {"error": "arquivo nao existe"}
-
-    return {"conteudo": feedback_path.read_text(encoding="utf-8")}
-
-
-@app.get("/download/feedback")
-def download_feedback():
-    feedback_path = caminho_feedback()
-    if not os.path.exists(feedback_path):
-        return {"error": "arquivo nao existe"}
-
-    return FileResponse(feedback_path, filename="feedback.csv")
-
-
-@app.get("/debug/memory")
-def debug_memory():
-    memory_path = Path(caminho_memoria_negativa())
-    if not memory_path.exists():
-        return {"status": "missing", "file": str(memory_path)}
-
-    return {
-        "status": "ok",
-        "file": str(memory_path),
-        "content": memory_path.read_text(encoding="utf-8"),
-    }
-
-
-@app.post("/chat")
-def chat(request: ChatRequest):
-    texto = request.query or request.message
-
-    if not texto:
-        return {"error": "query ou message deve ser informado"}
-
-    resultado = get_agent().responder(texto, top_k=request.top_k)
-    salvar_log_busca(resultado)
-    return resultado
-
-
-@app.post("/feedback")
-def feedback(request: FeedbackRequest):
-    feedback_file = caminho_feedback()
-    print(
-        "Salvando feedback:",
-        {
-            "query": request.query,
-            "product_id": request.product_id,
-            "feedback_file": feedback_file,
-            "logs_dir_exists": os.path.exists(LOGS_DIR),
-            "google_sheets_enabled": google_sheets_habilitado(),
-        },
-    )
-
-    try:
-        return salvar_feedback(
-            query=request.query,
-            product_id=request.product_id,
-            product_name=request.product_name,
-            rating=request.rating,
-            is_helpful=request.is_helpful,
-            note=request.note,
-            categoria_inferida=request.categoria_inferida,
-            feedback=request.feedback,
-            motivo=request.motivo,
-            score_final=request.score_final,
-            score_semantico=request.score_semantico,
-            bonus_lexical=request.bonus_lexical,
-            penalidade_feedback=request.penalidade_feedback,
-            user_message=request.user_message,
-        )
-    except Exception as exc:
-        return {
-            "ok": False,
-            "saved_local": False,
-            "saved_google_sheets": False,
-            "detail": str(exc),
-            "feedback_file": feedback_file,
-            "logs_dir_exists": os.path.exists(LOGS_DIR),
-            "google_sheets_enabled": google_sheets_habilitado(),
-        }