change web app

app.py

@@ -1,18 +1,53 @@
 import streamlit as st
 from extractive_model import summarize_pdf_with_textrank
 
-st.title("PDF Summarization App")
-
-pdf_file = st.file_uploader("Upload a PDF file", type=["pdf"])
-
-if pdf_file is not None and st.button("Summarize"):
-    # Save uploaded PDF to a temporary file
-    with open("temp_pdf.pdf", "wb") as f:
-        f.write(pdf_file.getbuffer())
-    
-    # Generate summary
-    summary = summarize_pdf_with_textrank("temp_pdf.pdf")
-    
-    # Display summary
-    st.write("Summary:")
-    st.write(summary)
+# Set page to wide mode
+st.set_page_config(layout="wide")
+
+# Function to handle file upload and return its content
+def load_pdf(file):
+    pdf_reader = PyPDF2.PdfReader(file)
+    pdf_text = ""
+    for page_num in range(len(pdf_reader.pages)):
+        pdf_text += pdf_reader.pages[page_num].extract_text()
+    return pdf_text
+
+# Main app
+def main():
+
+    st.title("Terms of Service Summarizer")
+
+    # Layout: 3 columns
+    col1, col2, col3 = st.columns([1, 3, 2], gap="large")
+
+    # Left column: Dropdown menu
+    with col1:
+        dropdown_options = ['Abstractive', 'Extractive']
+        dropdown_selection = st.selectbox("Choose type of summerizer:", dropdown_options)
+
+    # Middle column: Text input and File uploader
+    with col2:
+        user_input = st.text_input("Enter your text here:")
+        uploaded_file = st.file_uploader("Upload a PDF", type="pdf")
+        if st.button("Summarize"):
+            # Handling file upload
+            if uploaded_file is not None:
+                file_content = load_pdf(uploaded_file)
+                st.write("PDF uploaded successfully.")
+                # summary = summarizer(file_content)
+                summary = file_content
+            elif user_input is not None:
+                # summary = summarizer(user_input)
+                summary = user_input
+            else:
+                st.wirte("Upload a PDF or put in your text!")
+            st.session_state.summary = summary
+
+    # Right column: Displaying text after pressing 'Summarize'
+    with col3:
+        st.write("Output:")
+        if 'summary' in st.session_state:
+            st.write(st.session_state.summary)
+
+if __name__ == "__main__":
+    main()