Update app.py

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
 import os
 
-# If 'share_btn.py' is a custom module, ensure it's available; otherwise, comment out or remove these imports
+# If 'share_btn.py' is a custom module and contains 'community_icon_html' and 'loading_icon_html',
+# make sure it's available; otherwise, you can define these variables directly or remove them if not needed.
 # from share_btn import community_icon_html, loading_icon_html, share_js
 
 # Load interfaces from Hugging Face Spaces
@@ -16,48 +17,95 @@ def get_images(prompt):
 
 def get_prompts(prompt_text):
     # Generate expanded prompts using the text_gen interface
-    return text_gen(prompt_text)
+    expanded_prompt = text_gen(prompt_text)
+    return expanded_prompt
 
 css = '''
-/* Your CSS styles remain unchanged */
 .animate-spin {
     animation: spin 1s linear infinite;
 }
-/* ... rest of the CSS ... */
+@keyframes spin {
+    from {
+        transform: rotate(0deg);
+    }
+    to {
+        transform: rotate(360deg);
+    }
+}
+#share-btn-container {
+    display: flex; padding-left: 0.5rem !important; padding-right: 0.5rem !important; background-color: #000000; justify-content: center; align-items: center; border-radius: 9999px !important; width: 13rem;
+}
+#share-btn {
+    all: initial; color: #ffffff;font-weight: 600; cursor:pointer; font-family: 'IBM Plex Sans', sans-serif; margin-left: 0.5rem !important; padding-top: 0.25rem !important; padding-bottom: 0.25rem !important;
+}
+#share-btn * {
+    all: unset;
+}
+#share-btn-container div:nth-child(-n+2){
+    width: auto !important;
+    min-height: 0px !important;
+}
+#share-btn-container .wrap {
+    display: none !important;
+}
+a {text-decoration-line: underline;}
 '''
 
 with gr.Blocks(css=css) as demo:
     gr.HTML("""
     <div style="text-align: center; max-width: 700px; margin: 0 auto;">
-        <div style="display: inline-flex; align-items: center; gap: 0.8rem; font-size: 1.75rem;">
-            <h1 style="font-weight: 900; margin-bottom: 7px; margin-top: 5px;">
-                Prompt Refinery
-            </h1>
+        <div
+        style="
+            display: inline-flex;
+            align-items: center;
+            gap: 0.8rem;
+            font-size: 1.75rem;
+        "
+        >
+        <h1 style="font-weight: 900; margin-bottom: 7px; margin-top: 5px;">
+            Prompt Refinery
+        </h1>
         </div>
         <p style="margin-bottom: 10px; font-size: 94%">
-            🏭 Prompt Refinery generates variations of your prompt using 
-            <a href="https://huggingface.co/spaces/Gustavosta/MagicPrompt-Stable-Diffusion" target="_blank">
-                MagicPrompt and Stable Diffusion
-            </a>
+           🏭 Prompt Refinery generates variations of your prompt using 
+           <a href="https://huggingface.co/spaces/Gustavosta/MagicPrompt-Stable-Diffusion" target="_blank">
+               MagicPrompt and Stable Diffusion
+           </a>
         </p>
     </div>
     """)
     with gr.Row():
         with gr.Column():
-            input_text = gr.Textbox(label="Input text prompt", lines=2, elem_id="input-text")
-            see_prompts = gr.Button("✍️Expand my prompts")
+            input_text = gr.Textbox(
+                label="Input text prompt",
+                lines=2,
+                elem_id="input-text"
+            )
+            see_prompts = gr.Button("✍️ Expand my prompts")
         with gr.Column():
-            text_output = gr.Textbox(label="🏭 Expanded text prompts", lines=8, elem_id="translated")
+            text_output = gr.Textbox(
+                label="🏭 Expanded text prompts",
+                lines=8,
+                elem_id="translated"
+            )
             diffuse_btn = gr.Button(value="🏭 Render Images for My Prompts")
         with gr.Column(elem_id="generated-gallery"):
             sd_output = gr.Gallery().style(grid=2, height="auto")
             with gr.Group(elem_id="share-btn-container"):
-                # Replace the content of community_icon_html and loading_icon_html with actual HTML or leave empty
+                # If you have HTML content for these icons, you can include it here
                 community_icon = gr.HTML("", visible=False)
                 loading_icon = gr.HTML("", visible=False)
 
-    # Define the interactions without 'api_name', which is deprecated
-    see_prompts.click(get_prompts, inputs=[input_text], outputs=[text_output])
-    diffuse_btn.click(get_images, inputs=[text_output], outputs=[sd_output, community_icon, loading_icon])
+    # Connect the buttons to their respective functions
+    see_prompts.click(
+        fn=get_prompts,
+        inputs=[input_text],
+        outputs=[text_output]
+    )
+    diffuse_btn.click(
+        fn=get_images,
+        inputs=[text_output],
+        outputs=[sd_output, community_icon, loading_icon]
+    )
 
 demo.launch(debug=True)