Update app.py

app.py

@@ -9,6 +9,7 @@ from search import search
 from get_paper import get_paper
 from get_pages import get_pages
 from tts import tts
+import IPython.display as ipd
 
 
 st.title("ArXiV Audio")
@@ -46,15 +47,15 @@ if paper:
   print("total pages=", tpages)
   pgs = [i+1 for i in range(tpages)]
   
-  start_page = pgs[0]
-  end_page = pgs[-1]-1
+  start_page = 1
+  end_page = 1
   
   with st.form(key = "page_form"):
     col1, col2 = st.columns(2)
     with col1:
-      start_page = st.selectbox(label = "Start Page", options = pgs)
+      s_page = st.selectbox(label = "Start Page", options = pgs)
     with col2:
-      end_page = st.selectbox(label = "End Page", options = pgs, index = end_page)
+      e_page = st.selectbox(label = "End Page", options = pgs, index = pgs[-1]-1)
     submit_pages = st.form_submit_button(label = "Convert To Audio")
     
   if submit_pages:
@@ -64,5 +65,6 @@ if paper:
       content = get_pages(name, start_page, end_page)
       
       wav, rate = tts(content)
+      ipd.Audio(wav, rate=rate)