making things slightly better when repo is gated

app.py

@@ -43,9 +43,12 @@ def repo_size(r_type, r_id):
                 timeout=1000,
             )
             response = response.json()
+            # get the status code
         except Exception:
             response = {}
-
+        if response.get("error") and "restricted" in response.get("error"):
+            gr.Warning(f"Branch information for {r_id} not available.")
+            return {}
         size = response.get("size")
         if size is not None:
             repo_sizes[branch.name] = size
@@ -71,10 +74,20 @@ def get_repo_info(r_type, r_id):
         .T.reset_index(names="ext")
         .sort_values(by="size", ascending=False)
     )
-    r_sizes_df = pd.DataFrame(repo_sizes, index=["size"]).T.reset_index(names="branch")
-    r_sizes_df["formatted_size"] = r_sizes_df["size"].apply(format_repo_size)
+    # check if repo_sizes is just {}
+    if not repo_sizes:
+        r_sizes_component = gr.Dataframe(visible=False)
+    else:
+        r_sizes_df = pd.DataFrame(repo_sizes, index=["size"]).T.reset_index(
+            names="branch"
+        )
+        r_sizes_df["formatted_size"] = r_sizes_df["size"].apply(format_repo_size)
+        r_sizes_df.columns = ["Branch", "bytes", "Size"]
+        r_sizes_component = gr.Dataframe(
+            value=r_sizes_df[["Branch", "Size"]], visible=True
+        )
+
     rf_sizes_df["formatted_size"] = rf_sizes_df["size"].apply(format_repo_size)
-    r_sizes_df.columns = ["Branch", "bytes", "Size"]
     rf_sizes_df.columns = ["Extension", "bytes", "Count", "Size"]
     rf_sizes_plot = px.pie(
         rf_sizes_df,
@@ -91,7 +104,7 @@ def get_repo_info(r_type, r_id):
             visible=True,
         ),
         gr.Plot(rf_sizes_plot, visible=True),
-        gr.Dataframe(value=r_sizes_df[["Branch", "Size"]], visible=True),
+        r_sizes_component,
     )