Update app.py

app.py

@@ -4,7 +4,7 @@ import arxiv
 #from pdfminer.high_level import extract_text, extract_pages
 #from pdfminer.layout import LTTextContainer
 from search import search
-#from get_paper import get_paper
+from get_paper import get_paper
 #from get_pages import get_pages
 #from tts import tts
 
@@ -22,15 +22,31 @@ with st.form(key = "search_form"):
   
 lst = search(query=query, sort_by=sort_by, sort_order=order_by)
 if len(lst) != 0:
-  label = "Papers for" + query
+  label = "Papers for " + query
   with st.form(key = "paper_form"):
-    paper = st.selectbox(label = label, options=lst)
+    pname = st.selectbox(label = label, options=lst)
     submit_paper = st.form_submit_button(label = "Fetch Paper")
 else:
   with st.form(key = "paper_form"):
     paper = st.selectbox(label = "NO PAPERS FOUND", options=lst)
     submit_paper = st.form_submit_button(label = "Fetch Paper")
-  
-#with st.form(key = "paper_form"):
+    
+paper = get_paper(pname)
+
+name = paper.title+'.pdf'
+name = name.replace('?', '')
+name = "downloads/" + name
+
+tpages = len(list(extract_pages(name)))
+print("total pages=", tpages)
+pgs = [i+1 for i in range(tpages)]
+
+with st.form(key = "paper_form"):
+  col1, col2 = st.columns(3)
+  with col1:
+    start_page = st.selectbox(label = "Start Page", options = pgs)
+  with col2:
+    end_page = st.selectbox(label = "End Page", options = pgs)
+  submit_pages = st.form_submit_button(label = "Convert To Audio")