Update app.py

app.py

@@ -21,17 +21,43 @@ class CognitiveNetworkDemo:
         ))
         
         # Add model directory to Python path
-        if str(self.repo_path) not in sys.path:
-            sys.path.insert(0, str(self.repo_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 load_model(self):
         """Load the model if not already loaded"""
         if self.model is None:
             try:
-                # Import here to ensure path is set up
-                from model_repo.cognitive_net.network import DynamicCognitiveNet
+                # 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
+                
                 self.model = DynamicCognitiveNet(input_size=5, output_size=2)
+                
             except ImportError as e:
+                print("Debug - sys.path:", sys.path)
                 raise ImportError(f"Gagal mengimpor model: {str(e)}")
         return self.model
 
@@ -58,7 +84,7 @@ class CognitiveNetworkDemo:
         except ValueError as e:
             return f"Error dalam format input: {str(e)}"
         except Exception as e:
-            return f"Error: {str(e)}"
+            return f"Error: {str(e)}\n\nDebug info:\nRepo path: {self.repo_path}"
 
 def main():
     # Initialize the demo