FEAT: Added logging to test_case function to debug missing PDF

main.py

@@ -51,30 +51,56 @@ test_prompt = ChatPromptTemplate.from_messages(
     )
 
 def test_case(code):
-    # ... (Prompt setup) ...
-    
-    test_chain = test_prompt | llm | StrOutputParser()
-    test_cases = test_chain.invoke({"code": code})
-
-    print("\n[LOG] 1. LLM Raw Output Length:", len(test_cases)) # <-- NEW LOG
+    # ... (LLM generation and cleaning remains the same) ...
 
-    # Aggressive cleaning...
-    # ... (cleaning code) ...
+    # -----------------------------------------------------------
+    # (Existing cleaning and check)
+    test_cases = re.sub(r"```.*?```", "", test_cases, flags=re.DOTALL)
+    test_cases = re.sub(r"```", "", test_cases)
+    test_cases = test_cases.strip()
 
+    # If the LLM returns nothing, force a known output
     if not test_cases:
-        test_cases = "Error: Test case generation failed or returned empty content."
-    
-    print("\n[LOG] 2. Cleaned Text:", test_cases) # <-- NEW LOG
-    
+        test_cases = "Error: Test case generation failed or returned empty content. Check LLM output in logs."
+    # -----------------------------------------------------------
+
     safe_text = test_cases
-    
     pdf = FPDF()
-    # ... (FPDF setup with DejaVuSans) ...
+    pdf.add_page()
     
-    pdf.multi_cell(0, 10, txt=safe_text)
+    # ... (FPDF Unicode font setup remains the same) ...
     
-    file = pdf.output(dest='S').encode('latin-1')
+    pdf.multi_cell(0, 10, txt="--- Generated Test Cases ---", align='C')
+    pdf.multi_cell(0, 10, txt=safe_text)
     
-    print("\n[LOG] 3. PDF Bytes Length:", len(file)) # <-- NEW LOG
     
-    return file
+    # -----------------------------------------------------------
+    # CRITICAL CHANGE: Get bytes object directly from FPDF2
+    try:
+        # dest='S' returns a string in FPDF2. We need to encode this string.
+        pdf_string = pdf.output(dest='S')
+        pdf_bytes = pdf_string.encode('latin-1')
+        
+        # Check to ensure the file is not empty before returning
+        if len(pdf_bytes) < 100:
+            print("[CRITICAL LOG] FPDF generated very small file, likely failed.")
+            # Return a non-empty, guaranteed failure PDF to force the button to display
+            # This is a temporary diagnostic tool.
+            
+            # Recreate PDF with ONLY the error message
+            error_pdf = FPDF()
+            error_pdf.add_page()
+            error_pdf.set_font("Arial", size=12)
+            error_pdf.multi_cell(0, 10, txt="ERROR: PDF generation failed to create content. Check logs.")
+            return error_pdf.output(dest='S').encode('latin-1')
+            
+        return pdf_bytes
+        
+    except Exception as e:
+        print(f"[FATAL LOG] PDF output failed with error: {e}")
+        # Return a guaranteed failure PDF to force button display
+        error_pdf = FPDF()
+        error_pdf.add_page()
+        error_pdf.set_font("Arial", size=12)
+        error_pdf.multi_cell(0, 10, txt=f"FATAL ERROR: PDF generation crashed. Reason: {e}")
+        return error_pdf.output(dest='S').encode('latin-1')