Update app.py

app.py

@@ -19,21 +19,21 @@ def search_hub(query: str, search_type: str) -> pd.DataFrame:
         data = []
     return pd.DataFrame(data)
 
-def open_url(row):
-    if row is not None and not row.empty:
-        url = row.iloc[0]['link']
+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>'
     else:
         return ""
 
-def load_metadata(row, search_type):
-    if row is not None and not row.empty:
-        item_id = row.iloc[0]['id']
+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):
+        item_id = df.iloc[evt.index[0]]['id']
         
         if search_type == "Models":
             try:
                 card = ModelCard.load(item_id)
-                return card
+                return str(card)
             except Exception as e:
                 return f"Error loading model card: {str(e)}"
         elif search_type == "Datasets":
@@ -95,7 +95,7 @@ with gr.Blocks() as demo:
         return df, aggregated
 
     search_button.click(search_and_aggregate, inputs=[search_query, search_type], outputs=[results_df, aggregated_output])
-    results_df.select(open_url, outputs=[web_view])
+    results_df.select(open_url, inputs=[results_df], outputs=[web_view])
     results_df.select(load_metadata, inputs=[results_df, search_type], outputs=[metadata_output])
 
 demo.launch(debug=True)