Update final_code.py

final_code.py

@@ -0,0 +1,67 @@
+import streamlit as st
+import pandas as pd
+import os
+
+# Function to get list of document IDs
+def get_document_ids():
+    # Assuming 'abc' is the folder containing documents
+    document_ids = [file.split('_')[0] for file in os.listdir('abc') if file.endswith('.png')]
+    return list(set(document_ids))
+
+# Function to load image based on selected document ID and page number
+def load_image(document_id, page_number):
+    image_path = f'abc/{document_id}_{page_number}.png'
+    if os.path.exists(image_path):
+        return image_path
+    else:
+        return None
+
+# Function to load dataframe based on selected document ID
+def load_dataframe(document_id):
+    csv_path = f'abc/{document_id}_1.csv'
+    if os.path.exists(csv_path):
+        return pd.read_csv(csv_path)
+    else:
+        return None
+
+def main():
+    st.set_page_config(layout="wide")  # Set layout to wide
+
+    # Slider to adjust the width of the columns
+    col1_width = st.sidebar.slider("Width of First Column", 0.1, 10.0, 1.0, 0.1)
+    col2_width = st.sidebar.slider("Width of Second Column", 0.1, 10.0, 3.0, 0.1)
+    col3_width = st.sidebar.slider("Width of Third Column", 0.1, 10.0, 2.0, 0.1)
+
+    # Divide the screen into three vertical panels with specified widths
+    col1, col2, col3 = st.columns([col1_width, col2_width, col3_width])
+
+    # Document Selection Panel
+    with col1:
+        st.write("### Document Selection")
+        document_id = st.selectbox("Select Document ID", options=get_document_ids())
+        page_number = st.number_input("Page Number", min_value=1, step=1, value=1)
+
+    # Display Image Panel
+    with col2:
+        st.write("### Display Image")
+        image_path = load_image(document_id, page_number)
+        if image_path:
+            st.image(image_path)
+        else:
+            st.write("Image not found")
+
+    # Display DataFrame Panel
+    with col3:
+        st.write("### Display DataFrame")
+        df = load_dataframe(document_id)
+        if df is not None:
+            columns_to_display = st.multiselect("Select Columns to Display", df.columns)
+            if len(columns_to_display) > 0:
+                st.write(df[columns_to_display])
+            else:
+                st.write("No columns selected")
+        else:
+            st.write("DataFrame not found")
+
+if __name__ == "__main__":
+    main()