Upload folder using huggingface_hub

evaluate_atomic_matching.py

@@ -11,6 +11,7 @@ import getpass
 import shutil
 import tempfile
 import subprocess
+import stat
 
 # Initialize instructor client
 client = instructor.from_litellm(completion)
@@ -202,6 +203,62 @@ def process_questions(llm_model: str, openai_api_Key: str, deploy_key: str, metr
         error_message = f"Error pushing to repository: {str(e)}\nResults were saved locally to: {filepath}"
         return error_message, filepath  # Return filepath even on git error
 
+def setup_ssh_key(ssh_key_content):
+    try:
+        validate_ssh_key(ssh_key_content)
+    except ValueError as e:
+        raise ValueError(f"Invalid SSH key: {str(e)}")
+    
+    # Create .ssh directory if it doesn't exist
+    ssh_dir = os.path.expanduser('~/.ssh')
+    os.makedirs(ssh_dir, exist_ok=True)
+    
+    # Save the key to a file
+    key_path = os.path.join(ssh_dir, 'github_deploy_key')
+    with open(key_path, 'w') as f:
+        f.write(ssh_key_content)
+    
+    # Set correct permissions (600)
+    os.chmod(key_path, stat.S_IRUSR | stat.S_IWUSR)
+    
+    # Configure ssh to use this key for github.com
+    config_path = os.path.join(ssh_dir, 'config')
+    config_content = f"""Host github.com
+    HostName github.com
+    IdentityFile {key_path}
+    User git
+    """
+    
+    with open(config_path, 'w') as f:
+        f.write(config_content)
+    
+    # Set correct permissions for .ssh directory
+    os.chmod(ssh_dir, stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR)
+    
+    return key_path
+
+def handle_ssh_key(ssh_key):
+    try:
+        key_path = setup_ssh_key(ssh_key)
+        # Test the SSH connection
+        result = subprocess.run(['ssh', '-T', 'git@github.com'], 
+                              capture_output=True, 
+                              text=True)
+        return f"SSH key setup complete. Key saved to {key_path}"
+    except Exception as e:
+        return f"Error setting up SSH key: {str(e)}"
+
+def validate_ssh_key(key_content):
+    # Check if it looks like a private key
+    if not key_content.startswith('-----BEGIN'):
+        raise ValueError("Invalid key format")
+    
+    # Check if it's complete
+    if not key_content.strip().endswith('-----END'):
+        raise ValueError("Incomplete key")
+    
+    return True
+
 # Create Gradio interface
 with gr.Blocks(css="footer {visibility: hidden}") as iface:
     gr.Markdown("# Question Evaluation Tool")