Update blackgat/app/main.py

blackgat/app/main.py

@@ -1,20 +1,15 @@
-# main.py
+# app/main.py
 from fastapi import FastAPI
 from pydantic import BaseModel
 from typing import Dict
-from ai_models import (
+from app.ai_models import (
     scribe_generate, killchain_ai,
     heatseeker_score, recon_gpt,
     exploit_suggestion
 )
 
-app = FastAPI(
-    title="BlackGat AI",
-    description="Cloud-based AI for bug bounty automation",
-    version="1.0.0"
-)
+app = FastAPI(title="BlackGat AI", version="1.0", description="Cloud AI for Bug Bounty Hunters")
 
-# Models
 class VulnReport(BaseModel):
     type: str
     url: str
@@ -35,7 +30,6 @@ class HeatInput(BaseModel):
 class ReconPrompt(BaseModel):
     prompt: str
 
-# API Routes
 @app.post("/scribe")
 def scribe(data: VulnReport):
     return scribe_generate(data.dict())