Update src/streamlit_app.py

src/streamlit_app.py

@@ -103,6 +103,7 @@ class GitHubRepoAnalyzer:
         if not tree:
             return "📁 No files found in the specified path."
         
+        # Header information
         output = []
         output.append(f"📦 Repository: {tree_data['owner']}/{tree_data['repo']}")
         output.append(f"🌿 Branch: {tree_data['branch']}")
@@ -111,15 +112,93 @@ class GitHubRepoAnalyzer:
         output.append(f"📊 Total items: {len(tree)}")
         output.append("\n" + "="*60 + "\n")
         
-        sorted_tree = sorted(tree, key=lambda x: (x['type'] != 'tree', x['path']))
-        
-        for item in sorted_tree:
-            icon = "📁" if item['type'] == 'tree' else "📄"
-            size_info = f" ({item['size']} bytes)" if item['type'] == 'blob' and item['size'] > 0 else ""
-            output.append(f"{icon} {item['path']}{size_info}")
+        # Build tree structure
+        tree_structure = self._build_tree_structure(tree, tree_data.get('subfolder', ''))
+        output.append(tree_structure)
         
         return "\n".join(output)
     
+    def _build_tree_structure(self, tree_items: List[Dict], subfolder: str = '') -> str:
+        """Build a hierarchical tree structure with proper indentation and lines."""
+        if not tree_items:
+            return ""
+        
+        # Create a directory structure
+        dir_structure = {}
+        
+        # Process all items and build directory structure
+        for item in tree_items:
+            path = item['path']
+            
+            # Remove subfolder prefix if present
+            if subfolder and path.startswith(subfolder):
+                path = path[len(subfolder):].lstrip('/')
+            
+            if not path:  # Skip empty paths
+                continue
+                
+            parts = path.split('/')
+            current = dir_structure
+            
+            # Build nested structure
+            for i, part in enumerate(parts):
+                if part not in current:
+                    current[part] = {
+                        '_type': 'tree' if i < len(parts) - 1 or item['type'] == 'tree' else 'blob',
+                        '_size': item.get('size', 0) if i == len(parts) - 1 else 0,
+                        '_children': {}
+                    }
+                current = current[part]['_children']
+        
+        # Generate tree display
+        return self._format_tree_recursive(dir_structure, "", True)
+    
+    def _format_tree_recursive(self, structure: Dict, prefix: str = "", is_root: bool = False) -> str:
+        """Recursively format the tree structure with proper tree lines."""
+        lines = []
+        items = list(structure.items())
+        
+        # Sort: directories first, then files, both alphabetically
+        items.sort(key=lambda x: (x[1]['_type'] != 'tree', x[0].lower()))
+        
+        for i, (name, data) in enumerate(items):
+            is_last = i == len(items) - 1
+            
+            # Determine the tree characters
+            if is_root:
+                current_prefix = ""
+                next_prefix = ""
+            else:
+                current_prefix = prefix + ("└── " if is_last else "├── ")
+                next_prefix = prefix + ("    " if is_last else "│   ")
+            
+            # Format current item
+            if data['_type'] == 'tree':
+                # Directory
+                display_name = f"{name}/"
+                if not is_root:
+                    lines.append(f"{current_prefix}{display_name}")
+                else:
+                    lines.append(f"{display_name}")
+                
+                # Recursively add children
+                children_output = self._format_tree_recursive(
+                    data['_children'], 
+                    next_prefix if not is_root else "", 
+                    False
+                )
+                if children_output:
+                    lines.append(children_output)
+            else:
+                # File
+                size_info = f" ({data['_size']} bytes)" if data['_size'] > 0 else ""
+                if not is_root:
+                    lines.append(f"{current_prefix}{name}{size_info}")
+                else:
+                    lines.append(f"{name}{size_info}")
+        
+        return "\n".join(lines)
+    
     def get_file_content(self, owner: str, repo: str, branch: str, file_path: str) -> str:
         """Get content of a specific file."""
         try:
@@ -196,7 +275,6 @@ if 'files_content' not in st.session_state:
 # Initialize analyzer
 analyzer = GitHubRepoAnalyzer()
 
-
 # Main title and description
 st.title("🌳 GitHub Repository Tree Generator")
 st.markdown("""