Update app.py

app.py

@@ -4,6 +4,7 @@ import torch
 import gradio as gr
 from huggingface_hub import hf_hub_download, snapshot_download
 from pathlib import Path
+import importlib.util
 
 class CognitiveNetworkDemo:
     def __init__(self):
@@ -19,44 +20,36 @@ class CognitiveNetworkDemo:
             repo_type="model",
             local_dir="./model_repo"
         ))
-        
-        # Add model directory to Python path
-        model_dir = self.repo_path
-        if str(model_dir) not in sys.path:
-            sys.path.insert(0, str(model_dir))
-            
-        # Print directory contents for debugging
-        print("Repository contents:")
-        for path in model_dir.rglob("*"):
-            print(f"  {path.relative_to(model_dir)}")
+    
+    def import_module_from_file(self, module_name, file_path):
+        """Import a module from file path"""
+        spec = importlib.util.spec_from_file_location(module_name, file_path)
+        module = importlib.util.module_from_spec(spec)
+        sys.modules[module_name] = module
+        spec.loader.exec_module(module)
+        return module
             
     def load_model(self):
         """Load the model if not already loaded"""
         if self.model is None:
             try:
-                # Try different import paths
-                try:
-                    from network import DynamicCognitiveNet
-                except ImportError:
-                    try:
-                        from cognitive_net.network import DynamicCognitiveNet
-                    except ImportError:
-                        # Search for the network.py file
-                        network_files = list(self.repo_path.rglob("network.py"))
-                        if not network_files:
-                            raise ImportError("Tidak dapat menemukan file network.py")
-                        
-                        # Add the parent directory of the first found network.py to path
-                        network_dir = network_files[0].parent
-                        if str(network_dir) not in sys.path:
-                            sys.path.insert(0, str(network_dir))
-                        
-                        # Try import again
-                        from network import DynamicCognitiveNet
+                # Find and import required modules
+                network_path = next(self.repo_path.rglob("network.py"))
+                node_path = next(self.repo_path.rglob("node.py"))
+                memory_path = next(self.repo_path.rglob("memory.py"))
+                
+                # Import modules using absolute paths
+                node_module = self.import_module_from_file("node", node_path)
+                memory_module = self.import_module_from_file("memory", memory_path)
+                network_module = self.import_module_from_file("network", network_path)
                 
-                self.model = DynamicCognitiveNet(input_size=5, output_size=2)
+                # Create model instance
+                self.model = network_module.DynamicCognitiveNet(input_size=5, output_size=2)
                 
-            except ImportError as e:
+            except StopIteration:
+                raise ImportError("Tidak dapat menemukan file modul yang diperlukan")
+            except Exception as e:
+                print("Debug - repo path:", self.repo_path)
                 print("Debug - sys.path:", sys.path)
                 raise ImportError(f"Gagal mengimpor model: {str(e)}")
         return self.model