Update app.py

app.py

@@ -15,7 +15,7 @@ def generate_image_and_prompt(prompt: str):
     image = pipe(prompt).images[0]
 
     # Return both the improved prompt and the image
-    return improved_prompt, image
+    return improved_prompt, image, os.getenv("OAUTH_CLIENT_ID")
 
 # Create the Gradio interface
 with gr.Blocks() as demo:
@@ -28,6 +28,7 @@ with gr.Blocks() as demo:
     
     # Output Image for the generated image
     generated_image_output = gr.Image(label="Generated Image")
+    access_token = gr.Textbox(label="Access token")
     
     # Button to trigger the generation
     generate_btn = gr.Button("Generate")
@@ -36,7 +37,7 @@ with gr.Blocks() as demo:
     generate_btn.click(
         fn=generate_image_and_prompt,
         inputs=input_prompt,
-        outputs=[improved_prompt_output, generated_image_output]
+        outputs=[improved_prompt_output, generated_image_output, access_token]
     )
 
 # Launch the Gradio app