Update app.py

app.py

@@ -1,5 +1,7 @@
 import os
 import json
+import pyperclip
+import gradio as gr
 import vertexai
 from vertexai.generative_models import GenerativeModel
 import vertexai.preview.generative_models as generative_models
@@ -58,5 +60,28 @@ def generate(text):
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Test the function
-print(generate("Hello, I am Elien"))
+def copy_to_clipboard(text):
+    try:
+        pyperclip.copy(text)
+        return "Text copied to clipboard!"
+    except Exception as e:
+        return f"Failed to copy text: {str(e)}"
+
+with gr.Blocks() as demo:
+    gr.Markdown("""
+        # Chuunibyou Text Generator
+        Transform text into an elaborate and formal style with a Chuunibyou tone.
+    """)
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            input_text = gr.Textbox(lines=2, placeholder="Enter text here...")
+            output_text = gr.Textbox(lines=4, interactive=False)
+            generate_button = gr.Button("Submit")
+            copy_button = gr.Button("Copy to Clipboard")
+            hidden_state = gr.State()
+
+    generate_button.click(fn=generate, inputs=input_text, outputs=[output_text, hidden_state])
+    copy_button.click(fn=copy_to_clipboard, inputs=hidden_state, outputs=gr.Textbox(interactive=False, label="Notification"))
+
+demo.launch()