Update app.py

app.py

@@ -1,37 +1,28 @@
-import streamlit as st
-# from gtts import gTTS
-# from gtts import *
+import gradio as gr
+from gtts import gTTS
 import PyPDF2
 from tqdm import tqdm
 
-
-st.title('Book Translator and Reader')
-pdf_file = st.file_uploader("Upload Book PDF", type=["pdf"])
-
-def pdf_to_text(pdf_file):
+def pdf_to_audio(pdf_file, start_page, end_page):
     whole = ''
-    pdfreader= PyPDF2.PdfReader(pdf_file)
-    # choice2 = st.sidebar.selectbox("Translated_Language",lang)
-    pages=len(pdfreader.pages)
-    
-    page = st.slider('Give the range of the Pages you want to access',0, pages, (25, 27   ))
-    # [start_page,end_page] = input("Enter the start page number give a space and enter end page number").split(" ")
-    for num in tqdm(range(int(page[0]),int(page[1]))):
-        Page=pdfreader.pages[num]
-        text=Page.extract_text()
-        
-        whole+=text   
-
-    # myobj = gTTS(text=text, lang='en',tld='co.in', slow=False) 
- 
-    # myobj.save("test.wav") 
-    return text
+    pdfreader = PyPDF2.PdfFileReader(pdf_file)
+    pages = pdfreader.getNumPages()
 
+    for num in tqdm(range(start_page-1, end_page)):
+        page = pdfreader.getPage(num)
+        text = page.extractText()
+        whole += text
 
+    # myobj = gTTS(text=whole, lang='en', slow=False)
+    # audio = myobj.get_audio_data()
+    return whole
 
-if pdf_file is not None:
+inputs = [
+    gr.inputs.File(label="Book PDF"),
+    gr.inputs.Number(label="Start Page", min_value=1),
+    gr.inputs.Number(label="End Page", min_value=1),
+]
 
-    T = pdf_to_text(pdf_file)
-    st.write(T)
-    # st.audio(A)
+output = gr.outputs.Text(label="Audio")
 
+gr.Interface(fn=pdf_to_audio, inputs=inputs, output=output, title="Book Translator and Reader").launch()