Update app.py

app.py

@@ -1,10 +1,10 @@
 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
+import gradio as gr
+import pyperclip
 
 # Read the service account key JSON file path from environment variable
 SERVICE_ACCOUNT_KEY_PATH = os.getenv("GOOGLE_APPLICATION_CREDENTIALS")
@@ -58,30 +58,34 @@ def generate(text):
         return response_text if response_text else "No valid response generated or response was blocked."
 
     except Exception as e:
-        return f"Error: {str(e)}"
+        return str(e)
 
 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)}"
+    pyperclip.copy(text)
+    return "Text copied to clipboard!"
+
+iface = gr.Interface(
+    fn=generate,
+    inputs=gr.Textbox(lines=2, placeholder="Enter text here..."),
+    outputs=gr.Textbox(visible=False),
+    title="Chuunibyou Text Generator",
+    description="Transform text into an elaborate and formal style with a Chuunibyou tone.",
+    live=False
+)
+
+def update_output(text):
+    return gr.update(value=text, visible=True)
+
+iface_after_submit = iface.set_event_handlers(
+    submit=lambda x: update_output(generate(x))
+)
 
-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()
+copy_button = gr.Button("Copy to Clipboard")
+copy_button.click(copy_to_clipboard, inputs=iface_after_submit.output_component, outputs=gr.Textbox(visible=True))
 
-    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"))
+iface_after_submit = gr.Column(
+    [iface_after_submit, copy_button]
+)
 
-demo.launch()
+if __name__ == "__main__":
+    iface_after_submit.launch()