Update app.py

app.py

@@ -11,7 +11,7 @@ from dateutil import parser
 import json
 import threading
 from concurrent.futures import ThreadPoolExecutor, TimeoutError, as_completed
-import signal
+
 
 # Initialize Anthropic client
 client = anthropic.Anthropic(
@@ -31,6 +31,7 @@ def check_cancellation():
 
 # === Enhanced Model functions with progress tracking ===
 
+
 def extract_publication_date(soup, url):
     """Extract publication date from article HTML - same as before"""
     try:
@@ -405,6 +406,7 @@ Text:
             progress(0.5, desc="Sending request to AI model...")
             
         message = client.messages.create(
+            # The model name is left as you provided.
             model="claude-sonnet-4-20250514",
             max_tokens=1500,
             temperature=0.1,
@@ -444,8 +446,9 @@ def search_only_enhanced(name: str, article_count: int, progress=gr.Progress()):
         
         articles_output = search_articles_enhanced(name.strip(), max_articles=article_count, progress=progress)
         
+        # This now correctly handles cancellation from the main thread
         if "[CANCELLED]" in articles_output:
-            return articles_output, ""
+            return "🛑 Search was cancelled by user.", ""
             
         elapsed = time.time() - start
         progress(1.0, desc=f"Search completed in {elapsed:.1f}s")
@@ -537,24 +540,29 @@ with gr.Blocks(title="Enhanced Founder Finder", theme=gr.themes.Soft()) as demo:
                 show_copy_button=True
             )
 
-    # Event handlers
-    search_click = search_btn.click(
+    # --- FIX 2: Implement proper Gradio event cancellation ---
+    # The search_btn.click event is assigned to a variable.
+    search_event = search_btn.click(
         fn=search_only_enhanced,
         inputs=[name_input, article_count_slider],
         outputs=[output1, search_state],
-        show_progress=True
+        show_progress="full" # Use "full" for better progress bar experience
     )
     
+    # The cancel_btn.click event now correctly cancels the search_event.
     cancel_btn.click(
         fn=cancel_search,
-        outputs=[status_output]
+        inputs=None,
+        outputs=[status_output],
+        # This is the crucial argument that links the cancel button to the search event.
+        cancels=[search_event]
     )
 
     extract_btn.click(
         fn=extract_only_enhanced,
         inputs=[search_state, name_input],
         outputs=[output2],
-        show_progress=True
+        show_progress="full"
     )
 
     # Add some example companies
@@ -569,11 +577,15 @@ with gr.Blocks(title="Enhanced Founder Finder", theme=gr.themes.Soft()) as demo:
         inputs=[name_input, article_count_slider],
     )
 
+# --- FIX 1: Enable the queue. This is essential for long-running tasks on Hugging Face Spaces. ---
+demo.queue()
+
 if __name__ == "__main__":
     demo.launch(
         share=False,
         show_error=True
     )
+
 '''
 import gradio as gr 
 import requests