Update app.py

app.py

@@ -154,7 +154,7 @@ def process_and_generate(image_input, text_prompt, processing_size=512):
         if model.config.pad_token_id is None:
             model.config.pad_token_id = model.config.eos_token_id
 
-        generated_ids = model.generate(**inputs, max_new_tokens=2048, do_sample=True, top_p=0.8, temperature=0.7) # Increased tokens for detailed prompt
+        generated_ids = model.generate(**inputs, max_new_tokens=2048, do_sample=True, top_p=0.8, temperature=0.7)
 
         generated_ids_trimmed = [
             out_ids[len(in_ids):] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
@@ -222,7 +222,13 @@ with gr.Blocks() as demo:
             submit_button = gr.Button("Generate Markdown")
 
     with gr.Row():
-        output_text = gr.Markdown(label="Model Output")
+        # --- CHANGED: Switched from gr.Markdown to gr.Textbox for copyable output ---
+        output_text = gr.Textbox(
+            label="Model Output",
+            lines=20,
+            interactive=False,
+            placeholder="Generated markdown will appear here..."
+        )
 
 
     def update_image(url, max_dimension, full_page_capture):
@@ -236,16 +242,17 @@ with gr.Blocks() as demo:
     def generate_markdown_with_loading(image, user_prompt, processing_size, use_template):
         # Return a dictionary of updates to show the loading state
         yield {
-            output_text: "## Processing, please wait... ⏳",
+            # --- CHANGED: Updated loading message for Textbox ---
+            output_text: "Processing, please wait... ⏳",
             submit_button: gr.update(interactive=False)
         }
-        
+
         # Determine which prompt to use
         final_prompt = DETAILED_ANALYSIS_PROMPT if use_template else user_prompt
-        
+
         # Process the data
         result = process_and_generate(image, final_prompt, processing_size)
-        
+
         # Return a dictionary of updates with the final result
         yield {
             output_text: result,