Update app.py

app.py

@@ -17,14 +17,42 @@ def search_hub(query: str, search_type: str) -> pd.DataFrame:
         data = [{"id": space.id, "author": space.author, "link": f"https://huggingface.co/spaces/{space.id}"} for space in results]
     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)
+    
     return pd.DataFrame(data)
 
-def open_url(evt: gr.SelectData, df: pd.DataFrame):
-    if df is not None and not df.empty and evt.index[0] < len(df):
-        url = df.iloc[evt.index[0]]['link']
-        return f'<iframe src="{url}" width="100%" height="600px"></iframe>'
+def format_link(item: Dict, number: int, search_type: str) -> str:
+    link = item['link']
+    readme_link = f"{link}/blob/main/README.md"
+    title = f"{number}. {item['id']}"
+    
+    metadata = f"Author: {item['author']}"
+    if 'downloads' in item:
+        metadata += f", Downloads: {item['downloads']}"
+    
+    html = f"""
+    <div style="margin-bottom: 10px;">
+        <strong>{title}</strong><br>
+        <a href="{link}" target="_blank" style="color: #4a90e2; text-decoration: none;">View {search_type[:-1]}</a> | 
+        <a href="{readme_link}" target="_blank" style="color: #4a90e2; text-decoration: none;">View README</a><br>
+        <small>{metadata}</small>
+    </div>
+    """
+    return html
+
+def display_results(df: pd.DataFrame):
+    if df is not None and not df.empty:
+        html = "<div style='max-height: 400px; overflow-y: auto;'>"
+        for _, row in df.iterrows():
+            html += row['formatted_link']
+        html += "</div>"
+        return html
     else:
-        return ""
+        return "<p>No results found.</p>"
 
 def load_metadata(evt: gr.SelectData, df: pd.DataFrame, search_type: str):
     if df is not None and not df.empty and evt.index[0] < len(df):
@@ -84,18 +112,16 @@ with gr.Blocks() as demo:
         search_query = gr.Textbox(label="Search Query", value="awacke1")
         search_type = gr.Radio(["Models", "Datasets", "Spaces"], label="Search Type", value="Models")
         search_button = gr.Button("Search")
-    results_df = gr.DataFrame(label="Search Results", wrap=True, interactive=True)
-    web_view = gr.HTML(label="Web View")
+    results_html = gr.HTML(label="Search Results")
     metadata_output = gr.Textbox(label="Metadata", lines=10)
     aggregated_output = gr.JSON(label="Aggregated Content")
 
     def search_and_aggregate(query, search_type):
         df = search_hub(query, search_type)
         aggregated = SwarmyTime(df.to_dict('records'))
-        return df, aggregated
+        html_results = display_results(df)
+        return html_results, aggregated
 
-    search_button.click(search_and_aggregate, inputs=[search_query, search_type], outputs=[results_df, aggregated_output])
-    results_df.select(open_url, inputs=[results_df], outputs=[web_view])
-    results_df.select(load_metadata, inputs=[results_df, search_type], outputs=[metadata_output])
+    search_button.click(search_and_aggregate, inputs=[search_query, search_type], outputs=[results_html, aggregated_output])
 
 demo.launch(debug=True)