Update app.py

app.py

@@ -125,7 +125,7 @@ def start_feed(tag, current_index, feed_items):
     
     # Set loading state
     is_loading = True
-    yield tag, current_index, feed_items, generate_html([], False, 0, tag, is_loading), is_loading
+    html_content = generate_html([], False, 0, tag, is_loading)
 
     try:
         ideas = generate_ideas(tag)
@@ -136,10 +136,31 @@ def start_feed(tag, current_index, feed_items):
         print(f"Error in start_feed: {e}")
         feed_items = []
         current_index = 0
+        html_content = """
+        <div style="
+            display: flex;
+            flex-direction: column;
+            align-items: center;
+            justify-content: center;
+            max-width: 360px;
+            margin: 0 auto;
+            background-color: #000;
+            height: 640px;
+            border: 1px solid #333;
+            border-radius: 10px;
+            color: white;
+            font-family: Arial, sans-serif;
+        ">
+            <p>Error generating content. Please try again!</p>
+        </div>
+        """
+        is_loading = False
+        return tag, current_index, feed_items, html_content, is_loading
 
     # 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
+    html_content = generate_html(feed_items, False, current_index, tag, is_loading)
+    return tag, current_index, feed_items, html_content, is_loading
 
 def load_next(tag, current_index, feed_items):
     """
@@ -154,7 +175,7 @@ def load_next(tag, current_index, feed_items):
         tuple: (current_tag, current_index, feed_items, html_content, is_loading)
     """
     is_loading = True
-    yield tag, current_index, feed_items, generate_html(feed_items, False, current_index, tag, is_loading), is_loading
+    html_content = generate_html(feed_items, False, current_index, tag, is_loading)
 
     try:
         if current_index + 1 < len(feed_items):
@@ -166,10 +187,30 @@ def load_next(tag, current_index, feed_items):
             current_index = len(feed_items) - 1
     except Exception as e:
         print(f"Error in load_next: {e}")
-        current_index = max(0, current_index)
+        html_content = """
+        <div style="
+            display: flex;
+            flex-direction: column;
+            align-items: center;
+            justify-content: center;
+            max-width: 360px;
+            margin: 0 auto;
+            background-color: #000;
+            height: 640px;
+            border: 1px solid #333;
+            border-radius: 10px;
+            color: white;
+            font-family: Arial, sans-serif;
+        ">
+            <p>Error generating content. Please try again!</p>
+        </div>
+        """
+        is_loading = False
+        return tag, current_index, feed_items, html_content, is_loading
 
     is_loading = False
-    return tag, current_index, feed_items, generate_html(feed_items, False, current_index, tag, is_loading), is_loading
+    html_content = generate_html(feed_items, False, current_index, tag, is_loading)
+    return tag, current_index, feed_items, html_content, is_loading
 
 def load_previous(tag, current_index, feed_items):
     """
@@ -185,7 +226,8 @@ def load_previous(tag, current_index, feed_items):
     """
     if current_index > 0:
         current_index -= 1
-    return tag, current_index, feed_items, generate_html(feed_items, False, current_index, tag, False), False
+    html_content = generate_html(feed_items, False, current_index, tag, False)
+    return tag, current_index, feed_items, html_content, False
 
 def generate_html(feed_items, scroll_to_latest=False, current_index=0, tag="", is_loading=False):
     """