Update app.py

app.py

@@ -125,7 +125,7 @@ def search_articles(name: str, max_articles: int = 2) -> str:
 
     return f"[INFO] No articles found for {name}"
 
-def extract_entities(search_results: str) -> str:
+def extract_entities(search_results: str, company_name: str) -> str:
     """Extract entities using Claude 4"""
     MAX_CHARS = 8000
     if len(search_results) > MAX_CHARS:
@@ -133,11 +133,13 @@ def extract_entities(search_results: str) -> str:
         last_period = trunc.rfind('. ')
         search_results = trunc[:last_period + 1] if last_period > 3000 else trunc
 
-    prompt = f"""Extract all named entities that are described as founders of the searched business from the following text.
+    prompt = f"""Extract all named entities that are described as founders of "{company_name}" specifically from the following text.
+Only include founders who are explicitly mentioned as founders of {company_name}. 
+Ignore founders of other companies that may be mentioned in the text.
 Return a JSON object with the following two keys:
-- "people": a list of names of people mentioned
+- "people": a list of names of people mentioned as founders of {company_name}
 - "organizations": a list of organization names mentioned
-Respond only with valid JSON. Do not include any explanations, comments, or additional formatting. Double check that you included all founders. Double check that the founders you included are indeed founders of the business described in the search.
+Respond only with valid JSON. Do not include any explanations, comments, or additional formatting. Double check that you included all founders. Double check that the founders you included are indeed founders of {company_name}.
 Text:
 {search_results}"""
 
@@ -176,13 +178,16 @@ def search_only(name: str, article_count: int):
     except Exception as e:
         return f"[ERROR] Search failed: {str(e)}", ""
 
-def extract_only(stored_results: str):
+def extract_only(stored_results: str, company_name: str):
     if not stored_results.strip():
         return "No search results available. Please search first."
+    
+    if not company_name.strip():
+        return "No company name provided. Please search first."
 
     try:
         start = time.time()
-        entities = extract_entities(stored_results)
+        entities = extract_entities(stored_results, company_name.strip())
         elapsed = time.time() - start
         return f"✅ Extraction completed in {elapsed:.1f}s\n\n{entities}"
     except Exception as e:
@@ -220,13 +225,11 @@ with gr.Blocks(title="Founder Finder") as demo:
 
     extract_btn.click(
         fn=extract_only,
-        inputs=[search_state],
+        inputs=[search_state, name_input],
         outputs=[output2]
     )
 
 if __name__ == "__main__":
     demo.launch()
 
-
-