Update app.py

app.py

@@ -1,14 +1,44 @@
+import os
+import sys
+import subprocess
+import importlib.util
+
+print("=" * 60)
+print("🚀 STARTING VIDEO PROJECT MANAGER")
+print("=" * 60)
+
+# Function to check and install packages
+def ensure_package(package_name, import_name=None):
+    if import_name is None:
+        import_name = package_name
+    
+    # Check if package is installed
+    if importlib.util.find_spec(import_name) is None:
+        print(f"📦 Installing {package_name}...")
+        try:
+            subprocess.check_call([sys.executable, "-m", "pip", "install", package_name])
+            print(f"✅ {package_name} installed successfully!")
+            return True
+        except Exception as e:
+            print(f"❌ Failed to install {package_name}: {e}")
+            return False
+    else:
+        print(f"✅ {package_name} already installed")
+        return True
+
+# Install required packages
+ensure_package("gradio")
+ensure_package("supabase")
+ensure_package("python-dotenv")
+
+# Now import after ensuring they're installed
 import gradio as gr
 import uuid
 import json
-import sys
-import os
 from datetime import datetime
 from supabase import create_client
 
-print("=" * 60)
-print("🚀 STARTING VIDEO PROJECT MANAGER WITH SUPABASE")
-print("=" * 60)
+print("\n" + "=" * 60)
 
 # =============================================
 # SUPABASE CONNECTION
@@ -30,8 +60,11 @@ if SUPABASE_URL and SUPABASE_KEY:
         print("✅ Supabase client initialized successfully!")
         
         # Test connection
-        test_result = supabase.table("projects").select("count", count="exact").limit(1).execute()
-        print(f"✅ Database connected! Projects found: {test_result.count}")
+        try:
+            test_result = supabase.table("projects").select("count", count="exact").limit(1).execute()
+            print(f"✅ Database connected! Projects found: {test_result.count}")
+        except Exception as e:
+            print(f"⚠️ Database connection test failed: {e}")
     except Exception as e:
         print(f"❌ Failed to connect to Supabase: {e}")
         supabase = None
@@ -76,16 +109,25 @@ def create_project(title, content, tags, image_prompt, channel):
                 result = supabase.table("projects").insert(project_data).execute()
                 db_result = result.data[0] if result.data else None
                 print(f"✅ Successfully saved to Supabase! Record ID: {db_result.get('id') if db_result else 'unknown'}")
+                
+                return {
+                    "status": "success",
+                    "project_id": project_id,
+                    "folder": folder_name,
+                    "title": title,
+                    "message": "Project created and saved to database!",
+                    "database_id": db_result.get('id') if db_result else None,
+                    "timestamp": datetime.now().isoformat()
+                }
+                
             except Exception as e:
                 print(f"❌ Supabase save failed: {e}")
-                # Still return success but note the database error
                 return {
                     "status": "partial_success",
                     "project_id": project_id,
                     "folder": folder_name,
                     "title": title,
-                    "message": "Project created in memory but database save failed",
-                    "error": str(e),
+                    "message": f"Project created but database save failed: {str(e)}",
                     "timestamp": datetime.now().isoformat()
                 }
         else:
@@ -99,20 +141,6 @@ def create_project(title, content, tags, image_prompt, channel):
                 "timestamp": datetime.now().isoformat()
             }
         
-        # Success response
-        response = {
-            "status": "success",
-            "project_id": project_id,
-            "folder": folder_name,
-            "title": title,
-            "message": "Project created and saved to database!",
-            "database_id": db_result.get('id') if db_result else None,
-            "timestamp": datetime.now().isoformat()
-        }
-        
-        print(f"✅ Returning success response")
-        return response
-        
     except Exception as e:
         print(f"❌ Error in create_project: {e}")
         import traceback
@@ -185,6 +213,8 @@ def health_check():
 # GRADIO INTERFACE
 # =============================================
 
+print("\n📝 Creating Gradio interface...")
+
 with gr.Blocks(title="Video Project Manager") as demo:
     gr.Markdown("# 🎬 Video Project Manager")
     
@@ -242,7 +272,8 @@ with gr.Blocks(title="Video Project Manager") as demo:
         # Debug Tab
         with gr.TabItem("🔧 Debug"):
             gr.Markdown("### System Information")
-            gr.JSON(label="Status", value=health_check)
+            status_json = gr.JSON(label="Status", value=health_check)
+            refresh_btn = gr.Button("🔄 Refresh Status")
             
             gr.Markdown("### Test API Commands")
             gr.Markdown(f"""
@@ -251,11 +282,6 @@ with gr.Blocks(title="Video Project Manager") as demo:
             curl -X POST "https://yukee1992-video-project-manager.hf.space/api/create" \\
               -H "Content-Type: application/json" \\
               -d '{{"data": ["Test", "Content", "tags", "prompt", "{{}}"]}}'
-            
-            # Test queue endpoint
-            curl -X POST "https://yukee1992-video-project-manager.hf.space/queue/join" \\
-              -H "Content-Type: application/json" \\
-              -d '{{"fn_index": 0, "data": ["Test", "Content", "tags", "prompt", "{{}}"], "session_hash": "test123"}}'
             ```
             """)
     
@@ -281,6 +307,12 @@ with gr.Blocks(title="Video Project Manager") as demo:
     )
     print("  ✅ Registered: /api/search")
     
+    refresh_btn.click(
+        fn=health_check,
+        inputs=[],
+        outputs=[status_json]
+    )
+    
     print("📡 API endpoints registered successfully!")
 
 # =============================================