Update app.py

app.py

@@ -9,8 +9,6 @@ from PIL import Image
 import tempfile
 import io
 import threading
-import tkinter as tk
-from tkinter import filedialog, Label, Button, Frame
 from shiny import App, render, ui, reactive
 from shiny.types import ImgData
 import base64
@@ -19,6 +17,9 @@ import base64
 MODEL_DIR = "models"
 TEMP_DIR = "temp"
 
+# Don't import tkinter at the top level
+# This allows the app to run in environments without Tkinter (like Hugging Face)
+
 def parse_args():
     parser = argparse.ArgumentParser(description='Advanced Virtual Try-On')
     parser.add_argument('--garment', type=str, help='Path to garment image')
@@ -870,6 +871,16 @@ class TkinterUI:
     """Tkinter UI for the virtual try-on application"""
     
     def __init__(self, webcam_index=0, resolution="640x480"):
+        # Import tkinter here instead of at the top level
+        try:
+            import tkinter as tk
+            from tkinter import filedialog, Label, Button, Frame
+            self.tk_available = True
+        except ImportError:
+            print("Tkinter is not available in this environment.")
+            self.tk_available = False
+            return
+            
         self.webcam_index = webcam_index
         self.width, self.height = map(int, resolution.split('x'))
         self.app = None
@@ -881,6 +892,14 @@ class TkinterUI:
     
     def start(self):
         """Start the Tkinter UI"""
+        if not self.tk_available:
+            print("Cannot start Tkinter UI: Tkinter is not available.")
+            return
+            
+        # Import tkinter here to use its symbols
+        import tkinter as tk
+        from tkinter import filedialog, Label, Button, Frame
+        
         self.root = tk.Tk()
         self.root.title("Virtual Try-On")
         self.root.geometry(f"{self.width + 300}x{self.height + 100}")
@@ -1114,6 +1133,16 @@ class TkinterUI:
 
 def run_tkinter_app(webcam_index=0, resolution="640x480"):
     """Run the application with Tkinter UI"""
+    # Check if Tkinter is available
+    try:
+        import tkinter as tk
+        tk_available = True
+    except ImportError:
+        print("Tkinter is not available in this environment. Cannot run Tkinter UI.")
+        print("Try running with --shiny instead.")
+        tk_available = False
+        return
+        
     ui = TkinterUI(webcam_index, resolution)
     ui.start()
 
@@ -1341,8 +1370,15 @@ def main():
             app = run_shiny_app()
             app.run(host="0.0.0.0", port=8000)
         elif args.tkinter or (not args.garment and not args.shiny):
-            # Use tkinter by default if no garment is specified
-            run_tkinter_app(args.webcam, args.resolution)
+            # Check if Tkinter is available before trying to use it
+            try:
+                import tkinter
+                # Use tkinter by default if no garment is specified
+                run_tkinter_app(args.webcam, args.resolution)
+            except ImportError:
+                print("Tkinter is not available. Falling back to Shiny mode.")
+                app = run_shiny_app()
+                app.run(host="0.0.0.0", port=8000)
         else:
             # Traditional command-line mode if garment is specified
             width, height = map(int, args.resolution.split('x'))