Update app.py

app.py

@@ -43,7 +43,7 @@ def get_content_type(url):
 
 def perform_ocr_file(file, ocr_method="Mistral OCR"):
     if ocr_method == "Mistral OCR":
-        if file.name.endswith('.pdf'):
+        if lower(file.name).endswith('.pdf'):
             uploaded_pdf = client.files.upload(
                 file={
                     "file_name": file.name,
@@ -62,7 +62,7 @@ def perform_ocr_file(file, ocr_method="Mistral OCR"):
             )
             client.files.delete(file_id=uploaded_pdf.id)
 
-        elif file.name.endswith(('.png', '.jpg', '.jpeg')):
+        elif lower(file.name).endswith(('.png', '.jpg', '.jpeg')):
             base64_image = encode_image(file.name)
             ocr_response = client.ocr.process(
                 model="mistral-ocr-latest",
@@ -72,7 +72,8 @@ def perform_ocr_file(file, ocr_method="Mistral OCR"):
                 },
                 include_image_base64=True
             )
-
+        else:
+            return "# Unsupported file type. Please provide a PDF or an image (png, jpeg, jpg).", ""
         combined_markdown, raw_markdown = get_combined_markdown(ocr_response)
         return combined_markdown, raw_markdown