Merge pull request #4 from MrinalGoel643/abby_api_updates

added api and display for analyzing department for specific artworks

.gitignore

@@ -1 +1,2 @@
-venv
+venv
+final_project

main.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import pandas as pd
 from met_api import get_objectsWithImages, get_images
 
 # Caching Section
@@ -10,8 +11,6 @@ def cache_objectsWithImages():
 def cache_images(total, objectIDs, limit):
     return get_images(total, objectIDs, limit)
 
-
-
 # Page setup
 st.set_page_config(page_title="The METrics", layout="centered")
 
@@ -82,6 +81,25 @@ st.markdown("""
 </div>
 """, unsafe_allow_html=True)
 
+# Department analytic
+st.write("")
+st.subheader("Departments for your search")
+
+query = st.text_input("Search term (for images only)", value="cats", key="dept_query")
+max_ids = st.slider("How many results to analyze", 20, 400, 150, 10, key="dept_max")
+
+if st.button("Run department analytic", key="dept_run"):
+    with st.spinner("Fetching and tallying departments…"):
+        rows = met_api.department_counts(q=query, max_ids=max_ids)
+
+    if not rows:
+        st.info("No results found (or the API call failed). Try another term.")
+    else:
+        df = pd.DataFrame(rows, columns=["Department", "Count"])
+        # show a quick bar chart + table
+        st.bar_chart(df.set_index("Department"))
+        st.dataframe(df, use_container_width=True)
+
 st.write("")
 
 # Footer

met_api.py

@@ -31,3 +31,31 @@ def get_images(totalObjects, objectIDs, limit):
             break
 
     return images
+
+def department_counts(q="*", max_ids=200):
+    """
+    Analytic: return a list of (department, count) for search results.
+    - q: search query (default '*' = anything)
+    - max_ids: cap how many object IDs to inspect (keeps it fast)
+
+    Uses the Met search endpoint (images only), then tallies the 'department'
+    field from each object's metadata.
+    """
+    try:
+        resp = requests.get(f"{URL}search", params={"q": q, "hasImages": True}, timeout=15)
+        resp.raise_for_status()
+        ids = (resp.json().get("objectIDs") or [])[:max_ids]
+    except Exception:
+        return []
+
+    counts = {}
+    for oid in ids:
+        try:
+            obj = get_object(oid)  # uses your existing helper
+            dep = obj.get("department") or "(unknown)"
+            counts[dep] = counts.get(dep, 0) + 1
+        except Exception:
+            continue
+
+    # return sorted (department, count) pairs, highest first
+    return sorted(counts.items(), key=lambda x: x[1], reverse=True)

requirements.txt

@@ -1,2 +1,3 @@
 streamlit==1.48.1
+pandas
 requests==2.32.5