Update app.py

app.py

@@ -94,72 +94,72 @@ def main():
     def change_photo_state():
         st.session_state["photo"]="done"
     with st.container():
-            c2, c3 = st.columns([1,1])
-            message = st.text_input("Type your text here!")
-            camera_photo = c2.camera_input("Capture a photo to summarize: ", on_change=change_photo_state)
-            uploaded_photo = save(c3.file_uploader("Upload your Images/PDF",type=['jpg','png','jpeg','pdf'], on_change=change_photo_state))
-            if st.session_state["photo"]=="done" or message:
-                if uploaded_photo and uploaded_photo.type=='application/pdf':
-                    tet = read_pdf(uploaded_photo)
-                    # with tempfile.NamedTemporaryFile(delete=False) as temp_file:
-                    #     temp_file.write(uploaded_photo.read())
-                    #     temp_file_path = temp_file.name
-                        
-                    # loader = PyPDFLoader(temp_file_path)
-                    # if loader:
-                    #     text.extend(loader.load())
-                    #     os.remove(temp_file_path)
-                    # text_splitter = CharacterTextSplitter(separator="\n", chunk_size=1000, chunk_overlap=100, length_function=len)
-                    # text_chunks = text_splitter.split_documents(text)
-                    values = st.slider('Select a approximate number of lines to see and summarize',value=[0, len(tet)//(7*100)])
-                    text = tet[values[0]*7*10:values[1]*10*100] if values[0]!=len(tet)//(10*100) else tet[len(tet)//(10*100):]
-                    #st.success(type(text_chunks))
-                    if st.button("English Pdf Summarize"):
-                        st.subheader("Selected text for summarize: ")
-                        st.success(text)
-                        st.subheader("Summarized Text: ")
-                        engsum(text)
-                
-                elif uploaded_photo and uploaded_photo.type !='application/pdf':
-                    text=None
-                    img = Image.open(uploaded_photo)
-                    img = img.save("img.png")
-                    img = cv2.imread("img.png")
-                    st.text("Select the summarization type:")
-                    c4, c5 = st.columns([1,1])
-                    if c4.button("BENGALI"):
-                        text =  pytesseract.image_to_string(img, lang="ben")
-                        st.subheader("সারাংশ/সারমর্ম")
-                        bansum(text)
-                    if c5.button("ENGLISH"): 
-                        text=pytesseract.image_to_string(img)
-                        st.subheader("Summarized Text")
-                        engsum(text)
-                    #st.success(text)
-                elif camera_photo:
-                    text=None
-                    img = Image.open(camera_photo)
-                    img = img.save("img.png")
-                    img = cv2.imread("img.png")
-                    #text = pytesseract.image_to_string(img) if st.checkbox("Bangla") else pytesseract.image_to_string(img, lang="ben")
-                    st.text("Select the summarization type:")
-                    c6, c7 = st.columns([1,1])
-                    if c6.button("Bangla"):
-                        text =  pytesseract.image_to_string(img, lang="ben")
-                        st.subheader("সারাংশ/সারমর্ম")
-                        bansum(text)
-                    if c7.button("English"): 
-                        text=pytesseract.image_to_string(img)
-                        st.subheader("Summarized Text")
-                        engsum(text)
-                else:
-                    text=None
-                    text = message
-                    c8, c9 = st.columns([1,1])
-                    if c8.button("Bangla"):
-                        bansum(text)
-                    if c9.button("English"): 
-                        engsum(text) 
+        c2, c3 = st.columns([1,1])
+        message = st.text_input("Type your text here!")
+        camera_photo = c2.camera_input("Capture a photo to summarize: ", on_change=change_photo_state)
+        uploaded_photo = save(c3.file_uploader("Upload your Images/PDF",type=['jpg','png','jpeg','pdf'], on_change=change_photo_state))
+        if st.session_state["photo"]=="done" or message:
+            if uploaded_photo and uploaded_photo.type=='application/pdf':
+                tet = read_pdf(uploaded_photo)
+                # with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+                #     temp_file.write(uploaded_photo.read())
+                #     temp_file_path = temp_file.name
+                    
+                # loader = PyPDFLoader(temp_file_path)
+                # if loader:
+                #     text.extend(loader.load())
+                #     os.remove(temp_file_path)
+                # text_splitter = CharacterTextSplitter(separator="\n", chunk_size=1000, chunk_overlap=100, length_function=len)
+                # text_chunks = text_splitter.split_documents(text)
+                values = st.slider('Select a approximate number of lines to see and summarize',value=[0, len(tet)//(7*100)])
+                text = tet[values[0]*7*10:values[1]*10*100] if values[0]!=len(tet)//(10*100) else tet[len(tet)//(10*100):]
+                #st.success(type(text_chunks))
+                if st.button("English Pdf Summarize"):
+                    st.subheader("Selected text for summarize: ")
+                    st.success(text)
+                    st.subheader("Summarized Text: ")
+                    engsum(text)
+            
+            elif uploaded_photo and uploaded_photo.type !='application/pdf':
+                text=None
+                img = Image.open(uploaded_photo)
+                img = img.save("img.png")
+                img = cv2.imread("img.png")
+                st.text("Select the summarization type:")
+                c4, c5 = st.columns([1,1])
+                if c4.button("BENGALI"):
+                    text =  pytesseract.image_to_string(img, lang="ben")
+                    st.subheader("সারাংশ/সারমর্ম")
+                    bansum(text)
+                if c5.button("ENGLISH"): 
+                    text=pytesseract.image_to_string(img)
+                    st.subheader("Summarized Text")
+                    engsum(text)
+                #st.success(text)
+            elif camera_photo:
+                text=None
+                img = Image.open(camera_photo)
+                img = img.save("img.png")
+                img = cv2.imread("img.png")
+                #text = pytesseract.image_to_string(img) if st.checkbox("Bangla") else pytesseract.image_to_string(img, lang="ben")
+                st.text("Select the summarization type:")
+                c6, c7 = st.columns([1,1])
+                if c6.button("Bangla"):
+                    text =  pytesseract.image_to_string(img, lang="ben")
+                    st.subheader("সারাংশ/সারমর্ম")
+                    bansum(text)
+                if c7.button("English"): 
+                    text=pytesseract.image_to_string(img)
+                    st.subheader("Summarized Text")
+                    engsum(text)
+            else:
+                text=None
+                text = message
+                c8, c9 = st.columns([1,1])
+                if c8.button("Bangla"):
+                    bansum(text)
+                if c9.button("English"): 
+                    engsum(text) 
              
     with st.container():
         from streamlit_chat import message as st_message
@@ -193,18 +193,6 @@ def main():
             from copyreg import clear_extension_cache
             for chat in st.session_state.history:
                 st_message(**chat)
-    # if st.button("English Text Generation"): 
-    #     def query(payload):
-    #     	response = requests.post(API_URL2, headers=headers2, json=payload)
-    #     	return response.json()
-            
-    #     out = query({
-    #     	"inputs": text,
-    #     })
-    #     if isinstance(out, list) and out[0].get("generated_text"):
-    #         text_output = out[0]["generated_text"]
-    #         st.success(text_output)
-    #         #text=text_output
 
 
 if __name__ == "__main__":