Update app.py

app.py

@@ -25,7 +25,6 @@ def search_hub(query: str, search_type: str) -> pd.DataFrame:
     else:
         data = []
     
-    # Add numbering and format the link
     for i, item in enumerate(data, 1):
         item['number'] = i
         item['formatted_link'] = format_link(item, i, search_type)
@@ -64,38 +63,38 @@ async def download_readme(session: aiohttp.ClientSession, item: Dict) -> tuple[s
     except Exception as e:
         return item_id.replace('/', '_'), f"# Error downloading README for {item_id}\nError: {str(e)}"
 
-async def download_all_readmes(data: List[Dict]) -> str:
+async def download_all_readmes(data: List[Dict]) -> tuple[str, str]:
     """Download all README files and create a zip archive."""
+    if not data:
+        return "", "No results to download"
+        
     zip_buffer = io.BytesIO()
+    status_message = "Downloading READMEs..."
     
     async with aiohttp.ClientSession() as session:
-        # Download all READMEs concurrently
         tasks = [download_readme(session, item) for item in data]
         results = await asyncio.gather(*tasks)
         
-        # Create zip file
         with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zip_file:
             for filename, content in results:
                 zip_file.writestr(f"{filename}.md", content)
     
-    # Convert to base64
     zip_buffer.seek(0)
     base64_zip = base64.b64encode(zip_buffer.getvalue()).decode()
-    return base64_zip
-
-def create_download_link(base64_zip: str) -> str:
-    """Create an HTML download link for the zip file."""
+    
     download_link = f"""
-    <a href="data:application/zip;base64,{base64_zip}" 
-       download="readmes.zip" 
-       style="display: inline-block; padding: 10px 20px; 
-              background-color: #4CAF50; color: white; 
-              text-decoration: none; border-radius: 5px;
-              margin-top: 10px;">
-        📥 Download READMEs Archive
-    </a>
+    <div style="margin-top: 10px;">
+        <a href="data:application/zip;base64,{base64_zip}" 
+           download="readmes.zip" 
+           style="display: inline-block; padding: 10px 20px; 
+                  background-color: #4CAF50; color: white; 
+                  text-decoration: none; border-radius: 5px;">
+            📥 Download READMEs Archive
+        </a>
+    </div>
     """
-    return download_link
+    
+    return download_link, "READMEs ready for download!"
 
 def display_results(df: pd.DataFrame):
     if df is not None and not df.empty:
@@ -138,51 +137,52 @@ with gr.Blocks() as demo:
         search_button = gr.Button("Search")
     
     results_html = gr.HTML(label="Search Results")
-    download_html = gr.HTML(label="Download Link")
+    download_button = gr.Button("📚 Download All READMEs", visible=False)
+    download_status = gr.Markdown("", label="Download Status")
+    download_area = gr.HTML("", label="Download Link")
     metadata_output = gr.Textbox(label="Metadata", lines=10)
     aggregated_output = gr.JSON(label="Aggregated Content")
     
-    current_results = gr.State([])  # Store current search results
+    current_results = gr.State([])
 
-    async def search_and_aggregate(query, search_type):
+    def search_and_aggregate(query, search_type):
         df = search_hub(query, search_type)
         data = df.to_dict('records')
         aggregated = SwarmyTime(data)
         html_results = display_results(df)
-        
-        # Create download button
-        download_button = """
-        <button onclick="downloadReadmes()" 
-                style="padding: 10px 20px; 
-                       background-color: #4CAF50; 
-                       color: white; 
-                       border: none; 
-                       border-radius: 5px; 
-                       cursor: pointer;">
-            📚 Download All READMEs
-        </button>
-        """
-        
-        return html_results, download_button, aggregated, data
+        show_download = len(data) > 0
+        return [
+            html_results,  # results_html
+            show_download,  # download_button visible
+            "",  # download_status
+            "",  # download_area
+            aggregated,  # aggregated_output
+            data  # current_results
+        ]
 
-    async def download_readmes(data):
+    async def handle_download(data):
         if not data:
-            return "No results to download"
-        
-        base64_zip = await download_all_readmes(data)
-        return create_download_link(base64_zip)
+            return ["No results to download", ""]
+        download_link, status = await download_all_readmes(data)
+        return [status, download_link]
 
     search_button.click(
         search_and_aggregate,
         inputs=[search_query, search_type],
-        outputs=[results_html, download_html, aggregated_output, current_results]
+        outputs=[
+            results_html,
+            download_button,
+            download_status,
+            download_area,
+            aggregated_output,
+            current_results
+        ]
     )
 
-    # Add download button click handler
-    download_html.click(
-        download_readmes,
+    download_button.click(
+        handle_download,
         inputs=[current_results],
-        outputs=[download_html]
+        outputs=[download_status, download_area]
     )
 
 demo.launch(debug=True)