Update app.py

app.py

@@ -28,25 +28,36 @@ else:  # Assuming data is a dictionary
 chapter_id = st.selectbox("Select Chapter", sorted(chapters.keys()), format_func=lambda x: f"Chapter {x}")
 
 # Display chapter summary if available
-chapter_summary = chapters[chapter_id].get('chapter_summary', 'No summary available')
-st.subheader(f"Chapter {chapter_id}")
-st.markdown(f"**Summary:**\n\n{chapter_summary}")
-
-# Extract verses for the selected chapter
-verses = {verse['verse_number']: verse for verse in chapters[chapter_id]['verses']}
-verse_number = st.selectbox("Select Verse", sorted(verses.keys()))
-
-# Display selected verse
-selected_verse = verses[verse_number]
-st.markdown(f"**Verse {verse_number}**")
-st.markdown(f"**Sanskrit:**\n\n{selected_verse['text']}")
-st.markdown(f"**Translation:**\n\n{selected_verse['translation']}")
-
-# Display all verses in the selected chapter
-st.subheader("All Verses in this Chapter")
-for verse_number in sorted(verses.keys()):
-    verse = verses[verse_number]
-    st.markdown(f"**Verse {verse_number}**")
-    st.markdown(f"**Sanskrit:** {verse['text']}")
-    st.markdown(f"**Translation:** {verse['translation']}")
-    st.markdown("---")
+chapter_details = chapters.get(chapter_id)
+if chapter_details:
+    chapter_summary = chapter_details.get('chapter_summary', 'No summary available')
+    st.subheader(f"Chapter {chapter_id}")
+    st.markdown(f"**Summary:**\n\n{chapter_summary}")
+
+    # Extract verses for the selected chapter if available
+    verses = chapter_details.get('verses', [])
+    if verses:
+        verses_dict = {verse['verse_number']: verse for verse in verses}
+
+        # Select Verse
+        verse_number = st.selectbox("Select Verse", sorted(verses_dict.keys()))
+
+        # Display selected verse
+        selected_verse = verses_dict[verse_number]
+        st.markdown(f"**Verse {verse_number}**")
+        st.markdown(f"**Sanskrit:**\n\n{selected_verse['text']}")
+        st.markdown(f"**Translation:**\n\n{selected_verse['translation']}")
+
+        # Display all verses in the selected chapter
+        st.subheader("All Verses in this Chapter")
+        for verse_number in sorted(verses_dict.keys()):
+            verse = verses_dict[verse_number]
+            st.markdown(f"**Verse {verse_number}**")
+            st.markdown(f"**Sanskrit:** {verse['text']}")
+            st.markdown(f"**Translation:** {verse['translation']}")
+            st.markdown("---")
+    else:
+        st.error("No verses found for this chapter.")
+else:
+    st.error("Chapter details not found.")
+