initial commit

새로운 repository 를 만들었습니다.

backend/app/main.py

@@ -0,0 +1,40 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from app.model_load import use_model  # predictor.py에서 함수 import
+from app.exe import predict_url_maliciousness
+from app.utils import convert_numpy_to_python_types
+from fastapi.middleware.cors import CORSMiddleware
+
+app = FastAPI()
+
+# FastAPI 인스턴스에 CORS 미들웨어 추가
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # 또는 ["http://localhost:3000"] (프론트 URL)
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# 요청 데이터 구조 정의
+class UrlRequest(BaseModel):
+    url: str
+
+@app.get("/")
+def root():
+    return {"message": "URL 악성 판별기 FastAPI 서버 정상 작동 중!"}
+
+@app.post("/predict")
+def predict(request: UrlRequest):
+     url = request.url
+
+     result_model1 = convert_numpy_to_python_types(use_model(url))
+     result_model2 = convert_numpy_to_python_types(predict_url_maliciousness(url))
+     
+     response_data = {
+         "url": url,
+         "model1": result_model1,
+         "model2": result_model2
+     }
+     
+     return convert_numpy_to_python_types(response_data)