Multi-process to single-process

mm_story_agent/__init__.py

@@ -45,16 +45,22 @@ class MMStoryAgent:
         for modality in self.modalities:
             agents[modality] = self.modality_agent_class[modality](config[modality + "_generation"])
 
-        processes = []
-        return_dict = mp.Manager().dict()
+        # processes = []
+        # return_dict = mp.Manager().dict()
 
-        for modality in self.modalities:
-            p = mp.Process(target=self.call_modality_agent, args=(agents[modality], pages, story_dir / modality, return_dict), daemon=False)
-            processes.append(p)
-            p.start()
+        # for modality in self.modalities:
+        #     p = mp.Process(target=self.call_modality_agent, args=(agents[modality], pages, story_dir / modality, return_dict), daemon=False)
+        #     processes.append(p)
+        #     p.start()
         
-        for p in processes:
-            p.join()
+        # for p in processes:
+        #     p.join()
+
+
+        return_dict = {}
+        
+        for modality in self.modalities:
+            self.call_modality_agent(agents[modality], pages, story_dir / modality, return_dict)
 
         for modality, result in return_dict.items():
             try: