Update app.py

app.py

@@ -4,7 +4,6 @@ import vertexai
 from vertexai.generative_models import GenerativeModel
 import vertexai.preview.generative_models as generative_models
 import gradio as gr
-import streamlit as st
 
 # Read the service account key JSON file path from environment variable
 SERVICE_ACCOUNT_KEY_PATH = os.getenv("GOOGLE_APPLICATION_CREDENTIALS")
@@ -60,29 +59,34 @@ def generate(text):
     except Exception as e:
         return str(e)
 
-# Custom JavaScript for "Copy to Clipboard" functionality
-custom_js = """
-function copyToClipboard(textbox_id) {
-    const textbox = document.getElementById(textbox_id);
-    if (textbox) {
-        textbox.select();
+# Custom HTML and JavaScript for "Copy to Clipboard" functionality
+custom_html = """
+<div style="display: flex; flex-direction: column; align-items: flex-start;">
+    <textarea id="output-textbox" rows="6" style="width: 100%;"></textarea>
+    <button onclick="copyToClipboard()">Copy to Clipboard</button>
+</div>
+<script>
+    function copyToClipboard() {
+        var copyText = document.getElementById("output-textbox");
+        copyText.select();
         document.execCommand("copy");
     }
-}
+</script>
 """
 
 iface = gr.Interface(
     fn=generate,
     inputs=gr.Textbox(lines=2, placeholder="Enter text here..."),
-    outputs=gr.Textbox(label="Output"),
+    outputs="html",
     title="Chuunibyou Text Generator",
     description="Transform text into an elaborate and formal style with a nobleman tone.",
     live=True
 )
 
-iface.launch(js=custom_js, additional_buttons=[
-    {
-        "label": "Copy to Clipboard",
-        "onclick": "copyToClipboard('output-textbox')"
-    }
-])
+def launch_custom_interface():
+    iface.launch()
+    with gr.TabbedInterface(fn=generate, inputs=gr.Textbox(lines=2, placeholder="Enter text here..."), outputs=gr.HTML(label="Output")) as ti:
+        ti.add(custom_html)
+
+if __name__ == "__main__":
+    launch_custom_interface()