New comparison feature.

modules/web_app.py

@@ -569,30 +569,65 @@ def refine_bible_route(id):
     if is_project_locked(id):
         flash("Project is locked. Clone it to make changes.")
         return redirect(url_for('view_project', id=id))
-
-    instruction = request.form.get('instruction')
-    if not instruction:
-        flash("Instruction required.")
-        return redirect(url_for('view_project', id=id))
-        
-    bible_path = os.path.join(proj.folder_path, "bible.json")
-    bible = utils.load_json(bible_path)
     
-    if bible:
-        try: ai.init_models()
-        except: pass
-        
-        if not ai.model_logic:
-            flash("AI models not initialized.")
-            return redirect(url_for('view_project', id=id))
+    # Handle JSON request (AJAX) or Form request
+    data = request.json if request.is_json else request.form
+    instruction = data.get('instruction')
+    
+    if not instruction:
+        return {"error": "Instruction required"}, 400
 
-        new_bible = story.refine_bible(bible, instruction, proj.folder_path)
-        
-        if new_bible:
-            with open(bible_path, 'w') as f: json.dump(new_bible, f, indent=2)
+    source_type = data.get('source', 'original')
+    selected_keys = data.get('selected_keys')
+    if isinstance(selected_keys, str):
+        try: selected_keys = json.loads(selected_keys)
+        except: selected_keys = []
+
+    # Start Background Task
+    task = refine_bible_task(proj.folder_path, instruction, source_type, selected_keys)
+    
+    return {"status": "queued", "task_id": task.id}
+
+@app.route('/project/<int:id>/refine_bible/confirm', methods=['POST'])
+@login_required
+def confirm_bible_refinement(id):
+    proj = db.session.get(Project, id) or Project.query.get_or_404(id)
+    if proj.user_id != current_user.id: return "Unauthorized", 403
+    
+    action = request.form.get('action')
+    draft_path = os.path.join(proj.folder_path, "bible_draft.json")
+    bible_path = os.path.join(proj.folder_path, "bible.json")
+    
+    if action == 'accept_all':
+        if os.path.exists(draft_path):
+            shutil.move(draft_path, bible_path)
             flash("Bible updated successfully.")
         else:
-            flash("AI failed to update Bible. Check logs.")
+            flash("Draft expired or missing.")
+            
+    elif action == 'accept_selected':
+        if os.path.exists(draft_path) and os.path.exists(bible_path):
+            selected_keys_json = request.form.get('selected_keys', '[]')
+            try:
+                selected_keys = json.loads(selected_keys_json)
+                draft = utils.load_json(draft_path)
+                original = utils.load_json(bible_path)
+                
+                original = _merge_selected_changes(original, draft, selected_keys)
+                                
+                with open(bible_path, 'w') as f: json.dump(original, f, indent=2)
+                os.remove(draft_path) # Cleanup draft after merge
+                flash(f"Merged {len(selected_keys)} changes into Bible.")
+                
+            except Exception as e:
+                flash(f"Merge failed: {e}")
+        else:
+            flash("Files missing.")
+            
+    elif action == 'decline':
+        if os.path.exists(draft_path):
+            os.remove(draft_path)
+            flash("Changes discarded.")
             
     return redirect(url_for('view_project', id=id))