Update app.py

app.py

@@ -14,7 +14,13 @@ import numpy as np
 def preprocess_image(img):
     """Preprocess image to improve OCR accuracy for handwritten text"""
     # Convert to grayscale
-    gray = cv2.cvtColor(np.array(img), cv2.COLOR_BGR2GRAY)
+    img_array = np.array(img)
+    
+    # Check if the image is already grayscale
+    if len(img_array.shape) == 2:
+        gray = img_array
+    else:
+        gray = cv2.cvtColor(img_array, cv2.COLOR_RGB2GRAY)
     
     # Apply thresholding
     _, binary = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
@@ -78,7 +84,7 @@ with gr.Blocks(title="Handwritten Text OCR Extractor") as app:
     
     with gr.Row():
         with gr.Column():
-            file_input = gr.File(label="Upload Image or PDF", file_types=["image", "pdf"])
+            file_input = gr.File(label="Upload Image or PDF")
             extract_button = gr.Button("Extract Text")
         
         with gr.Column():