Update app.py

app.py

@@ -8,7 +8,6 @@ import os
 from datetime import datetime
 from typing import List, Dict
 import requests
-from huggingface_hub import login  # Importing Hugging Face Spaces
 import gradio as gr
 import atexit
 import subprocess
@@ -23,35 +22,23 @@ warnings.filterwarnings('ignore', category=UserWarning)
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 
 def initialize_zero_gpu():
-    """Initialize Hugging Face ZeroGPU and handle Hugging Face Spaces"""
-    logger.info("Initializing Hugging Face ZeroGPU...")
+    """Initialize Hugging Face ZeroGPU"""
     try:
-        # Check if CUDA is available
         if torch.cuda.is_available():
-            # Initialize DeepSpeed with Zero Optimization
             deepspeed.init_distributed()
             print("ZeroGPU initialized successfully.")
         else:
             print("CUDA is not available. Using CPU instead.")
     except Exception as e:
         print(f"Error initializing Hugging Face ZeroGPU: {str(e)}")
-        logger.error(f"Error initializing Hugging Face ZeroGPU: {str(e)}")
-        
-    # Check for the missing autotune file
-    autotune_file = '/home/user/.triton/autotune'
-    if not os.path.exists(autotune_file):
-        logger.warning(f"Missing autotune file: {autotune_file}. This may affect performance.")
-        print(f"Warning: Missing autotune file: {autotune_file}. This may affect performance.")
 
 def initialize_environment():
     """Initialize application environment and configurations"""
-    # Create necessary directories
     directories = ['logs', 'resolutions', 'repos']
     for directory in directories:
         os.makedirs(directory, exist_ok=True)
 
-    # Configure logging
-    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',
@@ -61,7 +48,6 @@ def initialize_environment():
         ]
     )
 
-    # Set up global exception handler
     def handle_exception(exc_type, exc_value, exc_traceback):
         if issubclass(exc_type, KeyboardInterrupt):
             sys.__excepthook__(exc_type, exc_value, exc_traceback)
@@ -167,21 +153,21 @@ class GitHubBot:
                 f.write(f"# Resolution for Issue #{issue_number}\n\n{resolution}")
 
             # Clone the forked repo
-            subprocess.run(['git', 'clone', forked_repo, '/tmp/' + forked_repo.split('/')[-1]])
+            subprocess.run(['git', 'clone', forked_repo, '/tmp/' + forked_repo.split('/')[-1]], check=True)
 
             # Change to the cloned directory
             os.chdir('/tmp/' + forked_repo.split('/')[-1])
 
             # Assuming manual intervention now
-            input("Apply the fix manually and stage the changes (press ENTER)? ")
+            input(" Apply the fix manually and stage the changes (press ENTER)? ")
 
             # Commit and push the modifications
-            subprocess.run(['git', 'add', '.'])
-            subprocess.run(['git', 'commit', '-m', f"Resolved issue #{issue_number} ({urllib.parse.quote(resolution)})"])
-            subprocess.run(['git', 'push', 'origin', 'HEAD'])
+            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
-            webbrowser.open('https://github.com/' + forked_repo.split('/')[-1] + '/compare/master...' + username + ':' + forked_repo.split('/')[-1] + '_resolved_issue_' + str(issue_number))
+            webbrowser.open(f'https://github.com/{forked_repo.split("/")[-1]}/compare/master...{username}:{forked_repo.split("/")[-1]}_resolved_issue_{issue_number}')
 
             return f"Resolution saved: {resolution_file}"
 
@@ -420,4 +406,4 @@ if __name__ == "__main__":
         logger.error(f"Error launching application: {str(e)}")
         raise
     finally:
-        logger.info("Application shutdown")
+        logger.info("Application shutdown")