Lucii1 commited on
Commit
a759e39
·
1 Parent(s): 0a83deb
Files changed (2) hide show
  1. .env.template +7 -0
  2. api_server.py +27 -0
.env.template ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ SCRAPINGDOG_API_KEY=
2
+ GEMINI_API_KEY=
3
+ GOOGLE_SAFE_BROWSING_API_KEY=
4
+ GOOGLE_APPLICATION_CREDENTIALS=
5
+ IMGBB_API_KEY=
6
+ GOOGLE_CSE_CX=
7
+ ENV=
api_server.py CHANGED
@@ -19,6 +19,33 @@ from AIGVDet import run_video_to_json
19
  UPLOAD_DIR = "temp_uploads"
20
  MAX_WORKERS = 4
21
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
22
  app = FastAPI(
23
  title="Multimedia Analysis API (Polling Mode)",
24
  description="API phân tích đa phương tiện sử dụng cơ chế Polling để tránh Timeout.",
 
19
  UPLOAD_DIR = "temp_uploads"
20
  MAX_WORKERS = 4
21
 
22
+ ENV = os.getenv("ENV")
23
+ cred_json = os.getenv("GOOGLE_APPLICATION_CREDENTIALS")
24
+
25
+ if ENV == "hf":
26
+ if cred_json:
27
+ try:
28
+ json.loads(cred_json)
29
+
30
+ file_path = "google-credentials.json"
31
+ with open(file_path, "w") as f:
32
+ f.write(cred_json)
33
+
34
+ os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = file_path
35
+
36
+ print("[INFO] Google credentials saved to", file_path)
37
+
38
+ except json.JSONDecodeError:
39
+ print("[ERROR] GOOGLE_APPLICATION_CREDENTIALS is not valid JSON")
40
+
41
+ else:
42
+ print("[ERROR] GOOGLE_APPLICATION_CREDENTIALS is missing")
43
+
44
+ else:
45
+ # DEV mode (local)
46
+ print("[INFO] ENV != hf → skip Google credentials setup")
47
+
48
+
49
  app = FastAPI(
50
  title="Multimedia Analysis API (Polling Mode)",
51
  description="API phân tích đa phương tiện sử dụng cơ chế Polling để tránh Timeout.",