Update app.py

app.py

@@ -4,107 +4,176 @@ 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)
 
 # =============================================
-# EXTENSIVE DEBUGGING SETUP
+# SUPABASE CONNECTION
 # =============================================
 
-print("=" * 60)
-print("🚀 STARTING VIDEO PROJECT MANAGER WITH DEBUG")
-print("=" * 60)
-print(f"Python version: {sys.version}")
-print(f"Gradio version: {gr.__version__}")
-print(f"Current time: {datetime.now().isoformat()}")
-print("=" * 60)
+# Get Supabase credentials from environment variables
+SUPABASE_URL = os.environ.get("SUPABASE_URL")
+SUPABASE_KEY = os.environ.get("SUPABASE_KEY")
 
-# Debug function to log all requests
-def debug_log(func_name, *args, **kwargs):
-    """Log function calls with arguments"""
-    print(f"\n🔍 DEBUG - Function: {func_name}")
-    print(f"  Arguments: {json.dumps(args, default=str, indent=2) if args else 'None'}")
-    print(f"  Keyword args: {kwargs if kwargs else 'None'}")
-    print(f"  Timestamp: {datetime.now().isoformat()}")
-    sys.stdout.flush()  # Force output to appear immediately
+print(f"\n🔧 Supabase Configuration:")
+print(f"   URL: {'✅ Set' if SUPABASE_URL else '❌ Missing'}")
+print(f"   Key: {'✅ Set' if SUPABASE_KEY else '❌ Missing'}")
+
+# Initialize Supabase client
+supabase = None
+if SUPABASE_URL and SUPABASE_KEY:
+    try:
+        supabase = create_client(SUPABASE_URL, 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}")
+    except Exception as e:
+        print(f"❌ Failed to connect to Supabase: {e}")
+        supabase = None
+else:
+    print("⚠️ Supabase credentials missing - running in memory-only mode")
 
 # =============================================
-# CORE FUNCTIONS WITH DEBUGGING
+# CORE FUNCTIONS
 # =============================================
 
 def create_project(title, content, tags, image_prompt, channel):
-    """Create a new project with debug logging"""
-    debug_log("create_project", title, content[:50] + "...", tags, image_prompt[:30] + "...", channel)
+    """Create a new project and save to Supabase"""
+    print(f"\n📝 Creating project: {title}")
+    print(f"   Content length: {len(content)}")
+    print(f"   Tags: {tags}")
+    print(f"   Image prompt: {image_prompt[:30]}...")
     
     try:
-        # Validate inputs
-        if not title or not content:
-            print("❌ ERROR: Missing required fields")
-            return {
-                "status": "error",
-                "message": "Title and content are required",
-                "received": {
-                    "title": bool(title),
-                    "content": bool(content)
-                }
-            }
-        
-        # Generate response
+        # Generate project data
         project_id = str(uuid.uuid4())[:8]
         folder_name = title.lower().replace(' ', '_').replace('/', '-')[:20]
         
+        project_data = {
+            "project_id": project_id,
+            "folder_name": folder_name,
+            "title": title,
+            "content": content[:5000],  # Limit content length
+            "tags": tags,
+            "image_prompt": image_prompt,
+            "channel_details": channel,
+            "status": "created",
+            "created_at": datetime.now().isoformat()
+        }
+        
+        print(f"📦 Project data prepared: {project_id}")
+        
+        # Save to Supabase if connected
+        db_result = None
+        if supabase:
+            try:
+                print("💾 Attempting to save to Supabase...")
+                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'}")
+            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),
+                    "timestamp": datetime.now().isoformat()
+                }
+        else:
+            print("⚠️ Supabase not connected - project only in memory")
+            return {
+                "status": "memory_only",
+                "project_id": project_id,
+                "folder": folder_name,
+                "title": title,
+                "message": "Project created in memory only (no database connection)",
+                "timestamp": datetime.now().isoformat()
+            }
+        
+        # Success response
         response = {
             "status": "success",
             "project_id": project_id,
             "folder": folder_name,
             "title": title,
-            "message": "Project created successfully!",
-            "timestamp": datetime.now().isoformat(),
-            "received_data": {
-                "title": title,
-                "content_length": len(content),
-                "tags": tags,
-                "has_image_prompt": bool(image_prompt),
-                "channel": channel
-            }
+            "message": "Project created and saved to database!",
+            "database_id": db_result.get('id') if db_result else None,
+            "timestamp": datetime.now().isoformat()
         }
         
-        print(f"✅ SUCCESS: Created project {project_id}")
+        print(f"✅ Returning success response")
         return response
         
     except Exception as e:
-        print(f"❌ ERROR in create_project: {str(e)}")
+        print(f"❌ Error in create_project: {e}")
         import traceback
         traceback.print_exc()
         return {
             "status": "error",
-            "message": f"Internal error: {str(e)}",
+            "message": f"Error: {str(e)}",
             "timestamp": datetime.now().isoformat()
         }
 
 def search_projects(term):
-    """Search for projects with debug logging"""
-    debug_log("search_projects", term)
+    """Search for projects in Supabase"""
+    print(f"\n🔍 Searching for: {term}")
     
-    try:
-        results = [
-            ["Test Video 1", "completed", datetime.now().strftime("%Y-%m-%d"), "test_1"],
-            ["How to Make Videos", "processing", datetime.now().strftime("%Y-%m-%d"), "tutorial"],
-            [term or "Search Result", "created", datetime.now().strftime("%Y-%m-%d"), "search_result"]
+    if not supabase:
+        print("⚠️ No database connection - returning demo data")
+        return [
+            ["Demo Project 1", "memory", datetime.now().strftime("%Y-%m-%d"), "demo_1"],
+            ["How to Make Videos", "demo", datetime.now().strftime("%Y-%m-%d"), "tutorial"],
+            [term or "Search Demo", "demo", datetime.now().strftime("%Y-%m-%d"), "demo"]
         ]
+    
+    try:
+        if term and term.strip():
+            result = supabase.table("projects")\
+                .select("title, status, created_at, folder_name")\
+                .ilike("title", f"%{term}%")\
+                .limit(20)\
+                .execute()
+        else:
+            result = supabase.table("projects")\
+                .select("title, status, created_at, folder_name")\
+                .limit(20)\
+                .order("created_at", desc=True)\
+                .execute()
+        
+        projects = []
+        for p in result.data:
+            created = p.get("created_at", "")
+            if created and len(created) > 10:
+                created = created[:10]
+            projects.append([
+                p.get("title", "Untitled"),
+                p.get("status", "unknown"),
+                created,
+                p.get("folder_name", "")
+            ])
         
-        print(f"✅ SUCCESS: Found {len(results)} results")
-        return results
+        print(f"✅ Found {len(projects)} projects")
+        return projects if projects else [["No projects found", "", "", ""]]
         
     except Exception as e:
-        print(f"❌ ERROR in search_projects: {str(e)}")
-        return [["Error", str(e), "", ""]]
+        print(f"❌ Search error: {e}")
+        return [["Error", str(e)[:50], "", ""]]
 
 def health_check():
-    """Simple health check endpoint"""
-    debug_log("health_check")
+    """Check system health"""
     return {
         "status": "healthy",
+        "supabase_connected": bool(supabase),
         "timestamp": datetime.now().isoformat(),
-        "api_version": "1.0",
         "endpoints": {
             "create": "/api/create",
             "search": "/api/search",
@@ -112,76 +181,53 @@ def health_check():
         }
     }
 
-def debug_info():
-    """Return debug information"""
-    return {
-        "python_version": sys.version,
-        "gradio_version": gr.__version__,
-        "timestamp": datetime.now().isoformat(),
-        "environment": {
-            "space_id": os.environ.get("SPACE_ID", "unknown"),
-            "hf_token": "***" if os.environ.get("HF_TOKEN") else None
-        }
-    }
-
 # =============================================
 # GRADIO INTERFACE
 # =============================================
 
-print("\n📝 Creating Gradio interface...")
-
-with gr.Blocks(title="Video Project Manager - DEBUG") as demo:
-    gr.Markdown("# 🎬 Video Project Manager - DEBUG MODE")
-    gr.Markdown("This version includes extensive logging. Check the Space logs for debug output.")
+with gr.Blocks(title="Video Project Manager") as demo:
+    gr.Markdown("# 🎬 Video Project Manager")
     
     # Status indicator
-    gr.Markdown(f"### ✅ Status: Running | Gradio v{gr.__version__} | Python {sys.version.split()[0]}")
+    if supabase:
+        gr.Markdown("### ✅ Connected to Supabase")
+    else:
+        gr.Markdown("### ⚠️ Running in Memory Mode (No Database)")
     
     with gr.Tabs():
-        # =========================================
-        # TAB 1: CREATE PROJECT
-        # =========================================
+        # Create Project Tab
         with gr.TabItem("📁 Create Project"):
             with gr.Row():
                 with gr.Column():
                     title_input = gr.Textbox(
                         label="Title",
-                        placeholder="Enter video title",
-                        info="This will be logged in debug"
+                        placeholder="Enter video title"
                     )
                     content_input = gr.Textbox(
                         label="Content",
                         lines=3,
-                        placeholder="Enter TTS content",
-                        info="Content length will be logged"
+                        placeholder="Enter TTS content"
                     )
                     tags_input = gr.Textbox(
                         label="Tags",
-                        placeholder="tag1, tag2, tag3",
-                        info="Tags for categorization"
+                        placeholder="tag1, tag2, tag3"
                     )
                     image_input = gr.Textbox(
                         label="Image Prompt",
                         lines=2,
-                        placeholder="Describe images",
-                        info="Prompt for image generation"
+                        placeholder="Describe images"
                     )
                     channel_input = gr.Textbox(
                         label="Channel Details",
-                        value='{"platform": "youtube"}',
-                        info="JSON string with channel info"
+                        value='{"platform": "youtube"}'
                     )
                     
                     create_btn = gr.Button("🚀 Create Project", variant="primary")
                     
                 with gr.Column():
                     output = gr.JSON(label="Result")
-                    gr.Markdown("### Debug Info")
-                    gr.Markdown("Check Space logs for detailed function calls")
         
-        # =========================================
-        # TAB 2: SEARCH
-        # =========================================
+        # Search Tab
         with gr.TabItem("🔍 Search"):
             search_input = gr.Textbox(
                 label="Search Term",
@@ -190,102 +236,64 @@ with gr.Blocks(title="Video Project Manager - DEBUG") as demo:
             search_btn = gr.Button("🔍 Search", variant="secondary")
             results = gr.Dataframe(
                 headers=["Title", "Status", "Date", "Folder"],
-                value=[["Demo", "ready", "2026-02-16", "demo"]],
                 label="Search Results"
             )
         
-        # =========================================
-        # TAB 3: DEBUG INFO
-        # =========================================
+        # Debug Tab
         with gr.TabItem("🔧 Debug"):
             gr.Markdown("### System Information")
-            debug_json = gr.JSON(label="Debug Info", value=debug_info)
-            refresh_btn = gr.Button("🔄 Refresh Debug Info")
+            gr.JSON(label="Status", value=health_check)
             
-            gr.Markdown("### API Test")
-            gr.Markdown("""
+            gr.Markdown("### Test API Commands")
+            gr.Markdown(f"""
             ```bash
             # Test create endpoint
             curl -X POST "https://yukee1992-video-project-manager.hf.space/api/create" \\
               -H "Content-Type: application/json" \\
-              -d '{"data": ["Test", "Content", "tags", "prompt", "{}"]}'
+              -d '{{"data": ["Test", "Content", "tags", "prompt", "{{}}"]}}'
             
-            # Test health endpoint
-            curl -X POST "https://yukee1992-video-project-manager.hf.space/api/health" \\
+            # Test queue endpoint
+            curl -X POST "https://yukee1992-video-project-manager.hf.space/queue/join" \\
               -H "Content-Type: application/json" \\
-              -d '{}'
+              -d '{{"fn_index": 0, "data": ["Test", "Content", "tags", "prompt", "{{}}"], "session_hash": "test123"}}'
             ```
             """)
-            
-            health_btn = gr.Button("🏥 Test Health Endpoint")
-            health_output = gr.JSON(label="Health Check")
     
     # =============================================
-    # EVENT HANDLERS WITH API NAMES
+    # EVENT HANDLERS
     # =============================================
     
-    print("📡 Registering API endpoints...")
+    print("\n📡 Registering API endpoints...")
     
-    # Create endpoint
     create_btn.click(
         fn=create_project,
         inputs=[title_input, content_input, tags_input, image_input, channel_input],
         outputs=[output],
-        api_name="create"  # This creates /api/create
+        api_name="create"
     )
     print("  ✅ Registered: /api/create")
     
-    # Search endpoint
     search_btn.click(
         fn=search_projects,
         inputs=[search_input],
         outputs=[results],
-        api_name="search"  # This creates /api/search
+        api_name="search"
     )
     print("  ✅ Registered: /api/search")
     
-    # Health endpoint
-    health_btn.click(
-        fn=health_check,
-        inputs=[],
-        outputs=[health_output],
-        api_name="health"  # This creates /api/health
-    )
-    print("  ✅ Registered: /api/health")
-    
-    # Refresh debug
-    refresh_btn.click(
-        fn=debug_info,
-        inputs=[],
-        outputs=[debug_json]
-    )
-    
     print("📡 API endpoints registered successfully!")
 
 # =============================================
-# LAUNCH WITH DEBUG
+# LAUNCH
 # =============================================
 if __name__ == "__main__":
     print("\n" + "=" * 60)
     print("🌐 Launching Gradio server...")
-    print(f"   Server: http://0.0.0.0:7860")
-    print(f"   API prefix: /api")
-    print(f"   Registered endpoints:")
-    print(f"     - POST /api/create")
-    print(f"     - POST /api/search")
-    print(f"     - POST /api/health")
-    print("\n🔍 Test commands:")
-    print('   curl -X POST "https://yukee1992-video-project-manager.hf.space/api/health" -H "Content-Type: application/json" -d "{}"')
-    print('   curl -X POST "https://yukee1992-video-project-manager.hf.space/api/create" -H "Content-Type: application/json" -d \'{"data": ["Test", "Content", "tags", "prompt", "{}"]}\'')
-    print("=" * 60)
-    print("\n📋 IMPORTANT: Check the Space logs for debug output!")
     print("=" * 60)
     
-    # Launch with debug mode
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
         share=False,
-        debug=True,  # Enable debug mode
-        show_error=True  # Show errors in UI
+        debug=True
     )