Update app.py

app.py

@@ -15,10 +15,38 @@ openai.api_key = api_key
 client = OpenAI(api_key=api_key)
 pdf_folder = "pdf"
 
-st.title("Mahkamah Agung: NER & Summarization of Legal Documents")
+# CSS for the bold colored line
+bold_line_css = """
+<style>
+    .centered-title {
+        text-align: center;
+        margin-bottom: 0;  # No gap between the title and the line
+    }
+    .bold-colored-line {
+        border: none;
+        height: 3px;  # Thickness of the line
+        background-color: #FF6347;  # Tomato color
+        margin-top: 0;  # No gap at the top
+    }
+</style>
+"""
+
+# Add the custom CSS to the Streamlit app
+
+
+
+st.markdown(
+    """
+    <h1 style='text-align: center;'>
+        Mahkamah Agung: NER & Summarization of Legal Documents
+    </h1>
+    <hr>
+    """,
+    unsafe_allow_html=True
+)
+
 
 #---------------------PDF OVERVIEW----------------------
-st.subheader("PDF Folder Overview")
 def get_pdf_details(folder_path):
     pdf_details = []
     for filename in os.listdir(folder_path):
@@ -28,7 +56,7 @@ def get_pdf_details(folder_path):
                 with open(pdf_path, "rb") as file:
                     pdf_reader = PdfReader(file)
                     page_count = len(pdf_reader.pages)
-                    pdf_details.append({"Filename": filename, "Page Count": page_count})
+                    pdf_details.append({"Berkas Putusan": filename, "Jumlah Halaman": page_count})
             except Exception as e:
                 st.warning(f"Could not read {filename}: {str(e)}")
     return pdf_details
@@ -36,21 +64,23 @@ def get_pdf_details(folder_path):
 pdf_list = get_pdf_details(pdf_folder)
 pdf_df = pd.DataFrame(pdf_list)
 
-if not pdf_df.empty:
-    with st.expander('PDF Overview'):
-        st.dataframe(pdf_df)
-else:
-    st.warning("No PDFs found in the specified folder.")
+col1, col2= st.columns(2)
+with col1:
+    if not pdf_df.empty:
+        with st.expander('PDF Overview'):
+            st.dataframe(pdf_df)
+    else:
+        st.warning("No PDFs found in the specified folder.")
 
 #---------------------MULTISELECT AND TEXT EXTRACTION----------------------
-st.subheader("Select PDFs for Extraction and Analysis")
 
 pdf_files = [f for f in os.listdir(pdf_folder) if f.lower().endswith('.pdf')]
-selected_pdfs = st.multiselect("Select PDFs", pdf_files)
-
-def extract_text_from_pdf(uploaded_file, start_page, end_page):
-    text = extract_text(uploaded_file, page_numbers=range(start_page, end_page+1))
-    return text
+with col2:
+    with st.expander('PDF Documents'):
+        selected_pdfs = st.multiselect("Select PDFs", pdf_files)
+    def extract_text_from_pdf(uploaded_file, start_page, end_page):
+        text = extract_text(uploaded_file, page_numbers=range(start_page, end_page+1))
+        return text
 
 pdf_texts = {}
 
@@ -76,9 +106,10 @@ template = """
     \'Hakim Ketua\', \'Hakim Anggota\', \'Panitera\', \'Putusan\', \'Putusan Lainnya\', \'Catatan Putusan\', \'Tanggal Musyawarah\', \'Tanggal Pembacaan\', \'Jenis Institusi Yudisial\', \'Tanggal Pendaftaran\', \'Institusi Yudisial\', \'Nomor Kasus\', \'Pengadilan\', \'Nama Terdakwa\', \'Tempat Lahir Terdakwa\', \'Tanggal Lahir Terdakwa\', \'Usia Terdakwa\', \'Jenis Kelamin Terdakwa\', \'Kebangsaan Terdakwa\', \'Agama Terdakwa\', \'Pekerjaan Terdakwa\', \'Pasal Dakwaan\',\' Pelanggaran Dakwaan\', \'Vonis Hukuman\', \'Deskripsi Vonis Atribut Disita\', \'Vonis Atribut Disita Berat\', \'Denda\', dan,  \'Kesimpulan\'.
     # """
 
+if st.button("📝Process"):
+    st.session_state.summaries = True
 
-
-if st.button("Analyze Selected PDFs"):
+if st.session_state.summaries:
     summaries = []
     for pdf_name, text in pdf_texts.items():
         response = client.chat.completions.create(
@@ -95,8 +126,12 @@ if st.button("Analyze Selected PDFs"):
         df.columns = [f"Kesimpulan Putusan ({pdf_name})"]
         
         summaries.append(df)
+        st.session_state.summaries=True
 
     # Display the summaries for each selected PDF
     for summary in summaries:
-        with st.expander(f"Summary for {summary.columns[0]}"):
+        st.session_state.summaries=True
+        with st.expander(f"{summary.columns[0]}"):
             st.dataframe(summary)
+
+`