Gradio 6

app.py

@@ -96,33 +96,8 @@ 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 15px;
-        border-radius: 5px;
-    }
-    .gr-button:hover {
-        background-color: #45a049;
-    }
-    .gr-textbox {margin-bottom: 15px;}
-    .tall-radio .gr-radio-item {
-        padding: 15px 0;
-        min-height: 50px;
-        display: flex;
-        align-items: center;
-    }
-    .tall-radio label {font-size: 16px;}
-"""
-
-with gr.Blocks(
-    title="Mistral OCR Demo",
-    css=custom_css,
-    theme=gr.themes.Soft()
-) as demo:
+
+with gr.Blocks(title="Mistral OCR Demo") 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, Mistral OCR 3 (mistral-ocr-2512). You can also see markdown live.</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>")
@@ -158,7 +133,7 @@ with gr.Blocks(
             pdf_example = gr.Button("PDF") 
             img_example = gr.Button("Image") 
         with gr.Column(scale=2):
-            cleaned_output = gr.Textbox(label="Extracted Plain Text", lines=10, show_copy_button=True)
+            cleaned_output = gr.Textbox(label="Extracted Plain Text", lines=10, buttons=["copy"])
             markdown_output = gr.Markdown(label="Rendered Markdown Text")
             image_output = gr.Gallery(label="OCR Extracted Images", columns=2, height="auto")
 
@@ -186,4 +161,4 @@ with gr.Blocks(
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(theme=gr.themes.Soft())