Update app to handle optional package imports

app.py

@@ -7,7 +7,14 @@ print(f"Python version: {sys.version}")
 print(f"Working directory: {os.getcwd()}")
 print(f"Directory contents: {os.listdir('.')}")
 
-# Import required packages
+# Import gradio first - this is our most essential dependency
+import gradio as gr
+
+# Try importing other packages (but don't fail if they're missing)
+torch_available = False
+transformers_available = False
+diffusers_available = False
+
 try:
     import torch
     print(f"PyTorch version: {torch.__version__}")
@@ -17,54 +24,63 @@ try:
     torch_available = True
 except ImportError as e:
     print(f"PyTorch import error: {e}")
-    torch_available = False
 
-# Simple Gradio interface
-import gradio as gr
+try:
+    import transformers
+    print(f"Transformers version: {transformers.__version__}")
+    transformers_available = True
+except ImportError as e:
+    print(f"Transformers import error: {e}")
 
-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)}"
+try:
+    import diffusers
+    print(f"Diffusers version: {diffusers.__version__}")
+    diffusers_available = True
+except ImportError as e:
+    print(f"Diffusers import error: {e}")
 
+# Simple demo interface
 def hello(name):
     if not name:
         name = "World"
     
-    result = f"Hello, {name}!\n\n"
-    
+    status = []
     if torch_available:
-        # Create a simple tensor to test PyTorch
-        tensor = torch.rand(3, 3)
-        result += f"PyTorch is working. Random tensor:\n{tensor}\n\n"
+        status.append("PyTorch ✓")
     else:
-        result += "PyTorch is NOT available.\n\n"
-    
-    return result
-
-# Create the Gradio interface with tabs
-with gr.Blocks(title="SonicDiffusion - Setup Tests") as demo:
-    gr.Markdown("# SonicDiffusion - Environment Tests")
+        status.append("PyTorch ✗")
     
-    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)
+    if transformers_available:
+        status.append("Transformers ✓")
+    else:
+        status.append("Transformers ✗")
         
-    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 diffusers_available:
+        status.append("Diffusers ✓")
+    else:
+        status.append("Diffusers ✗")
+    
+    return f"Hello, {name}!\n\nPackage Status:\n" + "\n".join(status)
+
+# Create the Gradio interface
+demo = gr.Interface(
+    fn=hello,
+    inputs="text",
+    outputs="text",
+    title="SonicDiffusion - Setup Status",
+    description="This app shows which packages are successfully installed."
+)
 
 if __name__ == "__main__":
+    # Try installing packages at runtime if they're not available
+    if not torch_available:
+        print("Attempting to install PyTorch...")
+        try:
+            import subprocess
+            subprocess.check_call([sys.executable, "-m", "pip", "install", "torch==2.0.1"])
+            print("PyTorch installed successfully!")
+        except Exception as e:
+            print(f"Error installing PyTorch: {e}")
+    
+    # Launch the demo
     demo.launch()