Upload folder using huggingface_hub

app.py

@@ -1,9 +1,14 @@
-from flask import Flask, request, send_file
+from flask import Flask, request, send_file, redirect
 from flask_cors import CORS
-import subprocess, os, uuid
+import subprocess
+import os
+import uuid
 
 app = Flask(__name__)
-CORS(app) # This allows your GitHub site to talk to your Hugging Face engine
+CORS(app) 
+
+BASE_URL = "https://gabefromsimplebit-simplebit-signer.hf.space"
+
 @app.route('/')
 def home():
     return "SimpleBit Engine is Online!"
@@ -13,25 +18,93 @@ def sign():
     job_id = str(uuid.uuid4())[:8]
     os.makedirs(job_id, exist_ok=True)
     
+    # Input names from your HTML
     ipa = request.files['ipa']
     p12 = request.files['p12']
     prov = request.files['provision']
     password = request.form.get('password', '1234')
 
-    ipa_path = f"{job_id}/app.ipa"
+    ipa_path = os.path.join(job_id, "app.ipa")
+    p12_path = os.path.join(job_id, "cert.p12")
+    prov_path = os.path.join(job_id, "prov.mobileprovision")
+    out_path = os.path.join(job_id, "signed.ipa")
+
     ipa.save(ipa_path)
-    p12.save(f"{job_id}/cert.p12")
-    prov.save(f"{job_id}/prov.mobileprovision")
+    p12.save(p12_path)
+    prov.save(prov_path)
 
-    output_path = f"{job_id}/signed.ipa"
-    cmd = f"zsign -k {job_id}/cert.p12 -p {password} -m {job_id}/prov.mobileprovision -o {output_path} {ipa_path}"
+    # Optimization: -z 0 for instant repacking
+    command = [
+        "zsign",
+        "-k", p12_path,
+        "-p", password,
+        "-m", prov_path,
+        "-o", out_path,
+        "-z", "0", 
+        ipa_path
+    ]
     
-    result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
+    try:
+        subprocess.run(command, check=True)
+        plist_url = f"{BASE_URL}/plist/{job_id}"
+        # This triggers the iOS 'Install' popup
+        return redirect(f"itms-services://?action=download-manifest&url={plist_url}")
+        
+    except Exception as e:
+        return f"Signing failed: {str(e)}", 500
+
+@app.route('/plist/<job_id>')
+def serve_plist(job_id):
+    ipa_direct_url = f"{BASE_URL}/download/{job_id}"
+    
+    # Manifest required by iOS to install apps over the air
+    plist_content = f"""<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+    <key>items</key>
+    <array>
+        <dict>
+            <key>assets</key>
+            <array>
+                <dict>
+                    <key>kind</key>
+                    <string>software-package</string>
+                    <key>url</key>
+                    <string>{ipa_direct_url}</string>
+                </dict>
+            </array>
+            <key>metadata</key>
+            <dict>
+                <key>bundle-identifier</key>
+                <string>com.simplebit.signed</string>
+                <key>bundle-version</key>
+                <string>1.0</string>
+                <key>kind</key>
+                <string>software</string>
+                <key>title</key>
+                <string>SimpleBit Signed App</string>
+            </dict>
+        </dict>
+    </array>
+</dict>
+</plist>"""
     
-    if os.path.exists(output_path):
-        return send_file(output_path, as_attachment=True)
-    else:
-        return f"Error signing: {result.stderr}", 500
+    plist_path = os.path.join(job_id, "manifest.plist")
+    with open(plist_path, "w") as f:
+        f.write(plist_content)
+        
+    return send_file(plist_path, mimetype='text/xml')
+
+@app.route('/download/<job_id>')
+def download_ipa(job_id):
+    # iOS needs the correct MIME type to treat this as an app
+    return send_file(
+        f"{job_id}/signed.ipa", 
+        mimetype='application/octet-stream',
+        as_attachment=True,
+        download_name="signed.ipa"
+    )
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)