update app

implemented image scaling feature by adding a slider to the user interface and integrating the scaling logic into the document processing function. This allows you to upscale the image before it's sent to the model, which can significantly improve the accuracy of the text extraction, especially for documents with small characters.

app.py

@@ -129,7 +129,8 @@ def generate_and_preview_pdf(image: Image.Image, text_content: str, font_size: i
 @spaces.GPU
 def process_document_stream(
     image: Image.Image, 
-    prompt_input: str, 
+    prompt_input: str,
+    image_scale_factor: float, # New parameter for image scaling
     max_new_tokens: int,
     temperature: float,
     top_p: float,
@@ -146,6 +147,21 @@ def process_document_stream(
         yield "Please enter a prompt.", ""
         return
 
+    # --- IMPLEMENTATION: Image Scaling based on user input ---
+    if image_scale_factor > 1.0:
+        try:
+            original_width, original_height = image.size
+            new_width = int(original_width * image_scale_factor)
+            new_height = int(original_height * image_scale_factor)
+            print(f"Scaling image from {image.size} to ({new_width}, {new_height}) with factor {image_scale_factor}.")
+            # Use a high-quality resampling filter for better results
+            image = image.resize((new_width, new_height), Image.Resampling.LANCZOS)
+        except Exception as e:
+            print(f"Error during image scaling: {e}")
+            # Continue with the original image if scaling fails
+            pass
+    # --- END IMPLEMENTATION ---
+
     temp_image_path = None
     try:
         # --- FIX: Save the PIL Image to a temporary file ---
@@ -230,6 +246,16 @@ def create_gradio_interface():
                 image_input = gr.Image(label="Upload Image", type="pil", sources=['upload'])
                 
                 with gr.Accordion("Advanced Settings", open=False):
+                    # --- NEW UI ELEMENT: Image Scaling Slider ---
+                    image_scale_factor = gr.Slider(
+                        minimum=1.0, 
+                        maximum=3.0, 
+                        value=1.0, 
+                        step=0.1, 
+                        label="Image Upscale Factor",
+                        info="Increases image size before processing. Can improve OCR on small text. Default: 1.0 (no change)."
+                    )
+                    # --- END NEW UI ELEMENT ---
                     max_new_tokens = gr.Slider(minimum=512, maximum=8192, value=2048, step=256, label="Max New Tokens")
                     temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=1.0, step=0.05, value=0.7)
                     top_p = gr.Slider(label="Top-p (nucleus sampling)", minimum=0.05, maximum=1.0, step=0.05, value=0.8)
@@ -276,7 +302,8 @@ def create_gradio_interface():
 
         process_btn.click(
             fn=process_document_stream,
-            inputs=[image_input, prompt_input, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
+            # --- UPDATE: Add the new slider to the inputs list ---
+            inputs=[image_input, prompt_input, image_scale_factor, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
             outputs=[raw_output_stream, markdown_output]
         )