Update app.py

app.py

@@ -163,25 +163,33 @@ def index():
 
 @app.route("/api/store", methods=["POST"])
 def store():
-    """Store encrypted secret with enhanced features - MODIFIED FOR FRONTEND FILE ENCRYPTION"""
+    """Store encrypted secret with enhanced features - FIXED FOR JSON REQUEST"""
     try:
-        form = request.form
-        data = form.get("data")
+        # Check if request is JSON or form data
+        if request.is_json:
+            # Handle JSON request (from frontend)
+            json_data = request.get_json()
+            data = json_data.get("data")
+            ttl = int(json_data.get("ttl", 300))
+            view_once = json_data.get("view_once", False)
+            delay_seconds = int(json_data.get("delay_seconds", 0))
+            theme = json_data.get("theme", "default")
+            password_hint = json_data.get("password_hint", "")
+            has_file = json_data.get("has_file", False)
+        else:
+            # Handle form data (backward compatibility)
+            form = request.form
+            data = form.get("data")
+            ttl = int(form.get("ttl", 300))
+            view_once = form.get("view_once", "false").lower() == "true"
+            delay_seconds = int(form.get("delay_seconds", 0))
+            theme = form.get("theme", "default")
+            password_hint = form.get("password_hint", "")
+            has_file = form.get("has_file", "false").lower() == "true"
         
         if not data:
             return jsonify({"error": "Data is required"}), 400
         
-        # Parse parameters
-        ttl = int(form.get("ttl", 300))
-        view_once = form.get("view_once", "false").lower() == "true"
-        delay_seconds = int(form.get("delay_seconds", 0))
-        theme = form.get("theme", "default")
-        password_hint = form.get("password_hint", "")
-        has_file = form.get("has_file", "false").lower() == "true"  # NEW: File flag from frontend
-        
-        # REMOVED: File upload handling (no longer needed)
-        # The frontend now encrypts files into the data field
-        
         # Generate IDs
         secret_id = str(uuid.uuid4())
         short_id = generate_short_id()
@@ -190,15 +198,15 @@ def store():
         while short_id in SHORT_LINKS:
             short_id = generate_short_id()
         
-        # Store secret - SIMPLIFIED STRUCTURE
+        # Store secret
         SECRETS[secret_id] = {
-            "data": data,  # Now contains encrypted message + file data
+            "data": data,  # Contains encrypted message + file data
             "expire_at": time.time() + ttl,
             "view_once": view_once,
             "delay_seconds": delay_seconds,
             "theme": theme,
             "password_hint": password_hint,
-            "has_file": has_file,  # NEW: Flag to indicate if encrypted data contains file
+            "has_file": has_file,
             "created_at": time.time(),
             "creator_ip": get_client_ip(request),
             "access_count": 0
@@ -218,10 +226,11 @@ def store():
             "short_url": f"{base_url}/s/{short_id}",
             "qr_code": qr_code,
             "expires_at": SECRETS[secret_id]["expire_at"],
-            "has_file": has_file  # Return the file flag
+            "has_file": has_file
         })
         
     except Exception as e:
+        print(f"Error in store endpoint: {str(e)}")  # Add logging
         return jsonify({"error": str(e)}), 500
 
 @app.route("/api/fetch/<secret_id>")