Update app.py

app.py

@@ -182,96 +182,42 @@ class Search:
 
 import matplotlib.pyplot as plt
 
-def plot_pie_chart(category_similarity_scores):
-    """Plot a pie chart of category similarity scores."""
-    categories = list(category_similarity_scores.keys())
-    similarities = list(category_similarity_scores.values())
-    normalized_similarities = [sim / sum(similarities) for sim in similarities]
-
-    fig, ax = plt.subplots()
-    ax.pie(normalized_similarities, labels=categories, autopct="%1.11f%%", startangle=90)
-    ax.axis('equal')  # Equal aspect ratio ensures the pie chart is circular.
-    plt.show()
-
-
+def plot_pie_chart(category_simiarity_scores):
+    categories = list(category_simiarity_scores.keys())
+    cur_similarities = list(category_simiarity_scores.values())
 
+    similarities = [similar / sum(cur_similarities) for similar in cur_similarities]
 
-
-def plot_piechart(sorted_cosine_scores_items):
-    sorted_cosine_scores = np.array([
-            sorted_cosine_scores_items[index][1]
-            for index in range(len(sorted_cosine_scores_items))
-        ]
-    )
-    categories = st.session_state.categories.split(" ")
-    categories_sorted = [
-        categories[sorted_cosine_scores_items[index][0]]
-        for index in range(len(sorted_cosine_scores_items))
-    ]
     fig, ax = plt.subplots()
-    ax.pie(sorted_cosine_scores, labels=categories_sorted, autopct="%1.1f%%")
-    st.pyplot(fig)  # Figure
+    ax.pie(similarities, labels=categories,
+           autopct="%1.11f%%",
+           startangle=90)
+    ax.axis('equal')
+    plt.show()
 
 
 def plot_piechart_helper(sorted_cosine_scores_items):
-    sorted_cosine_scores = np.array(
-        [
-            sorted_cosine_scores_items[index][1]
-            for index in range(len(sorted_cosine_scores_items))
-        ]
-    )
-    categories = st.session_state.categories.split(" ")
-    categories_sorted = [
-        categories[sorted_cosine_scores_items[index][0]]
-        for index in range(len(sorted_cosine_scores_items))
-    ]
+    sorted_cosine_scores = np.array(list(sorted_cosine_scores_items.values()))
+    categories_sorted = list(sorted_cosine_scores_items.keys())
+
     fig, ax = plt.subplots(figsize=(3, 3))
     my_explode = np.zeros(len(categories_sorted))
     my_explode[0] = 0.2
     if len(categories_sorted) == 3:
-        my_explode[1] = 0.1  # explode this by 0.2
+        my_explode[1] = 0.1
     elif len(categories_sorted) > 3:
         my_explode[2] = 0.05
+
     ax.pie(
         sorted_cosine_scores,
         labels=categories_sorted,
-        autopct="%1.1f%%",
+        autopct="%1.11f%%",
         explode=my_explode,
     )
 
     return fig
 
 
-def plot_piecharts(sorted_cosine_scores_models):
-    scores_list = []
-    categories = st.session_state.categories.split(" ")
-    index = 0
-    for model in sorted_cosine_scores_models:
-        scores_list.append(sorted_cosine_scores_models[model])
-        # scores_list[index] = np.array([scores_list[index][ind2][1] for ind2 in range(len(scores_list[index]))])
-        index += 1
-
-    if len(sorted_cosine_scores_models) == 2:
-        fig, (ax1, ax2) = plt.subplots(2)
-
-        categories_sorted = [
-            categories[scores_list[0][index][0]] for index in range(len(scores_list[0]))
-        ]
-        sorted_scores = np.array(
-            [scores_list[0][index][1] for index in range(len(scores_list[0]))]
-        )
-        ax1.pie(sorted_scores, labels=categories_sorted, autopct="%1.1f%%")
-
-        categories_sorted = [
-            categories[scores_list[1][index][0]] for index in range(len(scores_list[1]))
-        ]
-        sorted_scores = np.array(
-            [scores_list[1][index][1] for index in range(len(scores_list[1]))]
-        )
-        ax2.pie(sorted_scores, labels=categories_sorted, autopct="%1.1f%%")
-
-    st.pyplot(fig)
-
 
 def plot_alatirchart(sorted_cosine_scores_models):
     models = list(sorted_cosine_scores_models.keys())
@@ -296,7 +242,9 @@ if 'text_search' not in st.session_state:
 
 embeddings_model = Embeddings()
 
-model_type = st.sidebar.selectbox("Choose the model", ("50d"), index=1)
+model_type = "50d"
+st.sidebar.write("Model Type: 50d")
+
 
 
 st.title("in in-class coding practice1 Demo")