Update app.py

app.py

@@ -303,6 +303,9 @@ def start_feed(user_input, generate_video, current_index, feed_items):
     if not user_input.strip():
         user_input = "trending"
     
+    # Update current_user_input with the new user_input
+    current_user_input = user_input
+    
     # Set loading state
     is_loading = True
     html_content = generate_html([], False, 0, user_input, is_loading)
@@ -341,19 +344,19 @@ def start_feed(user_input, generate_video, current_index, feed_items):
         </div>
         """
         is_loading = False
-        return user_input, current_index, feed_items, html_content, share_links, is_loading
+        return current_user_input, current_index, feed_items, html_content, share_links, is_loading
 
     # Set loading state to False and update UI
     is_loading = False
     html_content = generate_html(feed_items, False, current_index, user_input, is_loading)
-    return user_input, current_index, feed_items, html_content, share_links, is_loading
+    return current_user_input, current_index, feed_items, html_content, share_links, is_loading
 
 def load_next(user_input, generate_video, current_index, feed_items):
     """
     Load the next item in the feed.
     
     Args:
-        user_input (str): The user's input concept or idea.
+        user_input (str): The user's input concept or idea (updated from the textbox).
         generate_video (bool): Whether to generate a video.
         current_index (int): The current item index.
         feed_items (list): The current list of feed items.
@@ -361,6 +364,12 @@ def load_next(user_input, generate_video, current_index, feed_items):
     Returns:
         tuple: (current_user_input, current_index, feed_items, html_content, share_links, is_loading)
     """
+    # Update current_user_input with the latest user_input from the textbox
+    current_user_input = user_input if user_input.strip() else "trending"
+    
+    # Use current_user_input for generating the next item
+    user_input = current_user_input
+    
     is_loading = True
     html_content = generate_html(feed_items, False, current_index, user_input, is_loading)
     share_links = ""
@@ -399,11 +408,11 @@ def load_next(user_input, generate_video, current_index, feed_items):
         </div>
         """
         is_loading = False
-        return user_input, current_index, feed_items, html_content, share_links, is_loading
+        return current_user_input, current_index, feed_items, html_content, share_links, is_loading
 
     is_loading = False
     html_content = generate_html(feed_items, False, current_index, user_input, is_loading)
-    return user_input, current_index, feed_items, html_content, share_links, is_loading
+    return current_user_input, current_index, feed_items, html_content, share_links, is_loading
 
 def load_previous(user_input, generate_video, current_index, feed_items):
     """
@@ -418,6 +427,9 @@ def load_previous(user_input, generate_video, current_index, feed_items):
     Returns:
         tuple: (current_user_input, current_index, feed_items, html_content, share_links, is_loading)
     """
+    # Update current_user_input with the latest user_input
+    current_user_input = user_input if user_input.strip() else "trending"
+    
     if current_index > 0:
         current_index -= 1
     html_content = generate_html(feed_items, False, current_index, user_input, False)
@@ -426,7 +438,7 @@ def load_previous(user_input, generate_video, current_index, feed_items):
         feed_items[current_index]['video_base64'],
         feed_items[current_index]['text']
     )
-    return user_input, current_index, feed_items, html_content, share_links, False
+    return current_user_input, current_index, feed_items, html_content, share_links, False
 
 def generate_share_links(image_base64, video_base64, caption):
     """
@@ -812,7 +824,7 @@ with gr.Blocks(
     # Handle magic button click to generate next item
     magic_button.click(
         fn=load_next,
-        inputs=[current_user_input, generate_video_checkbox, current_index, feed_items],
+        inputs=[user_input, generate_video_checkbox, current_index, feed_items],  # Use user_input directly from textbox
         outputs=[current_user_input, current_index, feed_items, feed_html, share_html, is_loading]
     )
 
@@ -822,7 +834,7 @@ with gr.Blocks(
     # Handle click to go to previous item
     previous_button.click(
         fn=load_previous,
-        inputs=[current_user_input, generate_video_checkbox, current_index, feed_items],
+        inputs=[user_input, generate_video_checkbox, current_index, feed_items],
         outputs=[current_user_input, current_index, feed_items, feed_html, share_html, is_loading]
     )