Update app.py

app.py

@@ -226,10 +226,9 @@ def check_reference_order(full_text: str) -> Dict[str, Any]:
         "missing_references": missing_refs,
         "is_ordered": len(out_of_order) == 0 and len(missing_refs) == 0
     }
-
 def highlight_issues_in_pdf(file, language_matches: List[Dict[str, Any]]) -> bytes:
     """
-    Highlights language issues in the PDF, adds a comment box with text on the side of the page,
+    Highlights language issues in the PDF, adds a dynamic comment box with text on the side of the page,
     and draws arrows pointing from the highlighted text to the comment box.
     Returns the annotated PDF as bytes.
     """
@@ -299,18 +298,29 @@ def highlight_issues_in_pdf(file, language_matches: List[Dict[str, Any]]) -> byt
             highlight.set_colors(stroke=(1, 1, 0))  # Yellow color
             highlight.update()
 
-            # Create the comment box on the side of the page
-            comment_x = page.rect.width - 150  # Adjust this value as needed
+            # Dynamically calculate the position of the comment box
+            page_width, page_height = page.rect.width, page.rect.height
+            comment_box_width = min(140, page_width / 3)  # Ensure the comment box width is a reasonable fraction of the page width
+            comment_box_height = 100  # Set a reasonable height for the comment box
+
+            # Position the comment box dynamically
+            if initial_x < page_width / 2:  # If the highlighted text is on the left half of the page
+                comment_x = page_width - comment_box_width - 10  # Position it on the right side
+            else:  # If the highlighted text is on the right half of the page
+                comment_x = 10  # Position it on the left side
+
             comment_y = initial_y  # Position the comment box near the highlighted text
-            comment_rect = fitz.Rect(comment_x, comment_y, comment_x + 140, comment_y + 100)
+            comment_rect = fitz.Rect(comment_x, comment_y, comment_x + comment_box_width, comment_y + comment_box_height)
             page.add_freetext_annot(comment_rect, error_text)
 
             # Draw an arrow from the highlighted word to the comment box
-            arrow_start = (initial_x + final_x) / 2, (initial_y + final_y) / 2  # Center of the highlighted word
-            arrow_end = comment_rect.center  # Center of the comment box
+            arrow_start_x = (initial_x + final_x) / 2  # Center X of the highlighted word
+            arrow_start_y = (initial_y + final_y) / 2  # Center Y of the highlighted word
+            arrow_end_x = (comment_rect.x0 + comment_rect.x1) / 2  # Center X of the comment box
+            arrow_end_y = (comment_rect.y0 + comment_rect.y1) / 2  # Center Y of the comment box
 
             # Draw the arrow
-            page.add_arrow(arrow_start, arrow_end, color=(0, 0, 0), width=2)
+            page.add_arrow((arrow_start_x, arrow_start_y), (arrow_end_x, arrow_end_y), color=(0, 0, 0), width=2)
 
         # Save annotated PDF to bytes
         byte_stream = io.BytesIO()
@@ -331,6 +341,7 @@ def highlight_issues_in_pdf(file, language_matches: List[Dict[str, Any]]) -> byt
 
 
 
+
 # ------------------------------
 # Main Analysis Function
 # ------------------------------