Update app.py

app.py

@@ -35,6 +35,27 @@ def save_to_hf_space(json_path, repo_id=HF_REPO_ID, token=HF_TOKEN):
         st.warning(f"Erreur lors de la sauvegarde sur Hugging Face: {e}")
         return False
 
+# --- Save to Hugging Face Space from memory (no file write) ---
+def save_to_hf_space_from_memory(json_string, repo_id=HF_REPO_ID, token=HF_TOKEN):
+    if not HF_HUB_AVAILABLE:
+        st.warning("huggingface_hub n'est pas installé. Sauvegarde sur HF désactivée.")
+        return False
+    try:
+        import io
+        api = HfApi()
+        api.upload_file(
+            path_or_fileobj=io.BytesIO(json_string.encode('utf-8')),
+            path_in_repo=HF_JSON_FILENAME,
+            repo_id=repo_id,
+            repo_type="space",
+            token=token
+        )
+        st.toast("💾 Données sauvegardées sur Hugging Face Space!", icon="🤗")
+        return True
+    except Exception as e:
+        st.warning(f"Erreur lors de la sauvegarde sur Hugging Face: {e}")
+        return False
+
 # --- Load from Hugging Face Space ---
 def load_from_hf_space(repo_id=HF_REPO_ID, token=HF_TOKEN):
     if not HF_HUB_AVAILABLE:
@@ -436,11 +457,9 @@ def save_editable_prompts_to_local_and_hf():
             json_string = json.dumps(data_to_save, indent=4, ensure_ascii=False)
             local_ok = save_to_local_file(json_string)
             hf_ok = False
-            # Save to HF Space if possible
-            with open(LOCAL_DATA_FILENAME, 'w', encoding='utf-8') as f:
-                f.write(json_string)
+            # Save to HF Space from memory (no file write to avoid restart)
             if HF_HUB_AVAILABLE:
-                hf_ok = save_to_hf_space(LOCAL_DATA_FILENAME)
+                hf_ok = save_to_hf_space_from_memory(json_string)
             if local_ok and hf_ok:
                 st.toast("💾 Données sauvegardées localement et sur Hugging Face!", icon="💾")
             elif local_ok:
@@ -458,19 +477,10 @@ def save_to_hf_only():
         data_to_save = _preprocess_for_saving(st.session_state.editable_prompts)
         try:
             json_string = json.dumps(data_to_save, indent=4, ensure_ascii=False)
-            # Create temporary file for HF upload without overwriting local file
-            import tempfile
-            with tempfile.NamedTemporaryFile(mode='w', encoding='utf-8', suffix='.json', delete=False) as tmp_file:
-                tmp_file.write(json_string)
-                tmp_file_path = tmp_file.name
-            
+            # Save directly from memory to avoid any file writes
             hf_ok = False
             if HF_HUB_AVAILABLE:
-                hf_ok = save_to_hf_space(tmp_file_path)
-            
-            # Clean up temporary file
-            import os
-            os.unlink(tmp_file_path)
+                hf_ok = save_to_hf_space_from_memory(json_string)
             
             if hf_ok:
                 st.toast("💾 Configuration sauvegardée sur Hugging Face!", icon="🤗")
@@ -503,7 +513,7 @@ def load_editable_prompts_from_local_or_hf():
             json_string_init = json.dumps(data_to_save_init, indent=4, ensure_ascii=False)
             save_to_local_file(json_string_init)
             if HF_HUB_AVAILABLE:
-                save_to_hf_space(LOCAL_DATA_FILENAME)
+                save_to_hf_space_from_memory(json_string_init)
             st.info("Modèles par défaut sauvegardés pour initialisation.")
         except Exception as e: # pragma: no cover
             st.error(f"Erreur sauvegarde initiale: {e}")