Update app.py

app.py

@@ -7,7 +7,9 @@ from utils import (
     csv_to_dataframe,
     save_csv,
     get_pdf_metadata,
-    extract_csv_from_response
+    extract_csv_from_response,
+    pdf_to_images,
+    analyze_single_document
 )
 import base64
 from datetime import datetime
@@ -93,7 +95,7 @@ with st.sidebar:
     st.markdown("---")
     st.info("""
         **Features:**
-        - Direct PDF processing
+        - PDF processing using images partitioned by page
         - Individual analysis for each document
         - Downloadable CSV reports
     """)
@@ -126,11 +128,17 @@ if uploaded_files and api_key:
                     with st.spinner(f"Analyzing {uploaded_file.name}..."):
                         try:
                             # Analyze PDF directly
-                            raw_response = analyze_pdf_directly(
-                                pdf_bytes=uploaded_file.getvalue(),
-                                prompt=PROMPT,
-                                model_name="gemini-1.5-pro"  # or "gemini-1.5-flash"
-                            )
+                             # Convert PDF to images
+                            images = pdf_to_images(uploaded_file.getvalue())
+                            
+                            # Analyze document
+                            raw_response = analyze_single_document(images, PROMPT)
+
+                            # raw_response = analyze_pdf_directly(
+                            #     pdf_bytes=uploaded_file.getvalue(),
+                            #     prompt=PROMPT,
+                            #     model_name="gemini-1.5-pro"  # or "gemini-1.5-flash"
+                            # )
                             
                             # Process response
                             csv_data = extract_csv_from_response(raw_response)
@@ -139,8 +147,9 @@ if uploaded_files and api_key:
                             with st.expander("View Analysis Results", expanded=True):
                                 if csv_data:
                                     df = csv_to_dataframe(csv_data)
+                                    print(f"DataFrame: {df}")
                                     if not df.empty:
-                                        st.dataframe(df, use_container_width=True)
+                                        st.dataframe(df)
                                         
                                         # Download button
                                         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")