Update app.py

app.py

@@ -64,34 +64,30 @@ Provide full names with their roles/titles where mentioned."""
 # --- Pipeline Function with Progress & Status ---
 def process_name_with_progress(name: str, progress=gr.Progress(track_tqdm=True)):
     if not name.strip():
-        yield "", "", "", "Please enter a name."
-
-    search_results = ""
-    entities = ""
-    full_names = ""
+        return "", "", "", "Please enter a name."
 
     try:
         progress(0.1)
-        status = "🔍 Searching for articles..."
+        yield "", "", "", "🔍 Searching for articles..."
+        
         search_results = search_articles(name.strip())
-        yield search_results, "", "", status
-
         progress(0.4)
-        status = "📄 Extracting entities from articles..."
-        entities = extract_entities(search_results)
-        yield search_results, entities, "", status
+        yield search_results, "", "", "📄 Extracting entities from articles..."
 
+        entities = extract_entities(search_results)
         progress(0.7)
-        status = "🧠 Finding full names and roles..."
-        full_names = find_full_names(search_results, entities)
-        yield search_results, entities, full_names, status
+        yield search_results, entities, "", "🧠 Finding full names and roles..."
 
+        full_names = find_full_names(search_results, entities)
         progress(1.0)
         yield search_results, entities, full_names, "✅ Complete!"
 
     except Exception as e:
         err = f"❌ Error: {str(e)}"
-        yield search_results or err, entities or err, full_names or err, err
+        yield search_results if 'search_results' in locals() else "", \
+              entities if 'entities' in locals() else "", \
+              full_names if 'full_names' in locals() else "", \
+              err
 
 # --- Gradio Interface ---
 with gr.Blocks(title="Name Research Tool") as demo:
@@ -100,9 +96,7 @@ with gr.Blocks(title="Name Research Tool") as demo:
 
     with gr.Row():
         name_input = gr.Textbox(label="Name", placeholder="Enter business or project name")
-        with gr.Column():
-            search_btn = gr.Button("Search (Real-time)", variant="primary")
-            debug_btn = gr.Button("Search (Debug Mode)", variant="secondary")
+        search_btn = gr.Button("Search", variant="primary")
 
     with gr.Column():
         output1 = gr.Textbox(label="Search Results", lines=10, max_lines=20)
@@ -110,7 +104,7 @@ with gr.Blocks(title="Name Research Tool") as demo:
         output3 = gr.Textbox(label="Full Names", lines=5, max_lines=10)
         status_output = gr.Textbox(label="Status / Progress", lines=1, interactive=False)
 
-    # Real-time search with progress
+    # Search with progress
     search_btn.click(
         fn=process_name_with_progress,
         inputs=[name_input],
@@ -119,4 +113,3 @@ with gr.Blocks(title="Name Research Tool") as demo:
 
 if __name__ == "__main__":
     demo.launch()
-