From 9e1bcae5453e49e5b2c67db88c923bfdf67ec4e2 Mon Sep 17 00:00:00 2001
From: Abi Raja <abimanyuraja@gmail.com>
Date: Wed, 10 Apr 2024 14:51:15 -0400
Subject: [PATCH] remove access code/token functionality

---
 backend/access_token.py                       |  21 ---
 backend/routes/generate_code.py               |  27 +---
 frontend/src/App.tsx                          |   8 +-
 frontend/src/components/PicoBadge.tsx         |  30 +---
 frontend/src/components/SettingsDialog.tsx    |   6 -
 .../components/settings/AccessCodeSection.tsx | 142 ------------------
 frontend/src/types.ts                         |   1 -
 7 files changed, 16 insertions(+), 219 deletions(-)
 delete mode 100644 backend/access_token.py
 delete mode 100644 frontend/src/components/settings/AccessCodeSection.tsx

diff --git a/backend/access_token.py b/backend/access_token.py
deleted file mode 100644
index 35fa657..0000000
--- a/backend/access_token.py
+++ /dev/null
@@ -1,21 +0,0 @@
-import json
-import os
-import httpx
-
-
-async def validate_access_token(access_code: str):
-    async with httpx.AsyncClient() as client:
-        url = (
-            "https://backend.buildpicoapps.com/screenshot_to_code/validate_access_token"
-        )
-        data = json.dumps(
-            {
-                "access_code": access_code,
-                "secret": os.environ.get("PICO_BACKEND_SECRET"),
-            }
-        )
-        headers = {"Content-Type": "application/json"}
-
-        response = await client.post(url, content=data, headers=headers)
-        response_data = response.json()
-        return response_data
diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py
index 259983f..a7edb9b 100644
--- a/backend/routes/generate_code.py
+++ b/backend/routes/generate_code.py
@@ -16,7 +16,6 @@ from mock_llm import mock_completion
 from typing import Dict, List, cast, get_args
 from image_generation import create_alt_url_mapping, generate_images
 from prompts import assemble_imported_code_prompt, assemble_prompt
-from access_token import validate_access_token
 from datetime import datetime
 import json
 from prompts.claude_prompts import VIDEO_PROMPT
@@ -98,27 +97,13 @@ async def stream_code(websocket: WebSocket):
     # Get the OpenAI API key from the request. Fall back to environment variable if not provided.
     # If neither is provided, we throw an error.
     openai_api_key = None
-    if "accessCode" in params and params["accessCode"]:
-        print("Access code - using platform API key")
-        res = await validate_access_token(params["accessCode"])
-        if res["success"]:
-            openai_api_key = os.environ.get("PLATFORM_OPENAI_API_KEY")
-        else:
-            await websocket.send_json(
-                {
-                    "type": "error",
-                    "value": res["failure_reason"],
-                }
-            )
-            return
+    if params["openAiApiKey"]:
+        openai_api_key = params["openAiApiKey"]
+        print("Using OpenAI API key from client-side settings dialog")
     else:
-        if params["openAiApiKey"]:
-            openai_api_key = params["openAiApiKey"]
-            print("Using OpenAI API key from client-side settings dialog")
-        else:
-            openai_api_key = os.environ.get("OPENAI_API_KEY")
-            if openai_api_key:
-                print("Using OpenAI API key from environment variable")
+        openai_api_key = os.environ.get("OPENAI_API_KEY")
+        if openai_api_key:
+            print("Using OpenAI API key from environment variable")
 
     if not openai_api_key and code_generation_model == "gpt_4_vision":
         print("OpenAI API key not found")
diff --git a/frontend/src/App.tsx b/frontend/src/App.tsx
index 5a51bc6..2329105 100644
--- a/frontend/src/App.tsx
+++ b/frontend/src/App.tsx
@@ -66,7 +66,6 @@ function App() {
       codeGenerationModel: CodeGenerationModel.GPT_4_VISION,
       // Only relevant for hosted version
       isTermOfServiceAccepted: false,
-      accessCode: null,
     },
     "setting"
   );
@@ -362,7 +361,7 @@ function App() {
 
   return (
     <div className="mt-2 dark:bg-black dark:text-white">
-      {IS_RUNNING_ON_CLOUD && <PicoBadge settings={settings} />}
+      {IS_RUNNING_ON_CLOUD && <PicoBadge />}
       {IS_RUNNING_ON_CLOUD && (
         <TermsOfServiceDialog
           open={!settings.isTermOfServiceAccepted}
@@ -394,10 +393,7 @@ function App() {
 
           {appState !== AppState.CODE_READY && <TipLink />}
 
-          {IS_RUNNING_ON_CLOUD &&
-            !(settings.openAiApiKey || settings.accessCode) && (
-              <OnboardingNote />
-            )}
+          {IS_RUNNING_ON_CLOUD && !settings.openAiApiKey && <OnboardingNote />}
 
           {IS_OPENAI_DOWN && (
             <div className="bg-black text-white dark:bg-white dark:text-black p-3 rounded">
diff --git a/frontend/src/components/PicoBadge.tsx b/frontend/src/components/PicoBadge.tsx
index 70a4f20..bdd0083 100644
--- a/frontend/src/components/PicoBadge.tsx
+++ b/frontend/src/components/PicoBadge.tsx
@@ -1,6 +1,4 @@
-import { Settings } from "../types";
-
-export function PicoBadge({ settings }: { settings: Settings }) {
+export function PicoBadge() {
   return (
     <>
       <a
@@ -14,26 +12,14 @@ export function PicoBadge({ settings }: { settings: Settings }) {
           feature requests?
         </div>
       </a>
-      {!settings.accessCode && (
-        <a href="https://picoapps.xyz?ref=screenshot-to-code" target="_blank">
-          <div
-            className="fixed z-50 bottom-5 right-5 rounded-md shadow text-black
+      <a href="https://picoapps.xyz?ref=screenshot-to-code" target="_blank">
+        <div
+          className="fixed z-50 bottom-5 right-5 rounded-md shadow text-black
          bg-white px-4 text-xs py-3 cursor-pointer"
-          >
-            an open source project by Pico
-          </div>
-        </a>
-      )}
-      {settings.accessCode && (
-        <a href="mailto:support@picoapps.xyz" target="_blank">
-          <div
-            className="fixed z-50 bottom-5 right-5 rounded-md shadow text-black
-         bg-white px-4 text-xs py-3 cursor-pointer"
-          >
-            email support
-          </div>
-        </a>
-      )}
+        >
+          an open source project by Pico
+        </div>
+      </a>
     </>
   );
 }
diff --git a/frontend/src/components/SettingsDialog.tsx b/frontend/src/components/SettingsDialog.tsx
index ea23bc8..2e7814b 100644
--- a/frontend/src/components/SettingsDialog.tsx
+++ b/frontend/src/components/SettingsDialog.tsx
@@ -22,7 +22,6 @@ import {
   AccordionItem,
   AccordionTrigger,
 } from "./ui/accordion";
-import AccessCodeSection from "./settings/AccessCodeSection";
 
 interface Props {
   settings: Settings;
@@ -47,11 +46,6 @@ function SettingsDialog({ settings, setSettings }: Props) {
           <DialogTitle className="mb-4">Settings</DialogTitle>
         </DialogHeader>
 
-        {/* Access code */}
-        {IS_RUNNING_ON_CLOUD && (
-          <AccessCodeSection settings={settings} setSettings={setSettings} />
-        )}
-
         <div className="flex items-center space-x-2">
           <Label htmlFor="image-generation">
             <div>DALL-E Placeholder Image Generation</div>
diff --git a/frontend/src/components/settings/AccessCodeSection.tsx b/frontend/src/components/settings/AccessCodeSection.tsx
deleted file mode 100644
index 5135307..0000000
--- a/frontend/src/components/settings/AccessCodeSection.tsx
+++ /dev/null
@@ -1,142 +0,0 @@
-import { useEffect, useState } from "react";
-import { Settings } from "../../types";
-import { Button } from "../ui/button";
-import { Input } from "../ui/input";
-import { Label } from "../ui/label";
-import useThrottle from "../../hooks/useThrottle";
-import { Progress } from "../ui/progress";
-import { PICO_BACKEND_FORM_SECRET } from "../../config";
-
-interface Props {
-  settings: Settings;
-  setSettings: React.Dispatch<React.SetStateAction<Settings>>;
-}
-
-interface UsageResponse {
-  used_credits: number;
-  total_credits: number;
-  is_valid: boolean;
-}
-
-enum FetchState {
-  EMPTY = "EMPTY",
-  LOADING = "LOADING",
-  INVALID = "INVALID",
-  VALID = "VALID",
-}
-
-function AccessCodeSection({ settings, setSettings }: Props) {
-  const [isLoading, setIsLoading] = useState(false);
-  const [isValid, setIsValid] = useState(false);
-  const [usedCredits, setUsedCredits] = useState(0);
-  const [totalCredits, setTotalCredits] = useState(0);
-  const throttledAccessCode = useThrottle(settings.accessCode || "", 500);
-
-  const fetchState = (() => {
-    if (!settings.accessCode) return FetchState.EMPTY;
-    if (isLoading) return FetchState.LOADING;
-    if (!isValid) return FetchState.INVALID;
-    return FetchState.VALID;
-  })();
-
-  async function fetchUsage(accessCode: string) {
-    const res = await fetch(
-      "https://backend.buildpicoapps.com/screenshot_to_code/get_access_code_usage",
-      {
-        method: "POST",
-        headers: {
-          "Content-Type": "application/json",
-        },
-        body: JSON.stringify({
-          access_code: accessCode,
-          secret: PICO_BACKEND_FORM_SECRET,
-        }),
-      }
-    );
-    const usage = (await res.json()) as UsageResponse;
-
-    if (!usage.is_valid) {
-      setIsValid(false);
-    } else {
-      setIsValid(true);
-      setUsedCredits(usage.used_credits);
-      setTotalCredits(usage.total_credits);
-    }
-
-    setIsLoading(false);
-  }
-
-  useEffect(() => {
-    // Don't do anything if access code is empty
-    if (!throttledAccessCode) return;
-
-    setIsLoading(true);
-    setIsValid(true);
-
-    // Wait for 500 ms before fetching usage
-    setTimeout(async () => {
-      await fetchUsage(throttledAccessCode);
-    }, 500);
-  }, [throttledAccessCode]);
-
-  return (
-    <div className="flex flex-col space-y-4 bg-slate-200 p-4 rounded dark:text-white dark:bg-slate-800">
-      <Label htmlFor="access-code">
-        <div>Access Code</div>
-      </Label>
-
-      <Input
-        id="access-code"
-        className="border-gray-700 dark:border-gray-700 dark:bg-gray-800 dark:text-white"
-        placeholder="Enter your Screenshot to Code access code"
-        value={settings.accessCode || ""}
-        onChange={(e) =>
-          setSettings((s) => ({
-            ...s,
-            accessCode: e.target.value,
-          }))
-        }
-      />
-
-      {fetchState === "EMPTY" && (
-        <div className="flex items-center justify-between">
-          <a href="https://buy.stripe.com/8wM6sre70gBW1nqaEE" target="_blank">
-            <Button size="sm" variant="secondary">
-              Buy credits
-            </Button>
-          </a>
-        </div>
-      )}
-
-      {fetchState === "LOADING" && (
-        <div className="flex items-center justify-between">
-          <span className="text-xs text-gray-700">Loading...</span>
-        </div>
-      )}
-
-      {fetchState === "INVALID" && (
-        <>
-          <div className="flex items-center justify-between">
-            <span className="text-xs text-gray-700">Invalid access code</span>
-          </div>
-        </>
-      )}
-
-      {fetchState === "VALID" && (
-        <>
-          <Progress value={(usedCredits / totalCredits) * 100} />
-          <div className="flex items-center justify-between">
-            <span className="text-xs text-gray-700">
-              {usedCredits} out of {totalCredits} credits used
-            </span>
-            <a href="https://buy.stripe.com/8wM6sre70gBW1nqaEE" target="_blank">
-              <Button size="sm">Add credits</Button>
-            </a>
-          </div>
-        </>
-      )}
-    </div>
-  );
-}
-
-export default AccessCodeSection;
diff --git a/frontend/src/types.ts b/frontend/src/types.ts
index fd46082..bbabeff 100644
--- a/frontend/src/types.ts
+++ b/frontend/src/types.ts
@@ -16,7 +16,6 @@ export interface Settings {
   codeGenerationModel: CodeGenerationModel;
   // Only relevant for hosted version
   isTermOfServiceAccepted: boolean;
-  accessCode: string | null;
 }
 
 export enum AppState {