Update app.py

app.py

@@ -122,7 +122,7 @@ def generate_item(tag, ideas, max_retries=3):
             image_response = client.models.generate_images(
                 model='imagen-3.0-generate-002',
                 prompt=image_prompt,
-                config=types.GenerateImagesConfig(
+                config=types.GenerateContentConfig(
                     number_of_images=1,
                     aspect_ratio="9:16",
                     person_generation="DONT_ALLOW"
@@ -289,68 +289,82 @@ def load_previous(tag, current_index, feed_items):
 
 def generate_share_links(image_base64, caption):
     """
-    Generate share links for social media platforms.
+    Generate share links for social media platforms with a download link.
     
     Args:
         image_base64 (str): The base64-encoded image data.
         caption (str): The caption to share with the image.
     
     Returns:
-        str: HTML string with share links.
+        str: HTML string with share links and download instructions.
     """
-    # Create a data URL for the image
     image_data_url = f"data:image/png;base64,{image_base64}"
     encoded_caption = urllib.parse.quote(caption)
-    encoded_image_url = urllib.parse.quote(image_data_url)
 
-    # Generate share links for each platform
+    # Generate share links with instructions to download the image
     share_links = """
     <div style="
         display: flex;
-        flex-wrap: wrap;
-        justify-content: center;
+        flex-direction: column;
+        align-items: center;
         gap: 10px;
         margin-top: 10px;
         color: white;
         font-family: Arial, sans-serif;
     ">
-        <a href="https://www.tiktok.com/upload?caption={caption}" target="_blank" style="
-            background-color: #00f2ea;
-            color: #000;
-            padding: 5px 10px;
-            border-radius: 5px;
-            text-decoration: none;
-        ">Share on TikTok</a>
-        <a href="https://www.instagram.com/?url={image_url}&caption={caption}" target="_blank" style="
-            background-color: #e1306c;
-            color: white;
-            padding: 5px 10px;
-            border-radius: 5px;
-            text-decoration: none;
-        ">Share on Instagram</a>
-        <a href="https://www.facebook.com/sharer/sharer.php?u={image_url}&quote={caption}" target="_blank" style="
-            background-color: #4267b2;
-            color: white;
-            padding: 5px 10px;
-            border-radius: 5px;
-            text-decoration: none;
-        ">Share on Facebook</a>
-        <a href="https://twitter.com/intent/tweet?text={caption}&url={image_url}" target="_blank" style="
-            background-color: #1da1f2;
+        <p style="text-align: center;">To share, download the image and upload it to your platform:</p>
+        <a href="{image_url}" download="feed_item.png" style="
+            background-color: #4CAF50;
             color: white;
             padding: 5px 10px;
             border-radius: 5px;
             text-decoration: none;
-        ">Share on X</a>
-        <a href="https://pinterest.com/pin/create/button/?url={image_url}&description={caption}" target="_blank" style="
-            background-color: #bd081c;
-            color: white;
-            padding: 5px 10px;
-            border-radius: 5px;
-            text-decoration: none;
-        ">Share on Pinterest</a>
+        ">Download Image</a>
+        <div style="
+            display: flex;
+            flex-wrap: wrap;
+            justify-content: center;
+            gap: 10px;
+            margin-top: 10px;
+        ">
+            <a href="https://www.tiktok.com/upload?caption={caption}" target="_blank" style="
+                background-color: #00f2ea;
+                color: #000;
+                padding: 5px 10px;
+                border-radius: 5px;
+                text-decoration: none;
+            ">Share on TikTok</a>
+            <a href="https://www.instagram.com/?caption={caption}" target="_blank" style="
+                background-color: #e1306c;
+                color: white;
+                padding: 5px 10px;
+                border-radius: 5px;
+                text-decoration: none;
+            ">Share on Instagram</a>
+            <a href="https://www.facebook.com/sharer/sharer.php?quote={caption}" target="_blank" style="
+                background-color: #4267b2;
+                color: white;
+                padding: 5px 10px;
+                border-radius: 5px;
+                text-decoration: none;
+            ">Share on Facebook</a>
+            <a href="https://twitter.com/intent/tweet?text={caption}" target="_blank" style="
+                background-color: #1da1f2;
+                color: white;
+                padding: 5px 10px;
+                border-radius: 5px;
+                text-decoration: none;
+            ">Share on X</a>
+            <a href="https://pinterest.com/pin/create/button/?description={caption}" target="_blank" style="
+                background-color: #bd081c;
+                color: white;
+                padding: 5px 10px;
+                border-radius: 5px;
+                text-decoration: none;
+            ">Share on Pinterest</a>
+        </div>
     </div>
-    """.format(caption=encoded_caption, image_url=encoded_image_url)
+    """.format(image_url=image_data_url, caption=encoded_caption)
 
     return share_links
 
@@ -531,7 +545,7 @@ with gr.Blocks(
         .gr-button { width: 100%; margin-top: 10px; }
         .gr-form { background-color: #111; padding: 15px; border-radius: 10px; }
     """,
-    title="Create Your Feed"  # Updated title
+    title="Create Your Feed"
 ) as demo:
     # State variables
     current_tag = gr.State(value="")