File size: 5,695 Bytes
7e2816d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 |
# app.py - VERSÃO COMPLETA COM VOZ (BASE64) E VISÃO
import os
import base64
import io
import asyncio
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
from pydantic import BaseModel
from PIL import Image
from jade.core import JadeAgent
from jade.scholar import ScholarAgent
from jade.heavy_mode import JadeHeavyAgent
print("Iniciando a J.A.D.E. com FastAPI...")
jade_agent = JadeAgent()
scholar_agent = ScholarAgent()
# Instantiate Heavy Agent. It uses environment variables.
jade_heavy_agent = JadeHeavyAgent()
print("J.A.D.E. pronta para receber requisições.")
app = FastAPI(title="J.A.D.E. API")
app.add_middleware(
CORSMiddleware,
allow_origins=["*"], allow_credentials=True, allow_methods=["*"], allow_headers=["*"],
)
# Mount generated directory for static files (PDFs, Images)
os.makedirs("backend/generated", exist_ok=True)
app.mount("/generated", StaticFiles(directory="backend/generated"), name="generated")
# Dicionário global para armazenar sessões de usuários
# Structure: user_sessions[user_id] = { "jade": [...], "scholar": [...], "heavy": [...] }
user_sessions = {}
class UserRequest(BaseModel):
user_input: str
image_base64: str | None = None
user_id: str | None = None
agent_type: str = "jade" # "jade", "scholar", "heavy"
web_search: bool = False # Toggle para busca web na J.A.D.E.
@app.post("/chat")
async def handle_chat(request: UserRequest):
try:
user_id = request.user_id if request.user_id else "default_user"
agent_type = request.agent_type.lower()
if user_id not in user_sessions:
print(f"Nova sessão criada para: {user_id}")
user_sessions[user_id] = {
"jade": [jade_agent.system_prompt],
"scholar": [],
"heavy": []
}
# Ensure sub-keys exist
if "jade" not in user_sessions[user_id]: user_sessions[user_id]["jade"] = [jade_agent.system_prompt]
if "scholar" not in user_sessions[user_id]: user_sessions[user_id]["scholar"] = []
if "heavy" not in user_sessions[user_id]: user_sessions[user_id]["heavy"] = []
vision_context = None
if request.image_base64:
try:
header, encoded_data = request.image_base64.split(",", 1)
image_bytes = base64.b64decode(encoded_data)
pil_image = Image.open(io.BytesIO(image_bytes))
# Jade handles vision processing
vision_context = jade_agent.image_handler.process_pil_image(pil_image)
except Exception as img_e:
print(f"Erro ao processar imagem Base64: {img_e}")
vision_context = "Houve um erro ao analisar a imagem."
final_user_input = request.user_input if request.user_input else "Descreva a imagem em detalhes."
bot_response_text = ""
audio_path = None
if agent_type == "scholar":
current_history = user_sessions[user_id]["scholar"]
bot_response_text, audio_path, updated_history = scholar_agent.respond(
history=current_history,
user_input=final_user_input,
user_id=user_id,
vision_context=vision_context
)
user_sessions[user_id]["scholar"] = updated_history
elif agent_type == "heavy":
current_history = user_sessions[user_id]["heavy"]
# Heavy agent is async
bot_response_text, audio_path, updated_history = await jade_heavy_agent.respond(
history=current_history,
user_input=final_user_input,
user_id=user_id,
vision_context=vision_context
)
user_sessions[user_id]["heavy"] = updated_history
else:
# Default to J.A.D.E.
current_history = user_sessions[user_id]["jade"]
# Jade agent is synchronous, run directly
bot_response_text, audio_path, updated_history = jade_agent.respond(
history=current_history,
user_input=final_user_input,
user_id=user_id,
vision_context=vision_context,
web_search=request.web_search # Passa o toggle de busca web
)
user_sessions[user_id]["jade"] = updated_history
# Audio Logic
audio_base64 = None
if audio_path and os.path.exists(audio_path):
print(f"Codificando arquivo de áudio: {audio_path}")
with open(audio_path, "rb") as audio_file:
audio_bytes = audio_file.read()
audio_base64 = base64.b64encode(audio_bytes).decode('utf-8')
# Remove only if temp file
if "backend/generated" not in audio_path:
os.remove(audio_path)
return {
"success": True,
"bot_response": bot_response_text,
"audio_base64": audio_base64
}
except Exception as e:
print(f"Erro crítico no endpoint /chat: {e}")
return {"success": False, "error": str(e)}
@app.get("/")
def root():
return {"message": "Servidor J.A.D.E. com FastAPI está online."}
if __name__ == "__main__":
import uvicorn
port = int(os.environ.get("PORT", 7860))
print(f"Iniciando o servidor Uvicorn em http://0.0.0.0:{port}")
uvicorn.run(app, host="0.0.0.0", port=port)
|