Update app.py

app.py

@@ -10,12 +10,8 @@ import requests
 import gradio as gr
 import atexit
 import subprocess
-import webbrowser
-import urllib.parse
+from urllib.parse import urlparse
 import warnings
-import torch
-import spaces
-from accelerate import Accelerator
 
 # Constants
 INPUT_DIRECTORY = 'input'
@@ -37,12 +33,15 @@ 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)
-    return initialize_logger()
+
 
 # GitHub API handler
 class GitHubAPI:
@@ -98,6 +97,7 @@ class GitHubAPI:
             logger.error(f"Error fetching issues: {str(e)}")
             return []
 
+
 # GitHub Bot
 class GitHubBot:
     def __init__(self):
@@ -150,11 +150,13 @@ 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:
@@ -164,16 +166,6 @@ 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
 
-        if 'github.com' in url:
-            parts = url.split('/')
-            if len(parts) > 4:
-                owner = parts[3]
-                repo = parts[4]
-                issues = bot.fetch_issues(github_token, owner, repo)
-                info['issues'] = issues
-        elif 'huggingface.co' in url:
-            # Add Hugging Face specific handling if needed
-            pass
 
     except requests.HTTPError as e:
         info['error'] = f"HTTP error: {str(e)}"
@@ -181,37 +173,43 @@ 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');
-    /* Custom gradient background */
+
+    /* Futuristic Cyberpunk Style */
     body {
-        background: linear-gradient(to bottom right, #121212, #303030) !important;
+        background: linear-gradient(to bottom right, #1e1e1e, #303030) !important;
+        color: #f0f0f0; 
+        font-family: 'Roboto Mono', monospace;
     }
-    /* Card styling */
     .container {
         max-width: 800px;
         margin: 2rem auto;
         padding: 0 1rem;
     }
+    h1 {
+        color: #00c0ff;
+        text-shadow: 0 0 10px #00c0ff;  /* Glow effect */
+    }
     .card {
-        background: #2a2a2a;
+        background: rgba(20,20,20, 0.8);
         border-radius: 1rem;
-        box-shadow: 0 10px 15px -3px rgba(0, 0, 0, 0.1);
+        box-shadow: 0 10px 15px -3px rgba(0, 0, 0, 0.5); 
         padding: 2rem;
+        border: 2px solid #00c0ff;
     }
-    /* Form controls */
     .input, .select, .textarea {
-        background: #363636;
-        border: 1px solid #4a4a4a;
-        color: #ffffff;
+        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;
     }
-    /* Buttons */
     .button {
         width: 100%;
         padding: 0.75rem;
@@ -219,82 +217,111 @@ custom_css = """
         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: #570df8;
-        color: white;
+        background: #2c2c2c;
+        color: #f0f0f0;
+        border-color: #00c0ff;
+    }
+    .button-primary:hover {
+        background: #4c4c4c;
+        color: #f0f0f0;
+        border-color: #00ff00;
     }
     .button-success {
-        background: #36d399;
-        color: white;
+        background: #2c2c2c;
+        color: #f0f0f0;
+        border-color: #00c0ff;
     }
-    /* Labels */
-    .label {
-        color: #a3a3a3;
-        margin-bottom: 0.5rem;
-        display: block;
+    .button-success:hover {
+        background: #4c4c4c;
+        color: #f0f0f0;
+        border-color: #00ff00;
     }
-    /* Output area */
     .output-area {
         margin-top: 2rem;
         padding: 1rem;
-        background: #363636;
+        background: rgba(20, 20, 20, 0.9);
         border-radius: 0.5rem;
-        color: #ffffff;
+        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.HTML("""
-            <div class="container">
-                <h1 class="text-4xl font-bold text-white mb-8 text-center">GitHub Issue Manager</h1>
-                <div class="card">
-        """)
-
-        with gr.Group():
-            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.Group():
-            fetch_button = gr.Button(
-                "Fetch Issues",
-                elem_classes="button button-primary"
-            )
-            issue_dropdown = gr.Dropdown(
-                choices=[],
-                label="Select Issue",
-                interactive=True,
-                elem_classes="select select-primary"
-            )
 
-        with gr.Group():
-            resolution_text = gr.Textbox(
-                label="Resolution",
-                placeholder="Enter your resolution here...",
-                lines=5,
-                elem_classes="textarea textarea-primary"
-            )
-            forked_repo = gr.Textbox(
-                label="Forked Repository Path",
-                placeholder="Enter the path of the forked repository",
-                elem_classes="input input-bordered input-primary"
-            )
-            resolve_button = gr.Button(
-                "Resolve Issue",
-                elem_classes="button button-success"
-            )
+        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",
@@ -302,24 +329,26 @@ def create_gradio_interface():
             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"
         )
 
-        gr.HTML("</div></div>")  # Close card and container divs
-
         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"])
 
@@ -340,9 +369,8 @@ def create_gradio_interface():
 
         def extract_info_handler(url):
             info = extract_info_from_url(url)
-            return info
+            return str(info)
 
-        # Connect components
         fetch_button.click(
             fetch_issues_handler,
             inputs=[github_token, repo_link],
@@ -351,13 +379,7 @@ def create_gradio_interface():
 
         resolve_button.click(
             resolve_issue_handler,
-            inputs=[
-                github_token,
-                repo_link,
-                issue_dropdown,
-                resolution_text,
-                forked_repo
-            ],
+            inputs=[github_token, repo_link, issue_dropdown, resolution_text, forked_repo],
             outputs=[output_text]
         )
 
@@ -367,8 +389,11 @@ def create_gradio_interface():
             outputs=[output_text]
         )
 
+
     return demo
 
+
+
 # Cleanup function
 def cleanup():
     try:
@@ -380,12 +405,14 @@ 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)
@@ -393,7 +420,7 @@ if __name__ == "__main__":
     signal.signal(signal.SIGTERM, signal_handler)
 
     # Initialize logger
-    logger = initialize_environment(INPUT_DIRECTORY, OUTPUT_DIRECTORY)
+    initialize_environment(INPUT_DIRECTORY, OUTPUT_DIRECTORY)
 
     # Create Gradio interface
     demo = create_gradio_interface()