Update app.py

app.py

@@ -5,7 +5,7 @@ import logging
 import time
 import os
 from datetime import datetime
-from typing import List, Dict
+from typing import List, Dict, Any
 import requests
 import gradio as gr
 import atexit
@@ -38,7 +38,6 @@ def initialize_logger():
     return logging.getLogger(__name__)
 
 # Set up environment
-@spaces.GPU
 def initialize_environment(input_file, output_directory):
     directories = [LOGS_DIRECTORY, RESOLUTIONS_DIRECTORY, REPOS_DIRECTORY, input_file, output_directory]
     for directory in directories:
@@ -156,6 +155,32 @@ def handle_issue_selection(token, owner, repo, issue_number, resolution, forked_
     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:
+        response = requests.get(url)
+        response.raise_for_status()
+        info['status_code'] = response.status_code
+        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)}"
+    except Exception as e:
+        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');
@@ -219,7 +244,7 @@ custom_css = """
     }
 """
 
-def create_gradio_interface(input_file, output_directory):
+def create_gradio_interface():
     bot = GitHubBot()
 
     with gr.Blocks(css=custom_css, theme=gr.themes.Base()) as demo:
@@ -277,18 +302,28 @@ def create_gradio_interface(input_file, output_directory):
             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(choices=["Please provide all required fields"])
+                return gr.Dropdown.update(choices=["Please provide all required fields"])
 
             owner, repo = repo_link.split('/')[-2:]
             issues = bot.fetch_issues(token, owner, repo)
             if not issues:
-                return gr.Dropdown(choices=["No issues found or error occurred"])
+                return gr.Dropdown.update(choices=["No issues found or error occurred"])
 
-            return gr.Dropdown(
+            return gr.Dropdown.update(
                 choices=[f"#{issue['number']}: {issue['title']}" for issue in issues]
             )
 
@@ -303,6 +338,10 @@ def create_gradio_interface(input_file, output_directory):
             except Exception as e:
                 return f"Error: {str(e)}"
 
+        def extract_info_handler(url):
+            info = extract_info_from_url(url)
+            return info
+
         # Connect components
         fetch_button.click(
             fetch_issues_handler,
@@ -322,6 +361,12 @@ def create_gradio_interface(input_file, output_directory):
             outputs=[output_text]
         )
 
+        extract_button.click(
+            extract_info_handler,
+            inputs=[url_textbox],
+            outputs=[output_text]
+        )
+
     return demo
 
 # Cleanup function
@@ -347,12 +392,9 @@ if __name__ == "__main__":
     signal.signal(signal.SIGINT, signal_handler)
     signal.signal(signal.SIGTERM, signal_handler)
 
-    input_file = INPUT_DIRECTORY
-    output_directory = OUTPUT_DIRECTORY
-
     # Initialize logger
-    logger = initialize_environment(input_file, output_directory)
+    logger = initialize_environment(INPUT_DIRECTORY, OUTPUT_DIRECTORY)
 
     # Create Gradio interface
-    demo = create_gradio_interface(input_file, output_directory)
+    demo = create_gradio_interface()
     demo.launch()