Update app.py

app.py

@@ -400,8 +400,39 @@ def create_ui() -> gr.Blocks:
             
         def keyword_search_with_status(keyword: str, state: AppState) -> Tuple[pd.DataFrame, str]:
             """Search keywords with status update."""
-            df = keyword_search_and_update(keyword, state)
-            return df, ""
+            try:
+                if not keyword:
+                    return pd.DataFrame(columns=["repo id", "strength", "weaknesses", "speciality", "relevance rating"]), ""
+                    
+                keyword_list = [k.strip() for k in re.split(r'[\n,]+', keyword) if k.strip()]
+                repo_ids = []
+                
+                for kw in keyword_list:
+                    try:
+                        results = search_top_spaces(kw, limit=5)
+                        repo_ids.extend(results)
+                    except Exception as e:
+                        logger.error(f"Error searching for keyword {kw}: {e}")
+                        continue
+                
+                # Remove duplicates while preserving order
+                seen = set()
+                unique_repo_ids = []
+                for rid in repo_ids:
+                    if rid not in seen:
+                        unique_repo_ids.append(rid)
+                        seen.add(rid)
+                
+                state.repo_ids = unique_repo_ids
+                state.current_repo_idx = 0
+                
+                write_repos_to_csv(unique_repo_ids)
+                df = read_csv_as_text("repo_ids.csv")
+                return df, ""
+                
+            except Exception as e:
+                logger.error(f"Error in keyword search: {e}")
+                return pd.DataFrame(columns=["repo id", "strength", "weaknesses", "speciality", "relevance rating"]), f"Error: {str(e)}"
             
         def analyze_with_status(state: AppState) -> Tuple[str, str, pd.DataFrame, str]:
             """Analyze with status update."""
@@ -441,12 +472,20 @@ def create_ui() -> gr.Blocks:
             outputs=[df_output, submit_status]
         )
         
+        def search_click():
+            """Handle search button click."""
+            return "Searching..."
+            
+        def search_complete(keyword: str, state: AppState):
+            """Complete search operation."""
+            return keyword_search_with_status(keyword, state)
+            
         search_btn.click(
-            fn=lambda: "Searching...",
+            fn=search_click,
             inputs=[],
             outputs=[search_status]
         ).then(
-            fn=keyword_search_with_status,
+            fn=search_complete,
             inputs=[keyword_input, state],
             outputs=[df_output, search_status]
         )