Update app.py

app.py

@@ -28,10 +28,9 @@ def process_ocr(document_source, api_key):
     )
 
 def do_ocr(input_type, url, file, api_key):
-    # Check UI-provided API key first, then fall back to environment variable
     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 via the input field or set the MISTRAL environment variable.", "", []
+        return "Please provide a valid Mistral API key", "", []
     
     document_source = None
     
@@ -113,9 +112,9 @@ with gr.Blocks(
     css=custom_css,
     theme=gr.themes.Soft()
 ) as demo:
-    gr.Markdown("<h1 style='text-align: center; color: #333;'>Mistral OCR Demo</h1>")
-    gr.Markdown("<p style='text-align: center; color: #666;'>Extract text and images from PDFs or images using Mistral's latest OCR model. Visit <a href='https://console.mistral.ai/'>Mistral AI Console</a> to manage your API key. You can also see markdown live.</p>")
-    gr.Markdown("<p style='text-align: center; color: #666;'>Provide your Mistral API key below or set the MISTRAL environment variable.</p>")
+    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. Visit <a href='https://console.mistral.ai/'>Mistral AI Console</a> to manage your API key. You can also see markdown live.</p>")
+    gr.Markdown("<p style='text-align: center;'>Provide your Mistral API key below or set the MISTRAL as secret from space settings.</p>")
 
     with gr.Row():
         with gr.Column(scale=1):