fix

app.py

@@ -4,7 +4,6 @@ from io import BytesIO
 import gradio as gr
 from mistralai import Mistral
 from PIL import Image
-from pathlib import Path
 
 # Config
 VALID_DOCUMENT_EXTENSIONS = {".pdf"}
@@ -31,9 +30,9 @@ def do_ocr(input_type, url, file, api_key):
     api_key = api_key.strip() if api_key and api_key.strip() else os.environ.get("MISTRAL")
     if not api_key:
         return "Please provide a valid Mistral API key", "", []
-    
+
     document_source = None
-    
+
     if input_type == "URL":
         if not url or url.strip() == "":
             return "Please provide a valid URL.", "", []
@@ -96,9 +95,10 @@ def do_ocr(input_type, url, file, api_key):
 
     return extracted_text.strip(), rendered_markdown.strip(), images
 
+
 custom_css = """
     body {font-family: 'Helvetica Neue', Helvetica;}
-    .gr-button {background-color: #4CAF50; color: white; border: none; padding: 10px ละpx; border-radius: 5px;}
+    .gr-button {background-color: #4CAF50; color: white; border: none; padding: 10px 15px; border-radius: 5px;}
     .gr-button:hover {background-color: #45a049;}
     .gr-textbox {margin-bottom: 15px;}
     .example-button {background-color: #1E90FF; color: white; border: none; padding: 8px 15px; border-radius: 5px; margin: 5px;}
@@ -114,13 +114,14 @@ with gr.Blocks(
 ) as demo:
     gr.Markdown("<h1 style='text-align: center;'>Mistral OCR Demo</h1>")
     gr.Markdown("<p style='text-align: center;'>Extract text and images from PDFs or images using Mistral's latest OCR model. You can also see markdown live.</p>")
-    gr.Markdown("<p style='text-align: center;'>Provide your Mistral API key below or create MISTRAL secret from space settings. Visit <a href='https://console.mistral.ai/'>Mistral AI Console</a> to manage your API key.</p>")
+    gr.Markdown("<p style='text-align: center;'>Provide your Mistral API key below or set the `MISTRAL` environment variable. Visit <a href='https://console.mistral.ai/'>Mistral AI Console</a> to manage your API key.</p>")
 
     with gr.Row():
         with gr.Column(scale=1):
             api_key_input = gr.Textbox(
                 label="Mistral API Key",
                 placeholder="Paste your Mistral API key here",
+                value=os.environ.get("MISTRAL", ""),
                 type="password",
                 lines=1
             )
@@ -132,7 +133,7 @@ with gr.Blocks(
             )
             url_input = gr.Textbox(
                 label="Document or Image URL",
-                placeholder="e E.g., https://arxiv.org/pdf/2501.12948",
+                placeholder="e.g., https://arxiv.org/pdf/2501.12948",
                 visible=True,
                 lines=1
             )