Saqib
commited on
Commit
•
8e57d9e
1
Parent(s):
7b2b54d
Update modules/app.py
Browse files- modules/app.py +3 -10
modules/app.py
CHANGED
@@ -18,13 +18,6 @@ pixart_sigma_base_url = os.getenv("PIXART_SIGMA_BASE_URL")
|
|
18 |
allowed_user_agent = os.getenv("ALLOWED_USER_AGENT")
|
19 |
|
20 |
|
21 |
-
async def verify_user_agent(request: Request):
|
22 |
-
print(request.headers)
|
23 |
-
user_agent = request.headers.get('user-agent', None)
|
24 |
-
if user_agent != allowed_user_agent:
|
25 |
-
raise HTTPException(status_code=403, detail="Access denied")
|
26 |
-
return True
|
27 |
-
|
28 |
def generate_hash(length=12):
|
29 |
# Characters that can appear in the hash
|
30 |
characters = string.ascii_lowercase + string.digits
|
@@ -36,7 +29,7 @@ def generate_hash(length=12):
|
|
36 |
async def read_root():
|
37 |
return {"message": "Saqib's API"}
|
38 |
|
39 |
-
@app.post("/whisper"
|
40 |
async def whisper(request: Request):
|
41 |
data = await request.json() # Extracting JSON data from request
|
42 |
if "audio_url" not in data:
|
@@ -80,7 +73,7 @@ async def whisper(request: Request):
|
|
80 |
return f"API request failed: {post_resp.status}"
|
81 |
return await post_resp.json()
|
82 |
|
83 |
-
@app.post("/img2location"
|
84 |
async def img2location(request: Request):
|
85 |
request_json = await request.json()
|
86 |
image_url = request_json.get("image_url", None)
|
@@ -163,7 +156,7 @@ async def img2location(request: Request):
|
|
163 |
|
164 |
raise ValueError(f"Unexpected response: {json_response}")
|
165 |
|
166 |
-
@app.post("/pixart-sigma"
|
167 |
async def pixart_sigma(request: Request):
|
168 |
request_json = await request.json()
|
169 |
prompt = request_json.get("prompt", None)
|
|
|
18 |
allowed_user_agent = os.getenv("ALLOWED_USER_AGENT")
|
19 |
|
20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
def generate_hash(length=12):
|
22 |
# Characters that can appear in the hash
|
23 |
characters = string.ascii_lowercase + string.digits
|
|
|
29 |
async def read_root():
|
30 |
return {"message": "Saqib's API"}
|
31 |
|
32 |
+
@app.post("/whisper")
|
33 |
async def whisper(request: Request):
|
34 |
data = await request.json() # Extracting JSON data from request
|
35 |
if "audio_url" not in data:
|
|
|
73 |
return f"API request failed: {post_resp.status}"
|
74 |
return await post_resp.json()
|
75 |
|
76 |
+
@app.post("/img2location")
|
77 |
async def img2location(request: Request):
|
78 |
request_json = await request.json()
|
79 |
image_url = request_json.get("image_url", None)
|
|
|
156 |
|
157 |
raise ValueError(f"Unexpected response: {json_response}")
|
158 |
|
159 |
+
@app.post("/pixart-sigma")
|
160 |
async def pixart_sigma(request: Request):
|
161 |
request_json = await request.json()
|
162 |
prompt = request_json.get("prompt", None)
|