Spaces:

broadfield-dev
/

repo_to_md

Running

App Files Files Community

broadfield-dev commited on Feb 25

Commit

540d342

verified ·

1 Parent(s): 6321348

Update app.py

Browse files

Files changed (1) hide show

app.py +57 -18

app.py CHANGED Viewed

@@ -5,6 +5,7 @@ import markdown
 from bs4 import BeautifulSoup
 import os
 import mimetypes
 app = Flask(__name__)
@@ -33,15 +34,23 @@ def process_file_content(file_info, owner, repo):
         file_data = file_response.json()
         if 'content' in file_data:
-            # Guess the file type
             file_extension = file_path.split('.')[-1] if '.' in file_path else ''
             mime_type, _ = mimetypes.guess_type(file_path)
-            is_text = mime_type and mime_type.startswith('text') or file_extension in ['py', 'md', 'txt', 'js', 'html', 'css']
             if is_text:
                 try:
                     decoded_content = base64.b64decode(file_data['content']).decode('utf-8')
-                    content = f"### File: {file_path}\n```{(file_extension if file_extension else 'text')}\n{decoded_content}\n```\n\n"
                 except UnicodeDecodeError:
                     content = f"### File: {file_path}\n[Text content could not be decoded - possibly corrupted or non-UTF-8 text]\n\n"
             else:
@@ -53,7 +62,7 @@ def create_markdown_document(repo_url):
     """Create markdown document from repo contents"""
     owner, repo, contents = get_repo_contents(repo_url)
-    if isinstance(contents, str):  # Error case
         return f"Error: {contents}"
     markdown_content = f"# Repository: {owner}/{repo}\n\n"
@@ -119,6 +128,20 @@ html_template = """
             padding: 20px;
             background-color: #f9f9f9;
         }
     </style>
 </head>
 <body>
@@ -127,6 +150,7 @@ html_template = """
         <p>Enter a GitHub repository URL (e.g., https://github.com/username/repository)</p>
         <input type="text" id="repoUrl" style="width: 100%; padding: 8px;" placeholder="Enter GitHub repository URL">
         <button onclick="processRepo()">Convert to Markdown</button>
         <h2>Markdown Output:</h2>
         <textarea id="markdownOutput" readonly></textarea>
@@ -138,23 +162,38 @@ html_template = """
     <script>
         async function processRepo() {
             const repoUrl = document.getElementById('repoUrl').value;
-            const response = await fetch('/process', {
-                method: 'POST',
-                headers: {
-                    'Content-Type': 'application/json',
-                },
-                body: JSON.stringify({ repo_url: repoUrl })
-            });
-            const data = await response.json();
-            if (data.error) {
-                alert(data.error);
-                return;
             }
-            document.getElementById('markdownOutput').value = data.markdown;
-            document.getElementById('output').innerHTML = data.html;
         }
     </script>
 </body>

 from bs4 import BeautifulSoup
 import os
 import mimetypes
+import json
 app = Flask(__name__)
         file_data = file_response.json()
         if 'content' in file_data:
             file_extension = file_path.split('.')[-1] if '.' in file_path else ''
             mime_type, _ = mimetypes.guess_type(file_path)
+            is_text = (mime_type and mime_type.startswith('text')) or file_extension in ['py', 'md', 'txt', 'js', 'html', 'css', 'json']
             if is_text:
                 try:
                     decoded_content = base64.b64decode(file_data['content']).decode('utf-8')
+                    # Special handling for JSON files
+                    if file_extension == 'json':
+                        try:
+                            json_data = json.loads(decoded_content)
+                            formatted_json = json.dumps(json_data, indent=2)
+                            content = f"### File: {file_path}\n```json\n{formatted_json}\n```\n\n"
+                        except json.JSONDecodeError:
+                            content = f"### File: {file_path}\n```json\n{decoded_content}\n```\n[Note: Invalid JSON format]\n\n"
+                    else:
+                        content = f"### File: {file_path}\n```{(file_extension if file_extension else 'text')}\n{decoded_content}\n```\n\n"
                 except UnicodeDecodeError:
                     content = f"### File: {file_path}\n[Text content could not be decoded - possibly corrupted or non-UTF-8 text]\n\n"
             else:
     """Create markdown document from repo contents"""
     owner, repo, contents = get_repo_contents(repo_url)
+    if isinstance(contents, str):
         return f"Error: {contents}"
     markdown_content = f"# Repository: {owner}/{repo}\n\n"
             padding: 20px;
             background-color: #f9f9f9;
         }
+        .spinner {
+            display: none;
+            border: 4px solid #f3f3f3;
+            border-top: 4px solid #3498db;
+            border-radius: 50%;
+            width: 30px;
+            height: 30px;
+            animation: spin 1s linear infinite;
+            margin: 20px auto;
+        }
+        @keyframes spin {
+            0% { transform: rotate(0deg); }
+            100% { transform: rotate(360deg); }
+        }
     </style>
 </head>
 <body>
         <p>Enter a GitHub repository URL (e.g., https://github.com/username/repository)</p>
         <input type="text" id="repoUrl" style="width: 100%; padding: 8px;" placeholder="Enter GitHub repository URL">
         <button onclick="processRepo()">Convert to Markdown</button>
+        <div id="spinner" class="spinner"></div>
         <h2>Markdown Output:</h2>
         <textarea id="markdownOutput" readonly></textarea>
     <script>
         async function processRepo() {
             const repoUrl = document.getElementById('repoUrl').value;
+            const spinner = document.getElementById('spinner');
+            const button = document.querySelector('button');
+            // Show spinner, disable button
+            spinner.style.display = 'block';
+            button.disabled = true;
+            try {
+                const response = await fetch('/process', {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json',
+                    },
+                    body: JSON.stringify({ repo_url: repoUrl })
+                });
+                const data = await response.json();
+                if (data.error) {
+                    alert(data.error);
+                    return;
+                }
+                document.getElementById('markdownOutput').value = data.markdown;
+                document.getElementById('output').innerHTML = data.html;
+            } catch (error) {
+                alert('An error occurred: ' + error.message);
+            } finally {
+                // Hide spinner, enable button
+                spinner.style.display = 'none';
+                button.disabled = false;
             }
         }
     </script>
 </body>