File size: 1,207 Bytes
bdd2530
f68f72a
 
4cda30b
bf41ce7
bdd2530
27b5f6d
ce0cf0a
bdd2530
ce0cf0a
b7adcb8
 
 
bf41ce7
4cda30b
f68f72a
 
 
 
 
 
 
 
 
 
8996d88
 
 
 
8189d77
ce0cf0a
27b5f6d
bf41ce7
4cda30b
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
from Brain.src.firebase.firebase import initialize_app
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
import uvicorn

from Brain.src.router.browser_router import construct_blueprint_browser_api
from Brain.src.router.train_router import construct_blueprint_train_api
from Brain.src.router.email_router import construct_blueprint_email_api
from Brain.src.router.api import construct_blueprint_api

import gradio as gr
import random
import time

app = FastAPI()
# Set up CORS middleware
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],  # Allow all origins (domains)
    allow_credentials=True,
    allow_methods=["*"],  # Allow all methods (GET, POST, PUT, DELETE, OPTIONS, etc.)
    allow_headers=["*"],  # Allow all headers
)

# Set up routers
app.include_router(construct_blueprint_api(), tags=["ai_app"])
app.include_router(
    construct_blueprint_browser_api(), prefix="/browser", tags=["ai_browser"]
)
app.include_router(construct_blueprint_train_api(), prefix="/train", tags=["ai_train"])
app.include_router(construct_blueprint_email_api(), prefix="/email", tags=["ai_email"])

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