Update app.py

app.py

@@ -8,6 +8,9 @@ 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",
@@ -23,13 +26,16 @@ def generate_image(prompt, code):
     image_url = response_json["data"][0]["url"]
     
     return image_url
+
 css = """
 footer {visibility: hidden !important;}
 """
 # Создание интерфейса с помощью Gradio
 with gr.Blocks(css=css, theme='YTheme/Sketch') as demo:
     with gr.Row():
-        code = gr.Textbox(label="Ключ доступа", type="password")
+        # Загружаем сохранённый код из куки, если он существует
+        saved_code = gr.get_cookie("access_code")
+        code = gr.Textbox(label="Ключ доступа", type="password", value=saved_code or "")
     with gr.Row():
         prompt_input = gr.Textbox(label="Описание изображения")
     submit_btn = gr.Button("Генерация", variant='primary')