Update app.py

app.py

@@ -31,7 +31,10 @@ def fetch_github_file(github_url, ssh_private_key):
         ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
 
         # Load the private key
-        private_key = paramiko.RSAKey.from_private_key(io.StringIO(ssh_private_key))
+        try:
+            private_key = paramiko.RSAKey.from_private_key(io.StringIO(ssh_private_key.strip()))
+        except paramiko.ssh_exception.SSHException as e:
+            return f"Error: Invalid SSH key - {str(e)}"
 
         # Connect to GitHub using SSH
         ssh.connect('github.com', username='git', pkey=private_key)
@@ -93,6 +96,8 @@ def process_input(file, github_url, ssh_private_key, gemini_api_key):
     elif github_url and ssh_private_key:
         if not ssh_private_key.strip():
             return "Error: SSH Private Key is empty. Please provide a valid key."
+        if not ssh_private_key.strip().startswith("-----BEGIN"):
+            return "Error: Invalid SSH key format. Please ensure you've copied the entire key, including the BEGIN and END lines."
         try:
             file_content = fetch_github_file(github_url, ssh_private_key)
             if "error" in file_content:
@@ -114,7 +119,7 @@ iface = gr.Interface(
     inputs=[
         gr.File(label="Upload dependency file (e.g., requirements.txt, package.json, Gemfile)"),
         gr.Textbox(label="GitHub Repository URL (optional, SSH or HTTPS format)"),
-        gr.Textbox(label="SSH Private Key (required if using GitHub URL)", type="password"),
+        gr.Textarea(label="SSH Private Key (required if using GitHub URL)", lines=10),
         gr.Textbox(label="Gemini API Key", type="password"),
     ],
     outputs=gr.Textbox(label="License Information and Analysis"),