Update app.py

app.py

@@ -11,12 +11,15 @@ anthropic_client = anthropic.Anthropic(api_key=os.getenv("ANTHROPIC_API_KEY"))
 
 # === 1. Simplified Search Workflow ===
 
-def search_workflow(name: str, progress=gr.Progress()):
+def search_workflow(name: str, api_key: str, progress=gr.Progress()):
     """
     A simple function to search for articles, fetching exactly 8 news articles: 4 recent, 4 historical.
     """
     if not name or not name.strip():
         return "❌ Please enter a company name.", ""
+    
+    if not api_key or not api_key.strip():
+        return "❌ Please enter your Anthropic API key.", ""
 
     progress(0, desc="Starting search...")
     
@@ -77,12 +80,15 @@ def search_workflow(name: str, progress=gr.Progress()):
 
 # === 2. Simplified Extraction Workflow ===
 
-def extraction_workflow(raw_text: str, company_name: str, progress=gr.Progress()):
+def extraction_workflow(raw_text: str, company_name: str, api_key: str, progress=gr.Progress()):
     """
     A simple and robust tool to extract founders from text using the AI model.
     """
     if not raw_text or not raw_text.strip():
         return "❌ Please run a search first to get text to analyze."
+    
+    if not api_key or not api_key.strip():
+        return "❌ Please enter your Anthropic API key."
 
     progress(0, desc="Preparing prompt for AI...")
     
@@ -98,7 +104,11 @@ ARTICLES:
 """
     try:
         progress(0.5, desc="Sending request to AI model...")
-        message = anthropic_client.messages.create(
+        
+        # Create client with user's API key
+        client = anthropic.Anthropic(api_key=api_key)
+        
+        message = client.messages.create(
             model="claude-sonnet-4-20250514", # As requested
             max_tokens=1024,
             temperature=0.0,
@@ -135,30 +145,40 @@ ARTICLES:
 
 with gr.Blocks(title="Founder Name Extraction Tool", theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🔎 Founder Name Extraction")
-    gr.Markdown("A tool to find company founders. **Step 1:** Search for articles. **Step 2:** Extract founders' names from the results.")
+    gr.Markdown("A tool to find company founders. **Step 1:** Enter your API key and company name. **Step 2:** Search for articles. **Step 3:** Extract founders' names from the results.")
     
     # Hidden state to pass text from search to extraction
     search_results_for_ai = gr.State("")
     
     with gr.Row():
-        name_input = gr.Textbox(label="Company Name", placeholder="e.g., 'OpenAI', 'SpaceX'", scale=3)
-        search_btn = gr.Button("1. 🔍 Search for Articles", variant="primary", scale=1)
+        api_key_input = gr.Textbox(
+            label="Anthropic API Key", 
+            placeholder="sk-ant-...", 
+            type="password",
+            scale=2
+        )
+        name_input = gr.Textbox(
+            label="Company Name", 
+            placeholder="e.g., 'OpenAI', 'SpaceX'", 
+            scale=2
+        )
+        search_btn = gr.Button("2. 🔍 Search for Articles", variant="primary", scale=1)
     
     with gr.Row():
-        extract_btn = gr.Button("2. 📊 Extract Founders from Search Results", variant="secondary")
+        extract_btn = gr.Button("3. 📊 Extract Founders from Search Results", variant="secondary")
     
     # Display both sections without tabs
     gr.Markdown("### Search Results")
     output_search = gr.Markdown()
     
     gr.Markdown("### Founder Intelligence Report")
-    output_extract = gr.Markdown(value="*Waiting for extraction...*")  # Shows placeholder text
+    output_extract = gr.Markdown(value="*Waiting for extraction...*")
     
     # --- Event Wiring ---
     # Search button populates the search results and the hidden state
     search_btn.click(
         fn=search_workflow,
-        inputs=[name_input],
+        inputs=[name_input, api_key_input],  # Added api_key_input
         outputs=[output_search, search_results_for_ai],
         show_progress="full"
     )
@@ -166,7 +186,7 @@ with gr.Blocks(title="Founder Name Extraction Tool", theme=gr.themes.Soft()) as
     # Extract button uses the hidden state to populate the extraction
     extract_btn.click(
         fn=extraction_workflow,
-        inputs=[search_results_for_ai, name_input],
+        inputs=[search_results_for_ai, name_input, api_key_input],  # Added api_key_input
         outputs=[output_extract],
         show_progress="full"
     )