Add PyTorch testing to app

app.py

@@ -1,12 +1,22 @@
-# Minimal app.py that doesn't require torch
 import os
 import sys
 
-# Print environment information for debugging
+# Print environment information
 print("==== Environment Information ====")
 print(f"Python version: {sys.version}")
 print(f"Working directory: {os.getcwd()}")
-print(f"Directory contents: {os.listdir('.')}")
+
+# Try importing torch
+try:
+    import torch
+    print(f"PyTorch version: {torch.__version__}")
+    print(f"CUDA available: {torch.cuda.is_available()}")
+    if torch.cuda.is_available():
+        print(f"CUDA device: {torch.cuda.get_device_name(0)}")
+    torch_available = True
+except ImportError as e:
+    print(f"PyTorch import error: {e}")
+    torch_available = False
 
 # Simple Gradio interface
 import gradio as gr
@@ -14,25 +24,22 @@ import gradio as gr
 def hello(name):
     if not name:
         name = "World"
-    return f"Hello, {name}!"
+    
+    if torch_available:
+        # Create a simple tensor to test PyTorch
+        tensor = torch.rand(3, 3)
+        return f"Hello, {name}! PyTorch is working. Here's a random tensor:\n{tensor}"
+    else:
+        return f"Hello, {name}! PyTorch is NOT available."
 
-# Create a simple Gradio interface
+# Create the interface
 demo = gr.Interface(
     fn=hello,
     inputs="text",
     outputs="text",
-    title="SonicDiffusion - Setup Test",
-    description="This is a test app to verify the environment is working."
+    title="SonicDiffusion - PyTorch Test",
+    description="Testing if PyTorch is working properly in the environment."
 )
 
 if __name__ == "__main__":
-    # Try to print installed packages
-    try:
-        import subprocess
-        print("==== Installed Packages ====")
-        subprocess.run([sys.executable, "-m", "pip", "list"])
-    except Exception as e:
-        print(f"Error listing packages: {e}")
-    
-    # Launch the demo
     demo.launch()