Update app.py

app.py

@@ -47,6 +47,9 @@ def load_narratives_data():
 
 narratives_df = load_narratives_data()
 
+# Extract language from id
+narratives_df['language'] = narratives_df['id'].str.extract('-(rus|zho|fas)-')
+
 col1, col2 = st.columns([1, 3], gap="large")
 
 with st.sidebar:
@@ -54,7 +57,19 @@ with st.sidebar:
 
 with col1:
     st.title("Narratives")
-    narrative_ids = narratives_df["id"].tolist()
+    
+    # Add language filter
+    selected_language = st.selectbox(
+        "Select language",
+        ["All", "rus", "zho", "fas"]
+    )
+    
+    if selected_language != "All":
+        filtered_df = narratives_df[narratives_df['language'] == selected_language]
+    else:
+        filtered_df = narratives_df
+    
+    narrative_ids = filtered_df["id"].tolist()
     container_for_nav = st.container()
     
     def sync_from_drop():
@@ -88,7 +103,7 @@ with col2:
     narrative_index = narrative_number
 
     if narrative_index >= 0:
-        narrative = narratives_df.iloc[narrative_index]
+        narrative = filtered_df.iloc[narrative_index]
         
         st.markdown("<h1 style='text-align: center; color: black;text-decoration: underline;'>Editor</h1>", unsafe_allow_html=True)
 
@@ -106,8 +121,9 @@ with col2:
         container.subheader("Original Text")
         original_input = container.text_area("Edit the original text", value=narrative['original'], height=300)
 
-
     elif narrative_index < 0:
         st.title("Overview")
-        st.write(f"Total number of narratives: {len(narratives_df)}")
+        st.write(f"Total number of narratives: {len(filtered_df)}")
+        if selected_language != "All":
+            st.write(f"Selected language: {selected_language}")
         st.write("Select a narrative from the sidebar to view and edit its details.")