Update app.py

app.py

@@ -82,31 +82,26 @@ def bansum(text):
     if isinstance(out, list) and out[0].get("summary_text"):
        text_output = out[0]["summary_text"]
        st.success(text_output) 
-        
-#Summarize=True
+
 #@st.cache
 def main():
     import streamlit as st
     import time
-    if st.checkbox("Summarize"):
-        if "photo" not in st.session_state:
-            st.session_state["photo"]="not done"
-        def change_photo_state():
-            st.session_state["photo"]="done"
-        c0, c1, c2, c3 = st.columns([1,1,1,1])
+    if "photo" not in st.session_state:
+        st.session_state["photo"]="not done"
+    def change_photo_state():
+        st.session_state["photo"]="done"
+    c1, c2, c3 = st.columns([2,2,1])
+    if st.checkbox("Summarize from text/images/pdfs"):
         with st.container():
-            message = c0.text_input("Type your text here!")
-            uploaded_photo=None
-            camera_photo=None
-            if c1.button("Camera"):
-                camera_photo = c1.camera_input("Capture a photo to summarize: ", on_change=change_photo_state)
-                if c1.button("Stop camera"):
-                   Camera=False
-            if c2.button("Images"):
-                uploaded_photo = c2.file_uploader("Upload your Images/PDF",type=['jpg','png','jpeg'], on_change=change_photo_state)
-            if c3.button("PDF"):
-                uploaded_photo = c3.file_uploader("Upload your Images/PDF",type=['pdf'], on_change=change_photo_state)
-                #Summarize=True
+            message = c1.text_input("Type your text here!")
+            if c2.button("CaptureImage"):
+                camera_photo = c2.camera_input("Capture a photo to summarize: ", on_change=change_photo_state)
+                if c2.button("Stop camera"):
+                    CaptureImage =False
+            if c3.button("Upload Images/PDFs"):
+                time.sleep(3)
+                uploaded_photo = 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)
@@ -135,11 +130,12 @@ def main():
                     img = img.save("img.png")
                     img = cv2.imread("img.png")
                     st.text("Select the summarization type:")
-                    if st.button("BENGALI"):
+                    c4, c5 = st.columns([1,1])
+                    if c4.button("BENGALI"):
                         text =  pytesseract.image_to_string(img, lang="ben")
                         st.subheader("সারাংশ/সারমর্ম")
                         bansum(text)
-                    if st.button("ENGLISH"): 
+                    if c5.button("ENGLISH"): 
                         text=pytesseract.image_to_string(img)
                         st.subheader("Summarized Text")
                         engsum(text)
@@ -151,23 +147,24 @@ def main():
                     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:")
-                    if st.button("Bangla"):
+                    c6, c7 = st.columns([1,1])
+                    if c6.button("Bangla"):
                         text =  pytesseract.image_to_string(img, lang="ben")
                         st.subheader("সারাংশ/সারমর্ম")
                         bansum(text)
-                    if st.button("English"): 
+                    if c7.button("English"): 
                         text=pytesseract.image_to_string(img)
                         st.subheader("Summarized Text")
                         engsum(text)
                 else:
                     text=None
                     text = message
-                    if st.button("Bangla"):
+                    c8, c9 = st.columns([1,1])
+                    if c8.button("Bangla"):
                         bansum(text)
-                    if st.button("English"): 
+                    if c9.button("English"): 
                         engsum(text) 
-    if st.checkbox("Chat/Conversate"):         
-        Summarize=False
+    if st.checkbox("Conversate"):         
         with st.container():
             from streamlit_chat import message as st_message
             from transformers import BlenderbotTokenizer