Update app.py

app.py

@@ -58,10 +58,15 @@ Describe the content of the sidebar, including any navigation, filters, or adver
 def load_model(model_name):
     """Loads the specified model and processor from Hugging Face."""
     model_id = MODEL_OPTIONS[model_name]
-    yield f"Status: Loading {model_name} model ({model_id})... Please wait.", gr.update(interactive=False)
+    
+    # First yield: Update status to loading, disable button, clear previous model state.
+    # MUST return a value for all 4 outputs.
+    yield f"Status: Loading {model_name} model ({model_id})... Please wait.", gr.update(interactive=False), None, None
+    
+    model = None
+    processor = None
     
     try:
-        # Specific loading instructions for the FP8 model
         if "FP8" in model_id:
             model = Qwen3VLForConditionalGeneration.from_pretrained(
                 model_id,
@@ -69,23 +74,22 @@ def load_model(model_name):
                 device_map="auto",
                 trust_remote_code=True
             )
-        else: # Standard loading for other models
+        else:
             model = Qwen3VLForConditionalGeneration.from_pretrained(
                 model_id,
                 device_map="auto",
                 trust_remote_code=True
             )
-            
         processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)
         
-        yield f"Status: {model_name} model loaded successfully.", gr.update(interactive=True)
+        # Final yield on success: Update status, enable button, and return the loaded model/processor.
+        # MUST return a value for all 4 outputs.
+        yield f"Status: {model_name} model loaded successfully.", gr.update(interactive=True), model, processor
         
     except Exception as e:
-        yield f"Status: Error loading model: {e}", gr.update(interactive=True)
-        model, processor = None, None
-        
-    return model, processor
-
+        # Final yield on error: Update status, enable button, return None for model/processor.
+        # MUST return a value for all 4 outputs.
+        yield f"Status: Error loading model: {e}", gr.update(interactive=True), None, None
 
 # --- Playwright Screenshot Function ---
 def take_screenshot(url, max_dimension=1024, full_page_capture=True):
@@ -122,7 +126,7 @@ def take_screenshot(url, max_dimension=1024, full_page_capture=True):
 # --- Inference Function ---
 def process_and_generate(model, processor, image_input, text_prompt, processing_size=512):
     if model is None or processor is None:
-        return "Error: Model is not loaded. Please select a model and click 'Load Model'."
+        return "Error: Model is not loaded. Please select a model and click 'Load/Switch Model'."
     if image_input is None or not text_prompt.strip():
         return "Please provide both an image and a text prompt."
         
@@ -142,7 +146,6 @@ def process_and_generate(model, processor, image_input, text_prompt, processing_
 
 # --- Gradio Interface ---
 with gr.Blocks() as demo:
-    # State components to hold the loaded model and processor
     model_state = gr.State(None)
     processor_state = gr.State(None)
 
@@ -172,10 +175,9 @@ with gr.Blocks() as demo:
 
     output_text = gr.Textbox(label="Model Output", lines=20, interactive=False, placeholder="Generated markdown will appear here...")
 
-    # --- UI Event Handlers ---
     def update_image(url, max_dimension, full_page_capture):
         path = take_screenshot(url, max_dimension, full_page_capture)
-        if "Error" in path: raise gr.Error(path)
+        if "Error" in str(path): raise gr.Error(path)
         return path
 
     def generate_markdown_with_loading(model, processor, image, user_prompt, processing_size, use_template):
@@ -211,4 +213,4 @@ if __name__ == "__main__":
         print("Playwright installation complete.")
     except Exception as e:
         print(f"Could not install playwright dependencies: {e}")
-    demo.launch()
+    demo.launch(share=True)