Add download testing to app

app.py

@@ -5,8 +5,9 @@ import sys
 print("==== Environment Information ====")
 print(f"Python version: {sys.version}")
 print(f"Working directory: {os.getcwd()}")
+print(f"Directory contents: {os.listdir('.')}")
 
-# Try importing torch
+# Import required packages
 try:
     import torch
     print(f"PyTorch version: {torch.__version__}")
@@ -21,25 +22,49 @@ except ImportError as e:
 # Simple Gradio interface
 import gradio as gr
 
+def test_download():
+    """Test the download functionality"""
+    try:
+        from download_assets import download_test_file
+        success = download_test_file()
+        if success:
+            return "Download test successful! Check the assets folder."
+        else:
+            return "Download test failed. See logs for details."
+    except Exception as e:
+        return f"Error while testing download: {str(e)}"
+
 def hello(name):
     if not name:
         name = "World"
     
+    result = f"Hello, {name}!\n\n"
+    
     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}"
+        result += f"PyTorch is working. Random tensor:\n{tensor}\n\n"
     else:
-        return f"Hello, {name}! PyTorch is NOT available."
+        result += "PyTorch is NOT available.\n\n"
+    
+    return result
 
-# Create the interface
-demo = gr.Interface(
-    fn=hello,
-    inputs="text",
-    outputs="text",
-    title="SonicDiffusion - PyTorch Test",
-    description="Testing if PyTorch is working properly in the environment."
-)
+# Create the Gradio interface with tabs
+with gr.Blocks(title="SonicDiffusion - Setup Tests") as demo:
+    gr.Markdown("# SonicDiffusion - Environment Tests")
+    
+    with gr.Tab("Hello Test"):
+        gr.Markdown("Test basic functionality and PyTorch")
+        name_input = gr.Textbox(label="Enter your name")
+        hello_btn = gr.Button("Say Hello")
+        hello_output = gr.Textbox(label="Output")
+        hello_btn.click(fn=hello, inputs=name_input, outputs=hello_output)
+        
+    with gr.Tab("Download Test"):
+        gr.Markdown("Test downloading files from Google Drive")
+        download_btn = gr.Button("Test Download")
+        download_output = gr.Textbox(label="Download Result")
+        download_btn.click(fn=test_download, inputs=None, outputs=download_output)
 
 if __name__ == "__main__":
     demo.launch()