fix

app.py

@@ -15,7 +15,7 @@ model.eval()
 # Function to process PDF and extract structured information
 def extract_license_info(pdf_file):
     if pdf_file is None:
-        return "Please upload a PDF file.", None, None
+        return "Please upload a PDF file.", None, "**Status:** Waiting for PDF upload..."
     
     try:
         # Convert PDF to image
@@ -35,11 +35,24 @@ def extract_license_info(pdf_file):
             outputs = model.generate(**inputs)
         extracted_text = processor.batch_decode(outputs, skip_special_tokens=True)[0]
 
+        # Create status message
+        status = f"""
+        ✅ **Processing Complete!**
+        - Image size: {img.width} x {img.height} pixels
+        - Device: {device.upper()}
+        - Text extracted: {len(extracted_text)} characters
+        """
+
         # Return results
-        return extracted_text, img, None
+        return extracted_text, img, status
 
     except Exception as e:
-        return f"Error processing PDF: {str(e)}", None, None
+        error_status = f"""
+        ❌ **Error Processing PDF**
+        - Error: {str(e)}
+        - Please try with a different PDF file
+        """
+        return f"Error processing PDF: {str(e)}", None, error_status
 
 # Gradio interface
 with gr.Blocks(title="Driver's License Info Extractor") as demo: