Update app.py

app.py

@@ -6,11 +6,9 @@ from surya.model.recognition.model import load_model as load_rec_model
 from surya.model.recognition.processor import load_processor as load_rec_processor
 from PIL import Image
 import torch
-import tempfile
 import os
 import re
 import json
-import base64
 from groq import Groq
 from st_keyup import st_keyup
 
@@ -109,15 +107,17 @@ model_choice = st.sidebar.selectbox("Select OCR Model:", ("GOT_CPU", "GOT_GPU",
 uploaded_file = st.sidebar.file_uploader("Choose an image...", type=["png", "jpg", "jpeg"])
 
 # Input from clipboard
-clipboard_data = st.sidebar.text_area("Paste image data from Clipboard (base64 format):")
-if clipboard_data:
-    try:
-        image_data = base64.b64decode(clipboard_data)
-        uploaded_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
-        uploaded_file.write(image_data)
-        uploaded_file.seek(0)
-    except Exception as e:
-        st.sidebar.warning(f"Clipboard data is not a valid base64 encoded image or an error occurred: {str(e)}")
+# Paste image button
+paste_result = paste_image_button(
+    label="Paste Image from Clipboard",
+    text_color="#ffffff",
+    background_color="#3498db",
+    hover_background_color="#2980b9",
+    key='paste_button',
+    errors='ignore'
+)
+if paste_result:
+    uploaded_file = paste_result.image_data
 
 # Input from camera
 camera_file = st.sidebar.camera_input("Capture from Camera")