Update app.py

app.py

@@ -12,23 +12,21 @@ import atexit
 import subprocess
 import webbrowser
 import urllib.parse
-import http.client
-import spaces
+import warnings
 import torch
 from accelerate import Accelerator
 
-import warnings
 warnings.filterwarnings('ignore', category=UserWarning, module='__main__')
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 
 @spaces.GPU
 def initialize_environment(input_file, output_directory):
     """Initialize application environment and configurations"""
-    directories = ['logs', 'resolutions', 'repos']
+    directories = ['logs', 'resolutions', 'repos', input_file, output_directory]
     for directory in directories:
         os.makedirs(directory, exist_ok=True)
 
-    log_file = f"logs/github_bot_{datetime.now(). strftime('%Y%m%d_%H%M%S')}.log"
+    log_file = f"logs/github_bot_{datetime.now().strftime('%Y%m%d_%H%M%S')}.log"
     logging.basicConfig(
         level=logging.INFO,
         format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
@@ -47,7 +45,11 @@ def initialize_environment(input_file, output_directory):
     sys.excepthook = handle_exception
     return logging.getLogger(__name__)
 
-logger = initialize_environment()
+# Example user inputs for input and output directories
+input_directory = 'input'
+output_directory = 'output'
+
+logger = initialize_environment(input_directory, output_directory)
 
 class GitHubAPI:
     """GitHub API handler with rate limiting and error handling"""
@@ -116,7 +118,7 @@ class GitHubBot:
     def __init__(self):
         self.github_api = None
 
-    def initialize_api(self, token: str ):
+    def initialize_api(self, token: str):
         """Initialize GitHub API with token"""
         self.github_api = GitHubAPI(token)