Update app.py

app.py

@@ -86,14 +86,25 @@ def create_ui() -> gr.Blocks:
         
         with gr.Row():
             with gr.Column():
-                # Input Section
-                gr.Markdown("### Enter Repository Information")
-                repo_input = gr.Textbox(
-                    label="Enter repo IDs (comma or newline separated) or keywords to search",
-                    lines=5,
-                    placeholder="Enter repository IDs or keywords to search"
+                # Repository ID Input Section
+                gr.Markdown("### Enter Repository IDs")
+                repo_id_input = gr.Textbox(
+                    label="Enter repository IDs (comma or newline separated)",
+                    lines=3,
+                    placeholder="repo1, repo2\nrepo3"
                 )
-                submit_btn = gr.Button("Submit", variant="primary")
+                submit_repo_btn = gr.Button("Submit Repository IDs", variant="primary")
+                
+                # Keyword Search Section
+                gr.Markdown("### Or Search by Keywords")
+                keyword_input = gr.Textbox(
+                    label="Enter keywords to search",
+                    lines=2,
+                    placeholder="Enter keywords separated by commas"
+                )
+                search_btn = gr.Button("Search by Keywords", variant="primary")
+                
+                # Status
                 status = gr.Textbox(label="Status", visible=True)
                 
                 # Results Section
@@ -117,25 +128,46 @@ def create_ui() -> gr.Blocks:
                     send_btn = gr.Button("Send", variant="primary")
                     clear_btn = gr.Button("Clear Chat", variant="secondary")
         
-        def process_input(text: str) -> Tuple[pd.DataFrame, str, str, str]:
-            """Process input and return results."""
+        def process_repo_ids(text: str) -> Tuple[pd.DataFrame, str, str, str]:
+            """Process repository IDs and return results."""
+            try:
+                repo_ids = [rid.strip() for rid in re.split(r'[\n,]+', text) if rid.strip()]
+                
+                if not repo_ids:
+                    return pd.DataFrame(), "No repository IDs provided", "", ""
+                
+                # Remove duplicates
+                repo_ids = list(dict.fromkeys(repo_ids))
+                
+                # Update CSV
+                write_repos_to_csv(repo_ids)
+                
+                # Get first repo analysis
+                content, summary = analyze_repo(repo_ids[0])
+                
+                return read_csv_as_text(CSV_FILE), f"Found {len(repo_ids)} repositories", content, summary
+                
+            except Exception as e:
+                logger.error(f"Error processing repository IDs: {e}")
+                return pd.DataFrame(), f"Error: {str(e)}", "", ""
+        
+        def process_keywords(text: str) -> Tuple[pd.DataFrame, str, str, str]:
+            """Process keywords and return search results."""
             try:
-                # Check if input is keywords or repo IDs
-                if any(kw in text.lower() for kw in ['search', 'find', 'look for']):
-                    # Handle as keyword search
-                    keywords = [k.strip() for k in re.split(r'[\n,]+', text) if k.strip()]
-                    repo_ids = []
-                    for kw in keywords:
-                        repo_ids.extend(search_top_spaces(kw, limit=5))
-                else:
-                    # Handle as repo IDs
-                    repo_ids = [rid.strip() for rid in re.split(r'[\n,]+', text) if rid.strip()]
+                keywords = [k.strip() for k in re.split(r'[\n,]+', text) if k.strip()]
+                
+                if not keywords:
+                    return pd.DataFrame(), "No keywords provided", "", ""
+                
+                repo_ids = []
+                for kw in keywords:
+                    repo_ids.extend(search_top_spaces(kw, limit=5))
                 
                 # Remove duplicates
                 repo_ids = list(dict.fromkeys(repo_ids))
                 
                 if not repo_ids:
-                    return pd.DataFrame(), "No repositories found", "", ""
+                    return pd.DataFrame(), "No repositories found for the given keywords", "", ""
                 
                 # Update CSV
                 write_repos_to_csv(repo_ids)
@@ -146,7 +178,7 @@ def create_ui() -> gr.Blocks:
                 return read_csv_as_text(CSV_FILE), f"Found {len(repo_ids)} repositories", content, summary
                 
             except Exception as e:
-                logger.error(f"Error processing input: {e}")
+                logger.error(f"Error processing keywords: {e}")
                 return pd.DataFrame(), f"Error: {str(e)}", "", ""
         
         def send_message(message: str, history: List[Dict[str, str]]) -> Tuple[List[Dict[str, str]], str]:
@@ -163,9 +195,15 @@ def create_ui() -> gr.Blocks:
             return [], ""
         
         # Event handlers
-        submit_btn.click(
-            fn=process_input,
-            inputs=[repo_input],
+        submit_repo_btn.click(
+            fn=process_repo_ids,
+            inputs=[repo_id_input],
+            outputs=[df_output, status, content_output, summary_output]
+        )
+        
+        search_btn.click(
+            fn=process_keywords,
+            inputs=[keyword_input],
             outputs=[df_output, status, content_output, summary_output]
         )