From 19a4460d904eb585fb19a4595ebd343b731ee6dd Mon Sep 17 00:00:00 2001
From: Philipp Horstenkamp <philipp@horstenkamp.de>
Date: Sun, 12 Nov 2023 14:14:15 +0100
Subject: [PATCH] Added a small stability fix. (#374)

The current code has problems with an empty db.
---
 src/aki_prj23_transparenzregister/ui/pages/home.py        | 8 ++++++--
 .../utils/networkx/networkx_data.py                       | 5 +----
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/src/aki_prj23_transparenzregister/ui/pages/home.py b/src/aki_prj23_transparenzregister/ui/pages/home.py
index c20e8e3..93d5400 100644
--- a/src/aki_prj23_transparenzregister/ui/pages/home.py
+++ b/src/aki_prj23_transparenzregister/ui/pages/home.py
@@ -134,8 +134,12 @@ def layout() -> list[html]:
     """Generates the Layout of the Homepage."""
     person_relation_types = person_relation_type_filter()
     company_relation_types = company_relation_type_filter()
-    selected_company_relation_types: frozenset[str] = frozenset(
-        {company_relation_types[1]}
+    selected_company_relation_types: frozenset[str] = (
+        frozenset(
+            {company_relation_types[1]} if company_relation_types else frozenset({})
+        )
+        if company_relation_types
+        else frozenset({})
     )
     selected_person_relation_types: frozenset[str] = frozenset(
         {}
diff --git a/src/aki_prj23_transparenzregister/utils/networkx/networkx_data.py b/src/aki_prj23_transparenzregister/utils/networkx/networkx_data.py
index 4d20789..9697cfd 100644
--- a/src/aki_prj23_transparenzregister/utils/networkx/networkx_data.py
+++ b/src/aki_prj23_transparenzregister/utils/networkx/networkx_data.py
@@ -3,7 +3,6 @@ from functools import lru_cache
 
 import networkx as nx
 import pandas as pd
-from loguru import logger
 from sqlalchemy.orm import aliased
 
 from aki_prj23_transparenzregister.ui.session_handler import SessionHandler
@@ -155,11 +154,9 @@ def filter_relation_type(
         relation_dataframe (pd.DataFrame): The filtered DataFrame which now only contains entries with the selected Relation Type.
     """
     if selected_relation_type is not None:
-        filtered = relation_dataframe.loc[
+        return relation_dataframe.loc[
             relation_dataframe["relation_type"].isin(selected_relation_type)
         ]
-        logger.info(f"Filtered! {len(filtered.index) / len(relation_dataframe)}")
-        return filtered
     return relation_dataframe