removed json dump

app.py

@@ -1,7 +1,7 @@
 import json
 import os
 import base64
-from typing import Dict, List
+from typing import Dict, List, Any
 import requests
 import gradio as gr
 from dotenv import load_dotenv
@@ -22,7 +22,7 @@ class GitHubMCPServer:
             "Accept": "application/vnd.github.v3+json"
         }
     
-    def get_repository_info(self, owner: str, repo: str) -> str:
+    def get_repository_info(self, owner: str, repo: str) -> dict:
         """Get basic repository information"""
         try:
             url = f"https://api.github.com/repos/{owner}/{repo}"
@@ -30,33 +30,35 @@ class GitHubMCPServer:
             
             if response.status_code == 200:
                 data = response.json()
-                result = {
+                return {
                     "success": True,
-                    "data": {
-                        "name": data["name"],
-                        "full_name": data["full_name"],
-                        "description": data.get("description", ""),
-                        "language": data.get("language", ""),
-                        "size": data["size"],
-                        "stars": data["stargazers_count"],
-                        "forks": data["forks_count"],
-                        "default_branch": data["default_branch"],
-                        "created_at": data["created_at"],
-                        "updated_at": data["updated_at"]
-                    }
+                    "repository_name": data["name"],
+                    "full_name": data["full_name"],
+                    "description": data.get("description", "No description available"),
+                    "primary_language": data.get("language", "Unknown"),
+                    "size_kb": data["size"],
+                    "stars": data["stargazers_count"],
+                    "forks": data["forks_count"],
+                    "default_branch": data["default_branch"],
+                    "created_date": data["created_at"][:10],
+                    "last_updated": data["updated_at"][:10],
+                    "is_private": data["private"],
+                    "clone_url": data["clone_url"]
                 }
             else:
-                result = {
+                return {
                     "success": False,
-                    "error": f"GitHub API error: {response.status_code}"
+                    "error": f"Repository not found or inaccessible (HTTP {response.status_code})",
+                    "status_code": response.status_code
                 }
             
-            return json.dumps(result, indent=2)
-            
         except Exception as e:
-            return json.dumps({"success": False, "error": str(e)}, indent=2)
+            return {
+                "success": False,
+                "error": f"Failed to fetch repository information: {str(e)}"
+            }
     
-    def get_file_content(self, owner: str, repo: str, path: str) -> str:
+    def get_file_content(self, owner: str, repo: str, path: str) -> dict:
         """Get content of a specific file"""
         try:
             url = f"https://api.github.com/repos/{owner}/{repo}/contents/{path}"
@@ -66,52 +68,84 @@ class GitHubMCPServer:
                 data = response.json()
                 if data["type"] == "file" and "content" in data:
                     # Decode base64 content
-                    content = base64.b64decode(data["content"]).decode('utf-8')
-                    result = {
-                        "success": True,
-                        "data": {
-                            "path": path,
-                            "content": content[:2000] + "..." if len(content) > 2000 else content,  # Truncate for display
-                            "size": len(content),
+                    try:
+                        content = base64.b64decode(data["content"]).decode('utf-8')
+                        file_extension = path.split('.')[-1].lower() if '.' in path else 'unknown'
+                        
+                        return {
+                            "success": True,
+                            "file_path": path,
+                            "file_name": path.split('/')[-1],
+                            "file_extension": file_extension,
+                            "content": content,
+                            "content_length": len(content),
+                            "line_count": len(content.split('\n')),
                             "sha": data["sha"],
-                            "encoding": data.get("encoding", "base64")
+                            "download_url": data.get("download_url", ""),
+                            "is_binary": False
+                        }
+                    except UnicodeDecodeError:
+                        return {
+                            "success": False,
+                            "error": "File contains binary data that cannot be decoded as text",
+                            "file_path": path,
+                            "is_binary": True
                         }
-                    }
                 else:
-                    result = {
+                    return {
                         "success": False,
-                        "error": "Not a file or content not available"
+                        "error": f"Path '{path}' is not a file or content is not available",
+                        "item_type": data.get("type", "unknown")
                     }
             else:
-                result = {
+                return {
                     "success": False,
-                    "error": f"GitHub API error: {response.status_code}"
+                    "error": f"File not found or inaccessible (HTTP {response.status_code})",
+                    "file_path": path,
+                    "status_code": response.status_code
                 }
             
-            return json.dumps(result, indent=2)
-            
         except Exception as e:
-            return json.dumps({"success": False, "error": str(e)}, indent=2)
+            return {
+                "success": False,
+                "error": f"Failed to fetch file content: {str(e)}",
+                "file_path": path
+            }
     
-    def scan_repository(self, owner: str, repo: str, extensions: str = ".py,.js,.ts,.php,.java") -> str:
+    def scan_repository(self, owner: str, repo: str, extensions: str = ".py,.js,.ts,.php,.java") -> dict:
         """Scan repository for code files"""
         try:
-            ext_list = [ext.strip() for ext in extensions.split(",")]
+            ext_list = [ext.strip() for ext in extensions.split(",") if ext.strip()]
             all_files = []
             self._scan_directory_sync(owner, repo, "", ext_list, all_files)
             
-            result = {
+            # Group files by extension for better organization
+            files_by_extension = {}
+            for file_info in all_files:
+                ext = file_info.get('name', '').split('.')[-1].lower() if '.' in file_info.get('name', '') else 'no_extension'
+                if ext not in files_by_extension:
+                    files_by_extension[ext] = []
+                files_by_extension[ext].append(file_info)
+            
+            return {
                 "success": True,
-                "data": {
+                "total_files_found": len(all_files),
+                "extensions_scanned": ext_list,
+                "files_by_extension": files_by_extension,
+                "all_files": all_files[:100],  # Limit to first 100 files
+                "scan_summary": {
                     "total_files": len(all_files),
-                    "files": all_files[:50]  # Limit to first 50 files for display
+                    "extensions_found": list(files_by_extension.keys()),
+                    "largest_file": max(all_files, key=lambda x: x.get('size', 0)) if all_files else None
                 }
             }
             
-            return json.dumps(result, indent=2)
-            
         except Exception as e:
-            return json.dumps({"success": False, "error": str(e)}, indent=2)
+            return {
+                "success": False,
+                "error": f"Failed to scan repository: {str(e)}",
+                "total_files_found": 0
+            }
     
     def _scan_directory_sync(self, owner: str, repo: str, path: str, extensions: List[str], all_files: List[Dict]):
         """Recursively scan directory for files"""
@@ -191,4 +225,4 @@ if __name__ == "__main__":
     print("📡 Server will provide GitHub repository access via MCP")
     print("🛠️ Available tools: repository info, file content, repository scanner")
     
-    demo.launch(share=True, mcp_server=True)
+    demo.launch(mcp_server=True)

requirements.txt

@@ -1,14 +1,8 @@
-# core
 gradio[oauth,mcp]==5.45.0
 fastapi==0.115.2
 uvicorn==0.24.0
-
-# MCP
 mcp==1.10.1
-
-# other deps
 smolagents>=0.1.0
 requests>=2.28.0
 python-dotenv>=1.0.0
-
 pydantic>=2.11,<2.12