cloud450 commited on
Commit
c180de7
·
verified ·
1 Parent(s): ab62a41

Upload main.py

Browse files
Files changed (1) hide show
  1. main.py +36 -0
main.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI
2
+ from fastapi.middleware.cors import CORSMiddleware
3
+ from routers import predict, monitor, reports, upload, blockchain
4
+ from dotenv import load_dotenv
5
+ import os
6
+
7
+ # Load environment variables
8
+ load_dotenv()
9
+ # Trigger reload
10
+
11
+ app = FastAPI(
12
+ title="Network IDS API",
13
+ description="Backend API for Intrusion Detection System",
14
+ version="1.0.0"
15
+ )
16
+
17
+ # Allow Streamlit frontend
18
+ app.add_middleware(
19
+ CORSMiddleware,
20
+ allow_origins=["*"],
21
+ allow_credentials=True,
22
+ allow_methods=["*"],
23
+ allow_headers=["*"],
24
+ )
25
+
26
+ # Routers
27
+ app.include_router(predict.router, prefix="/predict", tags=["Prediction"])
28
+ app.include_router(monitor.router, prefix="/monitor", tags=["Live Monitor"])
29
+ app.include_router(reports.router, prefix="/reports", tags=["Threat Reports"])
30
+ app.include_router(upload.router, prefix="/upload", tags=["File Upload"])
31
+ app.include_router(blockchain.router, prefix="/blockchain", tags=["Blockchain"])
32
+
33
+
34
+ @app.get("/")
35
+ def home():
36
+ return {"message": "Network IDS Backend Running!"}