Add error handling to process_chem_image for debugging

app.py

@@ -103,33 +103,44 @@ def format_reactions_html(reactions_data):
 
 
 def process_chem_image(image, selected_task):
-    image_path = "temp_image.png"
-    image.save(image_path)
-
-    # Run local model (MolScribe + OCR) — no external API needed
-    predictions = model.predict_image_file(image_path, molscribe=True, ocr=True)
-
-    # Format predictions into structured data
-    reactions_data, smiles_list = format_local_predictions(predictions)
-
-    # Generate HTML output
-    detailed_reactions = format_reactions_html(reactions_data)
-
-    # Generate visualization
-    combined_image_path = generate_combined_image(predictions, image_path)
-
-    # Save JSON
-    json_file_path = "output.json"
-    with open(json_file_path, "w") as f:
-        json.dump(reactions_data, f, indent=4)
-
-    return (
-        "\n\n".join(detailed_reactions),
-        "\n".join(smiles_list),
-        combined_image_path,
-        example_diagram,
-        json_file_path,
-    )
+    import traceback
+    try:
+        image_path = "temp_image.png"
+        if image is None:
+            raise ValueError("No image provided")
+        image.save(image_path)
+
+        # Run local model (MolScribe + OCR) — no external API needed
+        predictions = model.predict_image_file(image_path, molscribe=True, ocr=True)
+
+        # Format predictions into structured data
+        reactions_data, smiles_list = format_local_predictions(predictions)
+
+        # Generate HTML output
+        detailed_reactions = format_reactions_html(reactions_data)
+
+        # Generate visualization
+        try:
+            combined_image_path = generate_combined_image(predictions, image_path)
+        except Exception:
+            combined_image_path = None
+
+        # Save JSON
+        json_file_path = "output.json"
+        with open(json_file_path, "w") as f:
+            json.dump(reactions_data, f, indent=4)
+
+        return (
+            "\n\n".join(detailed_reactions),
+            "\n".join(smiles_list),
+            combined_image_path,
+            example_diagram,
+            json_file_path,
+        )
+    except Exception as e:
+        tb = traceback.format_exc()
+        error_html = f"<b>Error:</b> {str(e)}<br><pre>{tb}</pre>"
+        return (error_html, "", None, example_diagram, None)
 
 
 prompts_with_names = list_prompt_files_with_names()