update app.py

app.py

@@ -32,6 +32,7 @@ def convert_to_dataframe():
             "url",
             "stargazers_count",
             "downloads",
+            "model_architectures",
             "source",
             "score",
             "first_commit",
@@ -134,6 +135,26 @@ def main():
                 index=0,
             )
 
+            model_architectures = df["model_architectures"].tolist()
+            model_architectures_counts = Counter(model_architectures)
+            del model_architectures_counts[None]
+
+            model_architectures_counts = sorted(
+                model_architectures_counts.items(),
+                key=lambda x: x[1],
+                reverse=True,
+            )
+            model_architectures_list = [""] + [
+                model_and_count[0]
+                for model_and_count in model_architectures_counts
+            ]
+
+            selected_model_architecture = st.selectbox(
+                "Choose a model architecture",
+                model_architectures_list,
+                index=0,
+            )
+
             min_downloads = int(df["downloads"].min())
             max_downloads = int(df["downloads"].max())
 
@@ -219,7 +240,13 @@ def main():
     contained_project_name = df["project_name"].str.contains(
         query, case=False, na=False
     )
-    df = df[contained_description | contained_project_name]
+    contained_model_arch = df["model_architectures"].str.contains(
+        query, case=False, na=False
+    )
+
+    df = df[
+        contained_description | contained_project_name | contained_model_arch
+    ]
 
     if selected_languges:
         df = df[
@@ -235,6 +262,12 @@ def main():
             )
         ]
 
+    if selected_model_architecture:
+        contained_model_arch = df["model_architectures"].str.contains(
+            selected_model_architecture, case=False, na=False
+        )
+        df = df[contained_model_arch]
+
     if latest_repos_btn:
         df = df.sort_values(by="first_commit", ascending=False)
         df = df[:num_show_repos]
@@ -321,6 +354,20 @@ def main():
             )
             st.altair_chart(chart, use_container_width=True)
 
+            model_architectures = df["model_architectures"].tolist()
+            model_architectures_counts = Counter(model_architectures)
+            del model_architectures_counts[None]
+            model_architectures_df = pd.DataFrame(
+                model_architectures_counts.items(),
+                columns=["Model_architectures", "Count"],
+            )
+            model_architectures_df = model_architectures_df.sort_values(
+                by="Count", ascending=False
+            )
+
+            st.markdown("### Model Architecture Table")
+            st.dataframe(model_architectures_df)
+
     with col2:
         if selected_source_type == "GitHub":
             vs_type = "stargazers_count"