Update app.py

app.py

@@ -8,9 +8,6 @@ def generate_image(prompt, code):
     if code != os.getenv("code"):
         raise gr.Error("❗ Не верный ключ!")
         return None
-    # Сохраняем код в куки после успешной проверки
-    gr.set_cookie("access_code", code)
-    
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json",
@@ -30,12 +27,25 @@ def generate_image(prompt, code):
 css = """
 footer {visibility: hidden !important;}
 """
+js = """
+document.addEventListener('DOMContentLoaded', (event) => {
+  if (document.cookie.split(';').some((item) => item.trim().startsWith('code='))) {
+    const codeValue = document.cookie.split('; ').find(row => row.startsWith('code=')).split('=')[1];
+    document.getElementById('code').value = codeValue;
+  }
+});
+
+function saveCode() {
+  const codeValue = document.getElementById('code').value;
+  document.cookie = 'code=' + codeValue + ';max-age=31536000'; // Сохраняем на 1 год
+}
+"""
+
 # Создание интерфейса с помощью Gradio
 with gr.Blocks(css=css, theme='YTheme/Sketch') as demo:
+    gr.HTML(f"<script>{js}</script>")
     with gr.Row():
-        # Загружаем сохранённый код из куки, если он существует
-        saved_code = gr.get_cookie("access_code")
-        code = gr.Textbox(label="Ключ доступа", type="password", value=saved_code or "")
+        code = gr.Textbox(label="Ключ доступа", type="password", elem_id="code", change=saveCode)
     with gr.Row():
         prompt_input = gr.Textbox(label="Описание изображения")
     submit_btn = gr.Button("Генерация", variant='primary')