Spaces:

acecalisto3
/

GitBot

Runtime error

App Files Files Community

acecalisto3 commited on Dec 15, 2024

Commit

e101753

verified ·

1 Parent(s): ef39663

Update app.py

Browse files

Files changed (1) hide show

app.py +271 -231

app.py CHANGED Viewed

@@ -10,8 +10,9 @@ import requests
 import gradio as gr
 import atexit
 import subprocess
-from urllib.parse import urlparse
 import warnings
 # Constants
 INPUT_DIRECTORY = 'input'
@@ -20,6 +21,12 @@ LOGS_DIRECTORY = 'logs'
 RESOLUTIONS_DIRECTORY = 'resolutions'
 REPOS_DIRECTORY = 'repos'
 # Set up logging
 def initialize_logger():
     log_file = f"{LOGS_DIRECTORY}/github_bot_{datetime.now().strftime('%Y%m%d_%H%M%S')}.log"
@@ -33,16 +40,10 @@ def initialize_logger():
     )
     return logging.getLogger(__name__)
 logger = initialize_logger()  # Initialize logger globally
-# Set up environment
-def initialize_environment(input_file, output_directory):
-    directories = [LOGS_DIRECTORY, RESOLUTIONS_DIRECTORY, REPOS_DIRECTORY, input_file, output_directory]
-    for directory in directories:
-        os.makedirs(directory, exist_ok=True)
 # GitHub API handler
 class GitHubAPI:
     def __init__(self, token: str):
@@ -97,7 +98,6 @@ class GitHubAPI:
             logger.error(f"Error fetching issues: {str(e)}")
             return []
 # GitHub Bot
 class GitHubBot:
     def __init__(self):
@@ -137,7 +137,7 @@ class GitHubBot:
             # Commit and push the modifications
             subprocess.run(['git', 'add', '.'], check=True)
-            subprocess.run(['git', 'commit', '-m', f"Resolved issue #{issue_number} ({urllib.parse.quote(resolution)})"], check=True)
             subprocess.run(['git', 'push', 'origin', 'HEAD'], check=True)
             # Open Pull Request page
@@ -150,13 +150,11 @@ class GitHubBot:
             logger.error(error_msg)
             return error_msg
 def handle_issue_selection(token, owner, repo, issue_number, resolution, forked_repo):
     bot = GitHubBot()
     result = bot.resolve_issue(token, owner, repo, issue_number, resolution, forked_repo)
     return result
 def extract_info_from_url(url: str) -> Dict[str, Any]:
     info = {}
     try:
@@ -166,6 +164,17 @@ def extract_info_from_url(url: str) -> Dict[str, Any]:
         info['headers'] = dict(response.headers)
         info['content'] = response.text[:500]  # Limit content to first 500 characters for brevity
     except requests.HTTPError as e:
         info['error'] = f"HTTP error: {str(e)}"
@@ -173,226 +182,262 @@ def extract_info_from_url(url: str) -> Dict[str, Any]:
         info['error'] = f"Error: {str(e)}"
     return info
-custom_css = """
-    /* DaisyUI and Tailwind are loaded from CDN */
-    @import url('https://cdn.jsdelivr.net/npm/[email protected]/dist/full.css');
-    @import url('https://cdn.tailwindcss.com');
-    /* Futuristic Cyberpunk Style */
-    body {
-        background: linear-gradient(to bottom right, #1e1e1e, #303030) !important;
-        color: #f0f0f0;
-        font-family: 'Roboto Mono', monospace;
-    }
-    .container {
-        max-width: 800px;
-        margin: 2rem auto;
-        padding: 0 1rem;
-    }
-    h1 {
-        color: #00c0ff;
-        text-shadow: 0 0 10px #00c0ff;  /* Glow effect */
-    }
-    .card {
-        background: rgba(20,20,20, 0.8);
-        border-radius: 1rem;
-        box-shadow: 0 10px 15px -3px rgba(0, 0, 0, 0.5);
-        padding: 2rem;
-        border: 2px solid #00c0ff;
-    }
-    .input, .select, .textarea {
-        background: rgba(20,20,20, 0.9);
-        border: 1px solid #00c0ff;
-        color: #f0f0f0;
-        border-radius: 0.5rem;
-        padding: 0.75rem;
-        width: 100%;
-        margin-bottom: 1rem;
-    }
-    .button {
-        width: 100%;
-        padding: 0.75rem;
-        border-radius: 0.5rem;
-        font-weight: bold;
-        cursor: pointer;
-        transition: all 0.2s;
-        background: rgba(20, 20, 20, 0.8);
-        border: 2px solid #00c0ff;
-        color: #f0f0f0;
-    }
-    .button:hover {
-        background: rgba(10,10,10, 0.9);
-        border-color: #00ff00; /* Green on hover */
-    }
-    .button-primary {
-        background: #2c2c2c;
-        color: #f0f0f0;
-        border-color: #00c0ff;
-    }
-    .button-primary:hover {
-        background: #4c4c4c;
-        color: #f0f0f0;
-        border-color: #00ff00;
-    }
-    .button-success {
-        background: #2c2c2c;
-        color: #f0f0f0;
-        border-color: #00c0ff;
-    }
-    .button-success:hover {
-        background: #4c4c4c;
-        color: #f0f0f0;
-        border-color: #00ff00;
-    }
-    .output-area {
-        margin-top: 2rem;
-        padding: 1rem;
-        background: rgba(20, 20, 20, 0.9);
-        border-radius: 0.5rem;
-        color: #f0f0f0;
-        border: 2px solid #00c0ff;
-        overflow-y: auto;
-        max-height: 300px;
-    }
-    .label {
-        color: #a3a3a3;
-        margin-bottom: 0.5rem;
-        display: block;
-    }
-    /* Add more styling as needed to refine the look */
-"""
-def create_gradio_interface():
-    bot = GitHubBot()
-    with gr.Blocks(css=custom_css, theme=gr.themes.Base()) as demo:
-        gr.Markdown("<h1 style='color: #00c0ff; text-shadow: 0 0 10px #00c0ff;'>GitHub Issue Manager</h1>")
-        with gr.Row():
-            with gr.Column(scale=1):
-                github_token = gr.Textbox(
-                    label="GitHub Token",
-                    placeholder="Enter your GitHub token",
-                    type="password",
-                    elem_classes="input input-bordered input-primary"
-                )
-                repo_link = gr.Textbox(
-                    label="Repository Link",
-                    placeholder="Enter repository link (e.g., https://github.com/owner/repo)",
-                    elem_classes="input input-bordered input-primary"
-                )
-            with gr.Column(scale=1):
-                fetch_button = gr.Button(
-                    "Fetch Issues",
-                    elem_classes="button button-primary"
-                )
-        with gr.Row():
-            with gr.Column(scale=1):
-                issue_dropdown = gr.Dropdown(
-                    choices=[],
-                    label="Select Issue",
-                    interactive=True,
-                    elem_classes="select select-primary"
-                )
-            with gr.Column(scale=1):
-                resolution_text = gr.Textbox(
-                    label="Resolution",
-                    placeholder="Enter your resolution here...",
-                    lines=5,
-                    elem_classes="textarea textarea-primary"
-                )
-        with gr.Row():
-            with gr.Column(scale=1):
-                forked_repo = gr.Textbox(
-                    label="Forked Repository Path",
-                    placeholder="Enter the path of the forked repository",
-                    elem_classes="input input-bordered input-primary"
-                )
-            with gr.Column(scale=1):
-                resolve_button = gr.Button(
-                    "Resolve Issue",
-                    elem_classes="button button-success"
-                )
-        output_text = gr.Textbox(
-            label="Output",
-            interactive=False,
-            elem_classes="output-area"
-        )
-        url_textbox = gr.Textbox(
-            label="URL",
-            placeholder="Enter a URL to extract information",
-            elem_classes="input input-bordered input-primary"
-        )
-        extract_button = gr.Button(
-            "Extract Info",
-            elem_classes="button button-primary"
-        )
-        def fetch_issues_handler(token, repo_link):
-            if not all([token, repo_link]):
-                return gr.Dropdown.update(choices=["Please provide all required fields"])
-            owner, repo = repo_link.split('/')[-2:]
-            bot.initialize_api(token)
-            issues = bot.fetch_issues(token, owner, repo)
-            if not issues:
-                return gr.Dropdown.update(choices=["No issues found or error occurred"])
-            return gr.Dropdown.update(
-                choices=[f"#{issue['number']}: {issue['title']}" for issue in issues]
-            )
-        def resolve_issue_handler(token, repo_link, selected_issue, resolution, forked_repo):
-            if not selected_issue:
-                return "Please select an issue first"
-            try:
-                owner, repo = repo_link.split('/')[-2:]
-                issue_number = int(selected_issue.split(':')[0][1:])
-                return handle_issue_selection(token, owner, repo, issue_number, resolution, forked_repo)
-            except Exception as e:
-                return f"Error: {str(e)}"
-        def extract_info_handler(url):
-            info = extract_info_from_url(url)
-            return str(info)
-        fetch_button.click(
-            fetch_issues_handler,
-            inputs=[github_token, repo_link],
-            outputs=[issue_dropdown]
-        )
-        resolve_button.click(
-            resolve_issue_handler,
-            inputs=[github_token, repo_link, issue_dropdown, resolution_text, forked_repo],
-            outputs=[output_text]
-        )
-        extract_button.click(
-            extract_info_handler,
-            inputs=[url_textbox],
-            outputs=[output_text]
-        )
-    return demo
 # Cleanup function
 def cleanup():
@@ -405,23 +450,18 @@ def cleanup():
     except Exception as e:
         print(f"Error during cleanup: {str(e)}")
 # Signal handler
 def signal_handler(signum, frame):
     logger.info(f"Received signal {signum}")
     cleanup()
     sys.exit(0)
 if __name__ == "__main__":
     # Register cleanup handlers
     atexit.register(cleanup)
     signal.signal(signal.SIGINT, signal_handler)
     signal.signal(signal.SIGTERM, signal_handler)
-    # Initialize logger
-    initialize_environment(INPUT_DIRECTORY, OUTPUT_DIRECTORY)
     # Create Gradio interface
     demo = create_gradio_interface()
     demo.launch()

 import gradio as gr
 import atexit
 import subprocess
+from urllib.parse import urlparse, quote
 import warnings
+import webbrowser
 # Constants
 INPUT_DIRECTORY = 'input'
 RESOLUTIONS_DIRECTORY = 'resolutions'
 REPOS_DIRECTORY = 'repos'
+# Set up environment
+def initialize_environment(input_file, output_directory):
+    directories = [LOGS_DIRECTORY, RESOLUTIONS_DIRECTORY, REPOS_DIRECTORY, input_file, output_directory]
+    for directory in directories:
+        os.makedirs(directory, exist_ok=True)
 # Set up logging
 def initialize_logger():
     log_file = f"{LOGS_DIRECTORY}/github_bot_{datetime.now().strftime('%Y%m%d_%H%M%S')}.log"
     )
     return logging.getLogger(__name__)
+# Initialize environment and logger
+initialize_environment(INPUT_DIRECTORY, OUTPUT_DIRECTORY)
 logger = initialize_logger()  # Initialize logger globally
 # GitHub API handler
 class GitHubAPI:
     def __init__(self, token: str):
             logger.error(f"Error fetching issues: {str(e)}")
             return []
 # GitHub Bot
 class GitHubBot:
     def __init__(self):
             # Commit and push the modifications
             subprocess.run(['git', 'add', '.'], check=True)
+            subprocess.run(['git', 'commit', '-m', f"Resolved issue #{issue_number} ({quote(resolution)})"], check=True)
             subprocess.run(['git', 'push', 'origin', 'HEAD'], check=True)
             # Open Pull Request page
             logger.error(error_msg)
             return error_msg
 def handle_issue_selection(token, owner, repo, issue_number, resolution, forked_repo):
     bot = GitHubBot()
     result = bot.resolve_issue(token, owner, repo, issue_number, resolution, forked_repo)
     return result
 def extract_info_from_url(url: str) -> Dict[str, Any]:
     info = {}
     try:
         info['headers'] = dict(response.headers)
         info['content'] = response.text[:500]  # Limit content to first 500 characters for brevity
+        parsed_url = urlparse(url)
+        if 'github.com' in parsed_url.netloc:
+            parts = parsed_url.path.split('/')
+            if len(parts) > 2:
+                owner = parts[1]
+                repo = parts[2]
+                issues = bot.fetch_issues(github_token, owner, repo)
+                info['issues'] = issues
+        elif 'huggingface.co' in parsed_url.netloc:
+            # Add Hugging Face specific handling if needed
+            pass
     except requests.HTTPError as e:
         info['error'] = f"HTTP error: {str(e)}"
         info['error'] = f"Error: {str(e)}"
     return info
+# Initialize GitHubBot globally
+bot = GitHubBot()
+# HTML and CSS integration
+custom_html = """
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>GitHub Issue Manager</title>
+    <!-- Tailwind CSS -->
+    <link href="https://cdn.jsdelivr.net/npm/[email protected]/dist/tailwind.min.css" rel="stylesheet">
+    <!-- DaisyUI -->
+    <link href="https://cdn.jsdelivr.net/npm/[email protected]/dist/full.css" rel="stylesheet" type="text/css" />
+    <style>
+        body {
+            background: linear-gradient(to bottom right, #121212, #303030) !important;
+            color: #e0e0e0;
+            font-family: 'Roboto', sans-serif;
+            overflow-x: hidden; /* Prevent horizontal scrollbar */
+        }
+        .card {
+            border-radius: 1.5rem;
+            box-shadow: 0 15px 25px rgba(0, 0, 0, 0.2);
+            margin-bottom: 20px;
+        }
+        .input, .select, .textarea {
+            border: 2px solid #4a4a4a;
+            border-radius: 0.75rem;
+            padding-inline: 1.5rem;
+            padding-block: 0.75rem;
+        }
+        h1, h2, h3 {
+            color: #e0e0e0;
+        }
+        .output-area {
+            padding: 1.5rem;
+            border-radius: 0.75rem;
+        }
+        .btn {
+            border-radius: 0.75rem;
+            padding-block: 0.75rem;
+            padding-inline: 1.5rem;
+        }
+        .btn-primary {
+            background-color: #7928CA;
+            color: white;
+            border: 2px solid #7928CA;
+        }
+        .btn-success {
+            background-color: #22c55e;
+            color: white;
+            border: 2px solid #22c55e;
+        }
+    </style>
+</head>
+<body class="bg-gray-900">
+<div class="container mx-auto p-4" >
+    <h1 class="text-4xl md:text-5xl font-extrabold text-center mb-8">GitHub Issue Manager</h1>
+    <!-- GitHub Token & Repo URL -->
+    <div class="card bg-gray-800 p-8">
+        <div class="form-control">
+            <label class="label">
+                <span class="label-text">GitHub Token</span>
+            </label>
+            <input type="password" placeholder="Enter your GitHub Personal Access Token" class="input input-bordered input-primary" id="github-token">
+        </div>
+        <div class="form-control">
+            <label class="label">
+                <span class="label-text">Repository URL</span>
+            </label>
+            <input type="text" placeholder="Enter the full GitHub repository URL" class="input input-bordered input-primary" id="repo-url">
+        </div>
+    </div>
+    <!-- Fetch & Resolve Section -->
+    <div class="card bg-gray-800 p-8">
+        <div class="flex justify-between gap-4">
+            <button class="btn btn-primary" id="fetch-issues">Fetch Issues</button>
+            <select class="select select-primary w-full max-w-xs" id="issue-dropdown" >
+                <option disabled selected>Select Issue</option>
+            </select>
+        </div>
+        <div class="form-control mt-4">
+            <label class="label">
+                <span class="label-text">Resolution</span>
+            </label>
+            <textarea class="textarea textarea-primary" placeholder="Enter the resolution details" id="resolution-textarea"></textarea>
+        </div>
+        <div class="form-control mt-4">
+            <label class="label">
+                <span class="label-text">Forked Repository URL</span>
+            </label>
+            <input type="text" placeholder="URL to your forked repository (Optional)" class="input input-bordered input-primary" id="forked-repo-url">
+        </div>
+        <div class="form-control mt-4">
+            <button class="btn btn-success" id="resolve-issue">Resolve Issue</button>
+        </div>
+    </div>
+    <!-- Output Area -->
+    <div class="card bg-gray-800 p-8 mt-4" >
+        <label class="label">
+            <span class="label-text">Output</span>
+        </label>
+        <textarea class="textarea textarea-primary h-48" id="output-textarea" readonly></textarea>
+    </div>
+    <!-- URL to Extract -->
+    <div class="card bg-gray-800 p-8 mt-4">
+        <div class="form-control">
+            <label class="label">
+                <span class="label-text">URL</span>
+            </label>
+            <input type="text" placeholder="Enter a URL to extract information" class="input input-bordered input-primary" id="url-textbox">
+        </div>
+        <div class="form-control">
+            <button class="btn btn-primary" id="extract-info">Extract Info</button>
+        </div>
+    </div>
+</div>
+<script>
+    const githubTokenInput = document.getElementById('github-token');
+    const repoUrlInput = document.getElementById('repo-url');
+    const fetchIssuesButton = document.getElementById('fetch-issues');
+    const issueDropdown = document.getElementById('issue-dropdown');
+    const resolutionTextarea = document.getElementById('resolution-textarea');
+    const forkedRepoUrlInput = document.getElementById('forked-repo-url');
+    const resolveIssueButton = document.getElementById('resolve-issue');
+    const outputTextarea = document.getElementById('output-textarea');
+    const urlTextbox = document.getElementById('url-textbox');
+    const extractInfoButton = document.getElementById('extract-info');
+    fetchIssuesButton.addEventListener('click', async () => {
+      const token = githubTokenInput.value;
+      const repoUrl = repoUrlInput.value;
+      if (!token || !repoUrl) {
+        outputTextarea.value = "Please provide both GitHub Token and Repository URL.";
+        return;
+      }
+      try {
+        const response = await fetch('/fetch_issues', {
+          method: 'POST',
+          headers: {
+            'Content-Type': 'application/json',
+          },
+          body: JSON.stringify({ token: token, repoUrl: repoUrl }),
+        });
+        if (!response.ok) {
+          throw new Error(`HTTP error! status: ${response.status}`);
+        }
+        const data = await response.json();
+        if (data.error) {
+            outputTextarea.value = data.error;
+        } else {
+            issueDropdown.innerHTML = '';
+            data.issues.forEach(issue => {
+                const option = document.createElement('option');
+                option.value = issue.number;
+                option.text = `${issue.number}: ${issue.title}`;
+                issueDropdown.add(option);
+            });
+        }
+      } catch (error) {
+        outputTextarea.value = `Error fetching issues: ${error.message}`;
+      }
+    });
+    resolveIssueButton.addEventListener('click', async () => {
+      const token = githubTokenInput.value;
+      const repoUrl = repoUrlInput.value;
+      const issueNumber = issueDropdown.value;
+      const resolution = resolutionTextarea.value;
+      const forkedRepoUrl = forkedRepoUrlInput.value;
+      if (!token || !repoUrl || !issueNumber || !resolution) {
+        outputTextarea.value ="Please provide all required fields.";
+        return;
+      }
+      try {
+        const response = await fetch('/resolve_issue', {
+          method: 'POST',
+          headers: {
+            'Content-Type': 'application/json',
+          },
+          body: JSON.stringify({ token: token, repoUrl: repoUrl, issueNumber: issueNumber, resolution: resolution, forkedRepoUrl: forkedRepoUrl }),
+        });
+        if (!response.ok) {
+          throw new Error(`HTTP error! status: ${response.status}`);
+        }
+        const data = await response.json();
+        if (data.error) {
+            outputTextarea.value = data.error;
+        } else {
+            outputTextarea.value = data.result;
+        }
+      } catch (error) {
+        outputTextarea.value = `Error resolving issue: ${error.message}`;
+      }
+    });
+    extractInfoButton.addEventListener('click', async () => {
+      const url = urlTextbox.value;
+      if (!url) {
+        outputTextarea.value = "Please provide a URL.";
+        return;
+      }
+      try {
+        const response = await fetch('/extract_info', {
+          method: 'POST',
+          headers: {
+            'Content-Type': 'application/json',
+          },
+          body: JSON.stringify({ url: url }),
+        });
+        if (!response.ok) {
+          throw new Error(`HTTP error! status: ${response.status}`);
+        }
+        const data = await response.json();
+        if (data.error) {
+            outputTextarea.value = data.error;
+        } else {
+            outputTextarea.value = JSON.stringify(data, null, 2);
+        }
+      } catch (error) {
+        outputTextarea.value = `Error extracting info: ${error.message}`;
+      }
+    });
+</script>
+</body>
+</html>
+"""
+def create_gradio_interface():
+    with gr.Blocks(css=None, theme=None) as demo:
+        gr.HTML(custom_html)
+    return demo
 # Cleanup function
 def cleanup():
     except Exception as e:
         print(f"Error during cleanup: {str(e)}")
 # Signal handler
 def signal_handler(signum, frame):
     logger.info(f"Received signal {signum}")
     cleanup()
     sys.exit(0)
 if __name__ == "__main__":
     # Register cleanup handlers
     atexit.register(cleanup)
     signal.signal(signal.SIGINT, signal_handler)
     signal.signal(signal.SIGTERM, signal_handler)
     # Create Gradio interface
     demo = create_gradio_interface()
     demo.launch()