Update app.py

app.py

@@ -636,11 +636,22 @@ with gr.Blocks(theme=gr.themes.Soft(), title="EmotionToAction") as demo:
          outputs=[status_output, plan_display, current_plan_state]
     )
 
-    # Find & Add Resources Button Click (Async)
+    # Find & Add Resources Button Click (Async) - CORRECTED
     async def handle_find_resources(username, current_analysis, faiss_index_s, faiss_metadata_s, progress=gr.Progress(track_tqdm=True)):
-        if not username: return "Please enter username.", faiss_index_s, faiss_metadata_s, "Idle"
-        if not current_analysis: return "Please analyze an entry first.", faiss_index_s, faiss_metadata_s, "Idle"
-        if not CRAWL4AI_AVAILABLE: return "crawl4ai library not installed.", faiss_index_s, faiss_metadata_s, "Error"
+        # --- Check 1 ---
+        if not username:
+            yield "Please enter username.", faiss_index_s, faiss_metadata_s, "Idle"
+            return # Stop execution
+
+        # --- Check 2 ---
+        if not current_analysis:
+            yield "Please analyze an entry first.", faiss_index_s, faiss_metadata_s, "Idle"
+            return # Stop execution
+
+        # --- Check 3 ---
+        if not CRAWL4AI_AVAILABLE:
+            yield "crawl4ai library not installed.", faiss_index_s, faiss_metadata_s, "Error"
+            return # Stop execution
 
         status_msg = "Starting resource finding..."
         yield status_msg, faiss_index_s, faiss_metadata_s, status_msg # Initial update
@@ -653,30 +664,44 @@ with gr.Blocks(theme=gr.themes.Soft(), title="EmotionToAction") as demo:
         status_msg = "Searching web..."
         yield status_msg, faiss_index_s, faiss_metadata_s, status_msg
         urls = get_web_urls(search_term, num_results=3) # Limit URLs
-        if not urls: yield "No relevant URLs found.", faiss_index_s, faiss_metadata_s, "No URLs found."; return
+
+        # --- Check 4 ---
+        if not urls:
+             yield "No relevant URLs found.", faiss_index_s, faiss_metadata_s, "No URLs found."
+             return # Stop execution
 
         progress(0.3, desc=f"Crawling {len(urls)} pages...")
         status_msg = f"Crawling {len(urls)} pages..."
         yield status_msg, faiss_index_s, faiss_metadata_s, status_msg
-        crawled_content = await crawl_webpages_simple(urls) # Async call
-        if not crawled_content: yield "Crawling failed or yielded no content.", faiss_index_s, faiss_metadata_s, "Crawling failed."; return
+        # Use await for the async crawl function
+        crawled_content = await crawl_webpages_simple(urls)
+
+        # --- Check 5 ---
+        if not crawled_content:
+             yield "Crawling failed or yielded no content.", faiss_index_s, faiss_metadata_s, "Crawling failed."
+             return # Stop execution
 
         progress(0.7, desc="Adding content to FAISS index...")
         status_msg = "Adding content to index..."
         yield status_msg, faiss_index_s, faiss_metadata_s, status_msg
-        # Note: add_to_faiss modifies the index/metadata objects in place
-        index_obj = faiss_index_s # Get current index from state
-        meta_obj = faiss_metadata_s # Get current metadata from state
+
+        # Get current index/metadata objects from state to modify them
+        index_obj = faiss_index_s
+        meta_obj = faiss_metadata_s
+        # add_to_faiss modifies the objects in place
         _, _, added_count = add_to_faiss(index_obj, meta_obj, crawled_content)
 
         if added_count > 0:
-             # IMPORTANT: Save the modified index and metadata back to disk
+             # Save the modified index and metadata back to disk
              save_faiss_index(index_obj, meta_obj)
              status_msg = f"Successfully added {added_count} content chunks to the index."
-             yield status_msg, index_obj, meta_obj, status_msg # Return updated state objects
+             # Yield the final update including potentially modified state objects
+             yield status_msg, index_obj, meta_obj, status_msg
         else:
              status_msg = "Crawled content, but failed to add anything to the index."
+             # Yield the final update (state objects might not have changed but yield them anyway)
              yield status_msg, index_obj, meta_obj, status_msg
+        # No explicit return needed here, generator finishes naturally
 
     # Use the wrapper for async function
     find_resources_button.click(