fix run server

Author: Koldim2001

main_optimized.py

@@ -74,7 +74,6 @@ def proc_show_node(queue_in: Queue, config: dict):
         video_saver_node = VideoSaverNode(config["video_saver_node"])
     if show_in_web:
         video_server_node = VideoServer(config)
-        video_server_node.run()
     while True:
         ts0 = time()
         frame_element = queue_in.get()
@@ -84,7 +83,6 @@ def proc_show_node(queue_in: Queue, config: dict):
             video_saver_node.process(frame_element)
         if show_in_web:
             if isinstance(frame_element, VideoEndBreakElement):
-                video_server_node.stop_server()
                 break
             video_server_node.update_image(frame_element.frame_result)
         ts2 = time()
@@ -96,7 +94,7 @@ def proc_show_node(queue_in: Queue, config: dict):
                 + f"put {(time()-ts2) * 1000:.0f}"
             )
         if isinstance(frame_element, VideoEndBreakElement):
-                break
+            break
 
 
 @hydra.main(version_base=None, config_path="configs", config_name="app_config")

nodes/FlaskServerVideoNode.py

@@ -28,7 +28,9 @@ def __init__(self, config):
         self.host_ip = config_server["host_ip"]
         self.port = config_server["port"]
         self.index_page = config_server["index_page"]
+        
         self._frame = np.zeros(shape=(640, 480), dtype=np.uint8)
+        self.run()
 
     def _index(self) -> str:
         return render_template(self.index_page)

nodes/SendInfoDBNode.py

@@ -52,6 +52,7 @@ def __init__(self, config: dict) -> None:
             try:
                 self.cursor.execute(drop_table_query)
                 self.connection.commit()
+                logger.info(f"The table has been deleted")
             except (Exception, psycopg2.Error) as error:
                 logger.error(f"Error while dropping table:: {error}")