Update a.py

a.py

@@ -269,9 +269,14 @@ def parse_url_params(request: gr.Request):
     """Parse taskid from URL parameters"""
     task_id = None
     if request:
-        query_params = parse_qs(urlparse(request.request.url).query)
-        if 'taskid' in query_params:
-            task_id = query_params['taskid'][0]
+        try:
+            query_params = parse_qs(urlparse(request.request.url).query)
+            if 'taskid' in query_params:
+                task_id = query_params['taskid'][0]
+                # Remove any whitespace
+                task_id = task_id.strip()
+        except Exception as e:
+            print(f"Error parsing URL params: {e}")
     
     return task_id
 
@@ -280,10 +285,10 @@ with gr.Blocks() as app:
     gr.Markdown("# 🎵 Suno Song Generator")
     gr.Markdown("Create songs from lyrics and style using Suno AI")
     
-    # Store for URL parameters
-    url_task_id = gr.State(value="")
+    # We'll use a hidden component to track initial load
+    initial_load_done = gr.State(value=False)
     
-    with gr.Tab("🎶 Generate Song", id=0) as tab_generate:
+    with gr.Tab("🎶 Generate Song", id="generate_tab") as tab_generate:
         with gr.Row():
             with gr.Column(scale=1):
                 # Lyrics Input
@@ -351,7 +356,7 @@ with gr.Blocks() as app:
                     value="### Ready to generate!\n\nEnter lyrics and settings, then click 'Generate Song'"
                 )
     
-    with gr.Tab("🔍 Check Any Task", id=1) as tab_check:
+    with gr.Tab("🔍 Check Any Task", id="check_tab") as tab_check:
         with gr.Row():
             with gr.Column(scale=1):
                 gr.Markdown("### Check Task Status")
@@ -380,7 +385,7 @@ with gr.Blocks() as app:
                     value="### Enter a Task ID above\n\nPaste any Suno Task ID to check its current status and results."
                 )
     
-    with gr.Tab("📚 Instructions", id=2):
+    with gr.Tab("📚 Instructions", id="instructions_tab"):
         gr.Markdown("""
         ## 📖 How to Use This App
         
@@ -439,7 +444,7 @@ with gr.Blocks() as app:
         - Generation may have failed
         """)
     
-    with gr.Tab("📚 Less Instructions", id=3):
+    with gr.Tab("📚 Less Instructions", id="less_instructions_tab"):
         gr.Markdown("""
         ## 📖 How to Use This App
         
@@ -513,16 +518,30 @@ with gr.Blocks() as app:
     def on_page_load(request: gr.Request):
         """Handle URL parameters when page loads"""
         task_id = parse_url_params(request)
+        
         if task_id:
-            # Set the task ID in the input field
-            return task_id, get_task_info(task_id), gr.update(selected=1)
-        return "", "### Enter a Task ID above\n\nPaste any Suno Task ID to check its current status and results.", gr.update(selected=0)
+            # We have a task ID from URL, return it and fetch results
+            task_result = get_task_info(task_id)
+            return (
+                task_id,  # For check_task_id
+                task_result,  # For check_output
+                gr.Tabs(selected="check_tab"),  # Switch to check tab
+                True  # Mark as loaded
+            )
+        else:
+            # No task ID in URL, stay on first tab
+            return (
+                "",  # Empty check_task_id
+                "### Enter a Task ID above\n\nPaste any Suno Task ID to check its current status and results.",  # Default message
+                gr.Tabs(selected="generate_tab"),  # Stay on generate tab
+                True  # Mark as loaded
+            )
     
     # Load URL parameters when the app starts
     app.load(
         fn=on_page_load,
         inputs=[],
-        outputs=[check_task_id, check_output, gr.Tabs(selected=tab_check)],
+        outputs=[check_task_id, check_output, gr.Tabs(), initial_load_done],
         queue=False
     )