Update app.py

app.py

@@ -18,7 +18,7 @@ Only return the list, nothing else. Example:
 def parse_list_boxes(text):
     """Improved parsing with better error handling"""
     try:
-        return eval(text)  # Safer alternative: Use ast.literal_eval
+        return eval(text)
     except:
         matches = re.findall(r'\[([\d\.]+),\s*([\d\.]+),\s*([\d\.]+),\s*([\d\.]+)\]', text)
         return [[float(x) for x in m] for m in matches]
@@ -50,42 +50,50 @@ uploaded_file = st.file_uploader("Upload PDF", type=["pdf"])
 if uploaded_file and st.button("Analyze"):
     with st.spinner("Processing..."):
         try:
-            images = convert_from_bytes(uploaded_file.read(), dpi=300)  # Increased DPI
-            client = genai.Client(api_key=os.getenv("KEY"))  # Verify env var name
+            images = convert_from_bytes(uploaded_file.read(), dpi=300)
+            client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))
             
-            for idx, image in enumerate(images):
-                with st.expander(f"Page {idx+1}", expanded=True):
-                    img_byte_arr = io.BytesIO()
-                    image.save(img_byte_arr, format='PNG')
-                    
- 
-                    # Get bounding boxes
-                    response = client.models.generate_content(
-                        model="gemini-2.0-flash-exp",
-                        contents=[
-                            DETECTION_PROMPT,
-                            types.Part.from_bytes(
-                                data=img_byte_arr.getvalue(),
-                                mime_type="image/png"
-                            )
-                        ]
-                    )
-
-                    
-                    # Debug output
-                    with st.expander("Raw API Response"):
-                        st.code(response.text)
+            # Create tabs for pages
+            tabs = st.tabs([f"Page {i+1}" for i in range(len(images))])
+            
+            for idx, (tab, image) in enumerate(zip(tabs, images)):
+                with tab:
+                    col1, col2 = st.columns(2)
                     
-                    # Parse and draw
-                    boxes = parse_list_boxes(response.text)
-                    annotated = draw_bounding_boxes(image.copy(), boxes)
+                    with col1:
+                        st.image(image, caption="Original", use_column_width=True)
                     
-                    # Display
-                    cols = st.columns(2)
-                    cols[0].image(image, caption="Original", use_column_width=True)
-                    cols[1].image(annotated, 
-                                 caption=f"Detected {len(boxes)} text regions", 
-                                 use_column_width=True)
+                    with col2:
+                        img_byte_arr = io.BytesIO()
+                        image.save(img_byte_arr, format='PNG')
+                        
+                        # Get bounding boxes
+                        response = client.models.generate_content(
+                            model="gemini-2.0-flash-exp",
+                            contents=[
+                                DETECTION_PROMPT,
+                                types.Part.from_bytes(
+                                    data=img_byte_arr.getvalue(),
+                                    mime_type="image/png"
+                                )
+                            ]
+                        )
+                        
+                        # Parse and draw
+                        boxes = parse_list_boxes(response.text)
+                        annotated = draw_bounding_boxes(image.copy(), boxes)
+                        
+                        st.image(annotated, 
+                               caption=f"Detected {len(boxes)} text regions", 
+                               use_column_width=True)
+                        
+                        # Debug section
+                        debug_expander = st.expander("Debug Details")
+                        with debug_expander:
+                            st.write("**Raw API Response:**")
+                            st.code(response.text)
+                            st.write("**Parsed Boxes:**")
+                            st.write(boxes)
 
         except Exception as e:
             st.error(f"Error: {str(e)}")