Update app.py

app.py

@@ -16,7 +16,7 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
 #os.system('pip install -q pytesseract')
 #os.system('conda install -c conda-forge poppler')
 import streamlit as st
-#st.set_page_config(page_title="Summarization Tool", layout="wide", initial_sidebar_state="expanded")
+st.set_page_config(page_title="Summarization Tool", layout="wide", initial_sidebar_state="expanded")
 st.title("Bangla and English Summarizer: Upload Images/Pdf or input texts to summarize!")
 import torch
 import docx2txt
@@ -40,7 +40,7 @@ API_URL1 = "https://api-inference.huggingface.co/models/Michael-Vptn/text-summar
 headers1 = {"Authorization": "Bearer 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"}
 API_URL2 = "https://api-inference.huggingface.co/models/gpt2"
 headers2 = {"Authorization": "Bearer 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"}
-#@st.cache
+@st.cache
 def read_pdf(file):
 #     images=pdf2image.convert_from_path(file)
 #     # print(type(images))
@@ -85,7 +85,7 @@ def bansum(text):
     
 if "photo" not in st.session_state:
     st.session_state["photo"]="not done"
-c2, c3 = st.columns([2.5,1])
+c2, c3 = st.columns([2,1])
 def change_photo_state():
     st.session_state["photo"]="done"
 message = st.text_input("Type your text here!")
@@ -119,11 +119,12 @@ if st.session_state["photo"]=="done" or message:
         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)
@@ -135,20 +136,22 @@ if st.session_state["photo"]=="done" or message:
         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.button("English Text Generation"): 
     #     def query(payload):