more small fixes.

app.py

@@ -1,6 +1,78 @@
-import gradio as gr
+import os
+import glob
+from pathlib import Path
 import traceback
 
+import gradio as gr
+import jpype
+import jpype.imports
+from jpype import JClass, getDefaultJVMPath
+
+def _find_psl_jars() -> list[str]:
+    """
+    Priority:
+      1) Any *.jar inside PSL_JARS_DIR (if set)
+      2) Any *.jar in ./jars next to this script
+      3) Installed pslpython runtime jar
+    """
+    jars: list[str] = []
+
+    # 2) Local ./jars
+    this_dir = Path(__file__).resolve().parent
+    jars_dir = this_dir
+    jars.extend(glob.glob(str(jars_dir / "*.jar")))
+
+    # Deduplicate while preserving order
+    dedup = []
+    seen = set()
+    for j in jars:
+        if j not in seen and Path(j).is_file():
+            seen.add(j)
+            dedup.append(j)
+
+    return dedup
+
+def start_psl_jvm(verbose: bool = True) -> list[str]:
+    """
+    Start a JVM with a classpath that includes PSL jars.
+    Returns the list of jars used.
+    """
+    if jpype.isJVMStarted():
+        if verbose:
+            print("[PSL] JVM already started.")
+        return []
+
+    jars = _find_psl_jars()
+    if not jars:
+        raise RuntimeError(
+            "No PSL jars found. Set PSL_JARS_DIR to a folder with *.jar, "
+            "or place jars under ./jars, or ensure pslpython is installed with psl-runtime.jar."
+        )
+
+    classpath = os.pathsep.join(jars)
+
+    # If getDefaultJVMPath() fails on your platform, set JAVA_HOME first.
+    jvm_path = getDefaultJVMPath()
+
+    # Start JVM
+    jpype.startJVM(
+        jvm_path,
+        f"-Djava.class.path={classpath}",
+        # Optional: tighten memory or logging here
+        # "-Xms256m", "-Xmx1024m",
+    )
+
+    # Sanity check that PSL classes are visible
+    GA = JClass("org.linqs.psl.model.atom.GroundAtom")
+    if verbose:
+        print(f"[PSL] JVM started with {len(jars)} jars.")
+        for j in jars:
+            print(f"     - {j}")
+        print(f"[PSL] Sanity check: loaded {GA}")
+
+    return jars
+
+
 def _check_jvm():
     """
     Try starting the JVM and report status + jars used.