Update app.py

app.py

@@ -126,12 +126,15 @@ def take_screenshot(url):
     finally:
         if wd:
             wd.quit()
-    
+
+
 def create_ui():
-    spaces_list = get_most_liked_spaces()
-    print(f"Type of spaces_list: {type(spaces_list)}")
-    formatted_spaces = format_spaces(spaces_list)
-    print(f"Total spaces loaded: {len(formatted_spaces)}")
+    try:
+        spaces_list = get_most_liked_spaces()
+        print(f"Type of spaces_list: {type(spaces_list)}")
+        formatted_spaces = format_spaces(spaces_list)
+        print(f"Total spaces loaded: {len(formatted_spaces)}")
+        
 
     css = """
     footer {visibility: hidden;}
@@ -164,4 +167,18 @@ def create_ui():
                 outputs=[info_output, app_py_content, screenshot_output]
             )
 
-    return demo
+    return demo
+
+
+    except Exception as e:
+        print(f"Error in create_ui: {str(e)}")
+        print(traceback.format_exc())
+        raise
+
+if __name__ == "__main__":
+    try:
+        demo = create_ui()
+        demo.launch()
+    except Exception as e:
+        print(f"Error in main: {str(e)}")
+        print(traceback.format_exc())