Github/screenshot-to-code
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'main' into hosted

Browse Source
This commit is contained in:
Abi Raja 2023-12-06 10:50:42 -05:00
commit 4c4a19a40a
2 changed files with 4 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/access_token.py
View File

@ -18,10 +18,4 @@ async def validate_access_token(access_code: str):
response = await client.post(url, content=data, headers=headers) response = await client.post(url, content=data, headers=headers)
response_data = response.json() response_data = response.json()
return response_data
if response_data["success"]:
print("Access token is valid.")
return True
else:
print(f"Access token validation failed: {response_data['failure_reason']}")
return False

5
backend/main.py
View File

@ -111,13 +111,14 @@ async def stream_code(websocket: WebSocket):
openai_api_key = None openai_api_key = None
if "accessCode" in params and params["accessCode"]: if "accessCode" in params and params["accessCode"]:
print("Access code - using platform API key") print("Access code - using platform API key")
if await validate_access_token(params["accessCode"]): res = await validate_access_token(params["accessCode"])
if res["success"]:
openai_api_key = os.environ.get("PLATFORM_OPENAI_API_KEY") openai_api_key = os.environ.get("PLATFORM_OPENAI_API_KEY")
else: else:
await websocket.send_json( await websocket.send_json(
{ {
"type": "error", "type": "error",
"value": "Invalid access code or you're out of credits. Please try again.", "value": res["failure_reason"],
} }
) )
return return