QuantumChecker 0.2.5__tar.gz → 0.2.6__tar.gz

{quantumchecker-0.2.5 → quantumchecker-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: QuantumChecker
-Version: 0.2.5
+Version: 0.2.6
 Summary: A package to evaluate homework submissions in Python, SQL, PowerBI, and SSIS.
 Author: Qobiljon
 Author-email: qobiljonkhayrullayev@gmail.com

{quantumchecker-0.2.5 → quantumchecker-0.2.6}/QuantumCheck/powerbi_evaluator.py

@@ -13,6 +13,7 @@ from dotenv import load_dotenv
 from PIL import Image
 import io
 import base64
+import PyPDF2  # Added for PDF validation
 
 from .prompts import prompt_text_powerbi
 
@@ -97,16 +98,23 @@ class GeminiFlashModel:
         parts = [{"text": prompt}]
         for img in images:
             logger.debug("Processing image: %s", img.name)
-            with Image.open(img) as pil_img:
-                pil_img.thumbnail((1024, 1024))
-                img_buffer = io.BytesIO()
-                pil_img.save(img_buffer, format="PNG")
-                parts.append({
-                    "inline_data": {
-                        "mime_type": "image/png",
-                        "data": base64.b64encode(img_buffer.getvalue()).decode('utf-8')
-                    }
-                })
+            try:
+                with Image.open(img) as pil_img:
+                    if pil_img.size[0] == 0 or pil_img.size[1] == 0:
+                        logger.error("Invalid image dimensions for %s", img.name)
+                        raise ProcessingError(f"Invalid image dimensions for {img.name}")
+                    pil_img.thumbnail((1024, 1024))
+                    img_buffer = io.BytesIO()
+                    pil_img.save(img_buffer, format="PNG")
+                    parts.append({
+                        "inline_data": {
+                            "mime_type": "image/png",
+                            "data": base64.b64encode(img_buffer.getvalue()).decode('utf-8')
+                        }
+                    })
+            except Exception as e:
+                logger.error("Failed to process image %s: %s", img.name, str(e))
+                raise ProcessingError(f"Failed to process image {img.name}: {str(e)}")
         headers = {"Content-Type": "application/json"}
         data = {"contents": [{"parts": parts}]}
         logger.info("Sending visual evaluation API request to %s", self.endpoint)
@@ -124,8 +132,15 @@ class GeminiFlashModel:
         feedback_match = re.search(r"Feedback:\s*(.*)", output_text, re.DOTALL)
         result = {
             "score": int(score_match.group(1)) if score_match else 0,
-            "feedback": feedback_match.group(1).strip() if feedback_match else "No visual feedback generated"
+            "feedback": feedback_match.group(1).strip() if feedback_match else "No visual feedback generated",
+            "issues": []
         }
+        if not score_match:
+            result["issues"].append("Failed to parse score from visual API response")
+            logger.warning("Failed to parse score from visual API response")
+        if not feedback_match:
+            result["issues"].append("Failed to parse feedback from visual API response")
+            logger.warning("Failed to parse feedback from visual API response")
         logger.info("Visual evaluation completed: Score=%d, Feedback=%s", result["score"], result["feedback"][:50] + "..." if len(result["feedback"]) > 50 else result["feedback"])
         return result
 
@@ -221,10 +236,24 @@ class PowerBIProcessor:
             if not os.path.exists(pdf_path):
                 logger.error("PDF file does not exist: %s", pdf_path)
                 raise ProcessingError(f"PDF file not found: {pdf_path}")
+            # Validate PDF
+            try:
+                with open(pdf_path, "rb") as f:
+                    pdf_reader = PyPDF2.PdfReader(f)
+                    if len(pdf_reader.pages) == 0:
+                        logger.error("PDF is empty: %s", pdf_path)
+                        raise ProcessingError(f"PDF is empty: {pdf_path}")
+                    logger.info("PDF validated, contains %d pages", len(pdf_reader.pages))
+            except Exception as e:
+                logger.error("Invalid PDF file: %s", str(e))
+                raise ProcessingError(f"Invalid PDF file: {str(e)}")
             logger.debug("Creating output directory: %s", output_dir)
             os.makedirs(output_dir, exist_ok=True)
             logger.info("Converting PDF pages to images (max %d pages)", num_pages)
-            pages = convert_from_path(pdf_path, first_page=1, last_page=num_pages)
+            pages = convert_from_path(pdf_path, first_page=1, last_page=min(num_pages, len(pdf_reader.pages)))
+            if not pages:
+                logger.error("No pages converted from PDF: %s", pdf_path)
+                raise ProcessingError(f"No pages converted from PDF: {pdf_path}")
             image_paths = []
             for i, page in enumerate(pages):
                 image_path = os.path.join(output_dir, f"page_{i + 1}.png")
@@ -232,12 +261,12 @@ class PowerBIProcessor:
                 page.save(image_path, "PNG")
                 image_paths.append(image_path)
             logger.info("Successfully processed %d pages from PDF", len(image_paths))
-            logger.debug("Removing original PDF file: %s", pdf_path)
-            os.remove(pdf_path)
             return image_paths
         except Exception as e:
             logger.error("Failed to process PDF: %s", str(e))
-            raise ProcessingError(f"Failed to process PDF: {e}")
+            raise ProcessingError(f"Failed to process PDF: {str(e)}")
+        finally:
+            logger.debug("Not removing PDF file to allow debugging: %s", pdf_path)
 
     def extract_zip(self, zip_path: str, extract_path: str) -> tuple[str, str | None]:
         logger.info("Extracting ZIP file: %s", zip_path)
@@ -375,6 +404,9 @@ class PowerBIEvaluator:
                     logger.info("Processing PDF for visual evaluation: %s", pdf_path)
                     try:
                         image_paths = self.processor.process_pdf(pdf_path)
+                        if not image_paths:
+                            logger.error("No images generated from PDF: %s", pdf_path)
+                            raise ProcessingError("No images generated from PDF")
                         logger.info("Evaluating visuals with question: %s", questions[0])
                         visual_result = self.model.evaluate_visuals(questions[0], "outputimages")
                         result["score"] = (dax_result["score"] + visual_result["score"]) // 2
@@ -385,7 +417,11 @@ class PowerBIEvaluator:
                     except ProcessingError as e:
                         logger.warning("Failed to process PDF, proceeding with DAX evaluation only: %s", str(e))
                         result["issues"].append(f"Visual evaluation skipped: {str(e)}")
-                        result["recommendations"].append("Ensure a valid PDF is provided for visual evaluation if intended")
+                        result["recommendations"].append("Ensure a valid PDF with Power BI visuals is provided")
+                    except Exception as e:
+                        logger.error("Unexpected error during visual evaluation: %s", str(e))
+                        result["issues"].append(f"Visual evaluation failed: {str(e)}")
+                        result["recommendations"].append("Check PDF file and API connectivity")
                 else:
                     logger.info("No PDF provided, skipping visual evaluation")
 

{quantumchecker-0.2.5 → quantumchecker-0.2.6}/QuantumChecker.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: QuantumChecker
-Version: 0.2.5
+Version: 0.2.6
 Summary: A package to evaluate homework submissions in Python, SQL, PowerBI, and SSIS.
 Author: Qobiljon
 Author-email: qobiljonkhayrullayev@gmail.com

{quantumchecker-0.2.5 → quantumchecker-0.2.6}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="QuantumChecker",
-    version="0.2.5",
+    version="0.2.6",
     author="Qobiljon",
     author_email="qobiljonkhayrullayev@gmail.com",
     description="A package to evaluate homework submissions in Python, SQL, PowerBI, and SSIS.",