"""Temporary isolation test: set app_file: gpu_probe.py in Space README."""
import spaces
import torch
import gradio as gr
from importlib.metadata import version

print(f"Probe runtime: torch={torch.__version__}, cuda_build={torch.version.cuda}, spaces={version('spaces')}", flush=True)

@spaces.GPU(duration=10)
def test_gpu():
    print("GPU probe worker entered", flush=True)
    result = (torch.ones(1, device='cuda') + 1).cpu().item()
    return f"GPU works: {torch.cuda.get_device_name(0)}; result={result}"

demo = gr.Interface(fn=test_gpu, inputs=[], outputs='text', title='ZeroGPU allocation test')
demo.queue(default_concurrency_limit=1).launch(
    server_name="0.0.0.0",
    server_port=7860,
    prevent_thread_lock=False,
)