Update app.py

app.py

@@ -31,16 +31,16 @@ def generate_ideas(tag):
     Return the response as a JSON object with a single key 'ideas' containing a list of 5 ideas.
     Example: {{"ideas": ["A neon-lit gaming setup with RGB lights flashing", "A futuristic robot assembling a gadget"]}}
     """
-    response = client.models.generate_content(
-        model='gemini-2.5-flash-preview-04-17',
-        contents=[prompt],
-        config=types.GenerateContentConfig(temperature=1.2)
-    )
     try:
+        response = client.models.generate_content(
+            model='gemini-2.5-flash-preview-04-17',
+            contents=[prompt],
+            config=types.GenerateContentConfig(temperature=1.2)
+        )
         response_json = json.loads(response.text.strip())
-        ideas = response_json['ideas']
-        return ideas
-    except (json.JSONDecodeError, KeyError):
+        return response_json['ideas']
+    except Exception as e:
+        print(f"Error generating ideas: {e}")
         return [
             f"A vibrant {tag} scene at sunset",
             f"A close-up of {tag} with neon lights",
@@ -69,33 +69,37 @@ def generate_item(tag, ideas):
     The image prompt should describe the scene vividly, specify a perspective and style, and ensure no text or letters are included.
     Example: {{"caption": "Neon vibes only! 🌌 #tech", "image_prompt": "A close-up view of a neon-lit gaming setup with RGB lights flashing, in a futuristic style, no text or letters"}}
     """
-    response = client.models.generate_content(
-        model='gemini-2.5-flash-preview-04-17',
-        contents=[prompt],
-        config=types.GenerateContentConfig(temperature=1.2)
-    )
     try:
+        response = client.models.generate_content(
+            model='gemini-2.5-flash-preview-04-17',
+            contents=[prompt],
+            config=types.GenerateContentConfig(temperature=1.2)
+        )
         response_json = json.loads(response.text.strip())
         text = response_json['caption']
         image_prompt = response_json['image_prompt']
-    except (json.JSONDecodeError, KeyError):
+    except Exception as e:
+        print(f"Error generating item: {e}")
         text = f"Obsessed with {tag}! 🔥 #{tag}"
         image_prompt = f"A vivid scene of {selected_idea}, in a vibrant pop art style, no text or letters"
 
-    image_response = client.models.generate_images(
-        model='imagen-3.0-generate-002',
-        prompt=image_prompt,
-        config=types.GenerateImagesConfig(
-            number_of_images=1,
-            aspect_ratio="9:16",
-            person_generation="DONT_ALLOW"
+    try:
+        image_response = client.models.generate_images(
+            model='imagen-3.0-generate-002',
+            prompt=image_prompt,
+            config=types.GenerateImagesConfig(
+                number_of_images=1,
+                aspect_ratio="9:16",
+                person_generation="DONT_ALLOW"
+            )
         )
-    )
-
-    if image_response.generated_images and len(image_response.generated_images) > 0:
-        generated_image = image_response.generated_images[0]
-        image = Image.open(BytesIO(generated_image.image.image_bytes))
-    else:
+        if image_response.generated_images and len(image_response.generated_images) > 0:
+            generated_image = image_response.generated_images[0]
+            image = Image.open(BytesIO(generated_image.image.image_bytes))
+        else:
+            image = Image.new('RGB', (360, 640), color='gray')
+    except Exception as e:
+        print(f"Error generating image: {e}")
         image = Image.new('RGB', (360, 640), color='gray')
 
     buffered = BytesIO()
@@ -104,7 +108,7 @@ def generate_item(tag, ideas):
 
     return {'text': text, 'image_base64': img_str, 'ideas': ideas}
 
-def start_feed(tag, current_index, feed_items, is_loading):
+def start_feed(tag, current_index, feed_items):
     """
     Start or update the feed based on the tag.
     
@@ -112,7 +116,6 @@ def start_feed(tag, current_index, feed_items, is_loading):
         tag (str): The tag to generate content for.
         current_index (int): The current item index.
         feed_items (list): The current list of feed items.
-        is_loading (bool): Whether the feed is currently loading.
     
     Returns:
         tuple: (current_tag, current_index, feed_items, html_content, is_loading)
@@ -120,18 +123,25 @@ def start_feed(tag, current_index, feed_items, is_loading):
     if not tag.strip():
         tag = "trending"
     
+    # Set loading state
     is_loading = True
     yield tag, current_index, feed_items, generate_html([], False, 0, tag, is_loading), is_loading
 
-    ideas = generate_ideas(tag)
-    item = generate_item(tag, ideas)
-    feed_items = [item]
-    current_index = 0
+    try:
+        ideas = generate_ideas(tag)
+        item = generate_item(tag, ideas)
+        feed_items = [item]
+        current_index = 0
+    except Exception as e:
+        print(f"Error in start_feed: {e}")
+        feed_items = []
+        current_index = 0
 
+    # Set loading state to False and update UI
     is_loading = False
     return tag, current_index, feed_items, generate_html(feed_items, False, current_index, tag, is_loading), is_loading
 
-def load_next(tag, current_index, feed_items, is_loading):
+def load_next(tag, current_index, feed_items):
     """
     Load the next item in the feed.
     
@@ -139,7 +149,6 @@ def load_next(tag, current_index, feed_items, is_loading):
         tag (str): The tag to generate content for.
         current_index (int): The current item index.
         feed_items (list): The current list of feed items.
-        is_loading (bool): Whether the feed is currently loading.
     
     Returns:
         tuple: (current_tag, current_index, feed_items, html_content, is_loading)
@@ -147,18 +156,22 @@ def load_next(tag, current_index, feed_items, is_loading):
     is_loading = True
     yield tag, current_index, feed_items, generate_html(feed_items, False, current_index, tag, is_loading), is_loading
 
-    if current_index + 1 < len(feed_items):
-        current_index += 1
-    else:
-        ideas = feed_items[-1]['ideas'] if feed_items else generate_ideas(tag)
-        new_item = generate_item(tag, ideas)
-        feed_items.append(new_item)
-        current_index = len(feed_items) - 1
+    try:
+        if current_index + 1 < len(feed_items):
+            current_index += 1
+        else:
+            ideas = feed_items[-1]['ideas'] if feed_items else generate_ideas(tag)
+            new_item = generate_item(tag, ideas)
+            feed_items.append(new_item)
+            current_index = len(feed_items) - 1
+    except Exception as e:
+        print(f"Error in load_next: {e}")
+        current_index = max(0, current_index)
 
     is_loading = False
     return tag, current_index, feed_items, generate_html(feed_items, False, current_index, tag, is_loading), is_loading
 
-def load_previous(tag, current_index, feed_items, is_loading):
+def load_previous(tag, current_index, feed_items):
     """
     Load the previous item in the feed.
     
@@ -166,14 +179,13 @@ def load_previous(tag, current_index, feed_items, is_loading):
         tag (str): The tag to generate content for.
         current_index (int): The current item index.
         feed_items (list): The current list of feed items.
-        is_loading (bool): Whether the feed is currently loading.
     
     Returns:
         tuple: (current_tag, current_index, feed_items, html_content, is_loading)
     """
     if current_index > 0:
         current_index -= 1
-    return tag, current_index, feed_items, generate_html(feed_items, False, current_index, tag, is_loading), is_loading
+    return tag, current_index, feed_items, generate_html(feed_items, False, current_index, tag, False), False
 
 def generate_html(feed_items, scroll_to_latest=False, current_index=0, tag="", is_loading=False):
     """
@@ -189,7 +201,6 @@ def generate_html(feed_items, scroll_to_latest=False, current_index=0, tag="", i
     Returns:
         str: HTML string representing the feed.
     """
-    # Define loading messages based on the tag
     loading_messages = [
         f"Cooking up a {tag} masterpiece... 🍳",
         f"Snapping a vibrant {tag} moment... 📸",
@@ -392,34 +403,33 @@ with gr.Blocks(
         outputs=tag_input
     ).then(
         fn=start_feed,
-        inputs=[tag_input, current_index, feed_items, is_loading],
+        inputs=[tag_input, current_index, feed_items],
         outputs=[current_tag, current_index, feed_items, feed_html, is_loading]
     )
 
     # Handle Enter keypress in the custom tag input
     tag_input.submit(
         fn=start_feed,
-        inputs=[tag_input, current_index, feed_items, is_loading],
+        inputs=[tag_input, current_index, feed_items],
         outputs=[current_tag, current_index, feed_items, feed_html, is_loading]
     )
 
-    # Hidden buttonsAPPROXIMATELY Hidden buttons for navigation
+    # Hidden buttons for navigation
     next_button = gr.Button("Next", elem_id="next-button", visible=False)
     previous_button = gr.Button("Previous", elem_id="previous-button", visible=False)
 
     # Handle click to go to next item
     next_button.click(
         fn=load_next,
-        inputs=[current_tag, current_index, feed_items, is_loading],
+        inputs=[current_tag, current_index, feed_items],
         outputs=[current_tag, current_index, feed_items, feed_html, is_loading]
     )
 
     # Handle click to go to previous item
     previous_button.click(
         fn=load_previous,
-        inputs=[current_tag, current_index, feed_items, is_loading],
+        inputs=[current_tag, current_index, feed_items],
         outputs=[current_tag, current_index, feed_items, feed_html, is_loading]
     )
 
-# Launch the app with a public link
-demo.launch(share=True)
+demo.launch()