animate_bloch returns html not funcanimation

Author: vudangbaovy

quantum_app_backend/app.py

@@ -174,18 +174,18 @@ def Qtrace(gate, init_state, mag, t2):
 @app.route('/receive_data/qft', methods=['GET'])
 def Qfouriertransform():
     logger.debug("Startting QFT generator")
-    anim = None
+    anim_js = None
     try:
         qft_ode = QFTStepByStepODE(initial_state_str='011')
         qft_ode.run(n_step=6)
-        anim = qft_ode.animate_bloch()
+        anim_js = qft_ode.animate_bloch()
 
-        if not anim:
+        if not anim_js:
             raise Exception("Error generating QFT model")
     except Exception as e:
         logger.error(f"Error generating QFT model: {str(e)}", exc_info=True)
         raise
-    return anim
+    return anim_js
 
 @socketio.on('connect')
 def handle_connect():

quantum_app_backend/model_generators/QuantumFourierTransform.py

@@ -396,19 +396,24 @@ def animate(frame_idx):
             blit=False
         )
 
+        anim_js = anim.to_jshtml(fps=30)
         # plt.show()
         # or, to save:
-
-        anim_js = anim.to_jshtml(fps=30)
-        with open(f"qft_with_bloch_and_probs_{self.n_step}.html", "w") as f:
-            f.write(anim_js)
-
+        
+        ## Uncomment to save file locally
+        # path = os.path.abspath(os.path.join(f"quantum_app_backend/cache/qft/qft_with_bloch_and_probs_{self.n_step}.html"))
+        
+        # if not Path(path).exists():
+        #     with open(path, "w", encoding="utf-8") as f:
+        #         portalocker.lock(f, portalocker.LOCK_EX)
+        #         f.write(anim_js)
+        
         # html_writer = HTMLWriter()
         # anim.save(f"qft_with_bloch_and_probs_{self.n_step}.html", writer=html_writer)
 
         # print(f"Animation saved as qft_with_bloch_and_probs_{self.n_step}.html")        
         #
-        return anim
+        return anim_js
 
 
 ##############################################################################