QLBM: Upload Window

qlbm_embedded.py

@@ -1239,13 +1239,21 @@ def process_uploaded_job_result():
         _state.qlbm_job_upload_filename = filename
         log_to_console(f"Processing file: {filename}")
         
-        # Decode base64 content
+        # Decode content - handle both bytes and string (base64 data URI)
         content = file_data.get("content", "") if isinstance(file_data, dict) else ""
-        if content.startswith("data:"):
-            content = content.split(",", 1)[1]  # Remove data URI prefix
         
-        raw_bytes = base64.b64decode(content)
-        json_str = raw_bytes.decode("utf-8")
+        # Check if content is bytes or string
+        if isinstance(content, bytes):
+            # Content is already raw bytes, decode directly
+            json_str = content.decode("utf-8")
+            log_to_console("Content received as raw bytes")
+        else:
+            # Content is a string, possibly base64 data URI
+            if content.startswith("data:"):
+                content = content.split(",", 1)[1]  # Remove data URI prefix
+            raw_bytes = base64.b64decode(content)
+            json_str = raw_bytes.decode("utf-8")
+            log_to_console("Content decoded from base64 string")
         
         # Parse timesteps from user input
         # User provides Total Time T, we generate T_list = [1, 2, ..., T]
@@ -1375,8 +1383,8 @@ def process_uploaded_job_result():
         _state.qlbm_job_upload_success = f"✓ Successfully processed {len(output)} timestep(s) from {filename}"
         
         # Update the Plotly figure widget
-        if hasattr(_ctrl, "qlbm_qiskit_plot_update"):
-            _ctrl.qlbm_qiskit_plot_update(fig)
+        if hasattr(_ctrl, "qlbm_qiskit_result_update"):
+            _ctrl.qlbm_qiskit_result_update(fig)
         
         log_to_console(f"Results ready! {len(output)} frames generated.")
         
@@ -1579,48 +1587,18 @@ async def _run_simulation_async():
             
             output, plotly_fig, init_state_prep_circ = await loop.run_in_executor(executor, _run_ibm_qpu_blocking)
             
-            # Step 3: Creating Plots is already handled in get_job_result, now add T=0 snapshot
+            # Step 3: Finalizing results (T=0 snapshot removed - only show T=1 onwards)
             _state.qlbm_simulation_progress = 92
-            _state.qlbm_status_message = "Step 3: Adding T=0 snapshot..."
+            _state.qlbm_status_message = "Step 3: Finalizing results..."
             await _qlbm_flush_async()
             
-            # Generate T=0 initial snapshot and prepend to output
-            log_to_console("Generating T=0 initial distribution snapshot...")
-            try:
-                initial_snapshot = show_initial_distribution(
-                    n=params["n"],
-                    init_state_name=params["init_state_name"],
-                    sine_k_x=params["sine_k_x"],
-                    sine_k_y=params["sine_k_y"],
-                    sine_k_z=params["sine_k_z"],
-                    gauss_cx=params["gauss_cx"],
-                    gauss_cy=params["gauss_cy"],
-                    gauss_cz=params["gauss_cz"],
-                    gauss_sigma=params["gauss_sigma"],
-                    mdd_kx_log2=params["mdd_kx_log2"],
-                    mdd_ky_log2=params["mdd_ky_log2"],
-                    mdd_kz_log2=params["mdd_kz_log2"],
-                    plot=False,
-                    return_data=True,
-                    init_state_circuit=init_state_prep_circ,
-                )
-                if initial_snapshot is not None:
-                    initial_density, X = initial_snapshot
-                    # Prepend T=0 to outputs
-                    output = [initial_density] + list(output)
-                    extended_T_list = [0] + list(params["T_list"])
-                    # Recreate figure with T=0 included
-                    plotly_fig = _create_slider_figure(output, extended_T_list, X)
-                    log_to_console("T=0 snapshot added successfully.")
-                else:
-                    extended_T_list = params["T_list"]
-            except Exception as exc:
-                log_to_console(f"Warning: Could not generate T=0 snapshot: {exc}")
-                extended_T_list = params["T_list"]
+            # Use T_list directly (no T=0 prepend)
+            result_T_list = list(params["T_list"])
+            log_to_console(f"Results available for T={result_T_list}")
             
             # Store results
             simulation_data_frames = output
-            simulation_times = [float(t) for t in extended_T_list]
+            simulation_times = [float(t) for t in result_T_list]
             
             # Update UI
             if hasattr(_ctrl, "qlbm_qiskit_result_update"):
@@ -1628,7 +1606,7 @@ async def _run_simulation_async():
             
             _state.qlbm_max_time_step = len(output) - 1
             _state.qlbm_time_val = 0
-            _state.qlbm_time_slider_labels = [f"T={t}" for t in extended_T_list]
+            _state.qlbm_time_slider_labels = [f"T={t}" for t in result_T_list]
             _state.qlbm_simulation_has_run = True
             _state.qlbm_qiskit_mode = True
             
@@ -1696,48 +1674,18 @@ async def _run_simulation_async():
             
             output, plotly_fig, init_state_prep_circ = await loop.run_in_executor(executor, _run_ionq_qpu_blocking)
             
-            # Step 3: Creating Plots is already handled in get_job_result, now add T=0 snapshot
+            # Step 3: Finalizing results (T=0 snapshot removed - only show T=1 onwards)
             _state.qlbm_simulation_progress = 92
-            _state.qlbm_status_message = "Step 3: Adding T=0 snapshot..."
+            _state.qlbm_status_message = "Step 3: Finalizing results..."
             await _qlbm_flush_async()
             
-            # Generate T=0 initial snapshot and prepend to output
-            log_to_console("Generating T=0 initial distribution snapshot...")
-            try:
-                initial_snapshot = show_initial_distribution(
-                    n=params["n"],
-                    init_state_name=params["init_state_name"],
-                    sine_k_x=params["sine_k_x"],
-                    sine_k_y=params["sine_k_y"],
-                    sine_k_z=params["sine_k_z"],
-                    gauss_cx=params["gauss_cx"],
-                    gauss_cy=params["gauss_cy"],
-                    gauss_cz=params["gauss_cz"],
-                    gauss_sigma=params["gauss_sigma"],
-                    mdd_kx_log2=params["mdd_kx_log2"],
-                    mdd_ky_log2=params["mdd_ky_log2"],
-                    mdd_kz_log2=params["mdd_kz_log2"],
-                    plot=False,
-                    return_data=True,
-                    init_state_circuit=init_state_prep_circ,
-                )
-                if initial_snapshot is not None:
-                    initial_density, X = initial_snapshot
-                    # Prepend T=0 to outputs
-                    output = [initial_density] + list(output)
-                    extended_T_list = [0] + list(params["T_list"])
-                    # Recreate figure with T=0 included
-                    plotly_fig = _create_slider_figure(output, extended_T_list, X)
-                    log_to_console("T=0 snapshot added successfully.")
-                else:
-                    extended_T_list = params["T_list"]
-            except Exception as exc:
-                log_to_console(f"Warning: Could not generate T=0 snapshot: {exc}")
-                extended_T_list = params["T_list"]
+            # Use T_list directly (no T=0 prepend)
+            result_T_list = list(params["T_list"])
+            log_to_console(f"Results available for T={result_T_list}")
             
             # Store results
             simulation_data_frames = output
-            simulation_times = [float(t) for t in extended_T_list]
+            simulation_times = [float(t) for t in result_T_list]
             
             # Update UI
             if hasattr(_ctrl, "qlbm_qiskit_result_update"):
@@ -1745,7 +1693,7 @@ async def _run_simulation_async():
             
             _state.qlbm_max_time_step = len(output) - 1
             _state.qlbm_time_val = 0
-            _state.qlbm_time_slider_labels = [f"T={t}" for t in extended_T_list]
+            _state.qlbm_time_slider_labels = [f"T={t}" for t in result_T_list]
             _state.qlbm_simulation_has_run = True
             _state.qlbm_qiskit_mode = True