From 0263a694c6fae541192a313d5bc7d20eee1e70d2 Mon Sep 17 00:00:00 2001
From: Miao Xiang <one111eric@hotmail.com>
Date: Fri, 31 Mar 2023 17:36:32 -0700
Subject: [PATCH] make code cleaner

---
 javascript/extraNetworks.js  |  4 +---
 modules/ui_extra_networks.py | 18 ++++++++++--------
 2 files changed, 11 insertions(+), 11 deletions(-)

diff --git a/javascript/extraNetworks.js b/javascript/extraNetworks.js
index 20007a7ab..bb2884ea0 100644
--- a/javascript/extraNetworks.js
+++ b/javascript/extraNetworks.js
@@ -95,7 +95,7 @@ function saveCardPreview(event, tabname, filename){
     event.preventDefault()
 }
 
-function saveCardDescription(event, tabname, filename,descript){
+function saveCardDescription(event, tabname, filename, descript){
     var textarea = gradioApp().querySelector("#" + tabname + '_description_filename  > label > textarea')
     var button = gradioApp().getElementById(tabname + '_save_description')
     var description = gradioApp().getElementById(tabname+ '_description_input')
@@ -106,8 +106,6 @@ function saveCardDescription(event, tabname, filename,descript){
 
     button.click()
 
-    //alert("add description here!")
-
     event.stopPropagation()
     event.preventDefault()
 }
diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py
index 525cd9f3b..6a90281ec 100644
--- a/modules/ui_extra_networks.py
+++ b/modules/ui_extra_networks.py
@@ -215,7 +215,6 @@ class ExtraNetworksUi:
 
         self.button_save_description = None
         self.description_target_filename = None
-
         self.description_input = None
 
         self.tabname = None
@@ -320,18 +319,21 @@ def setup_ui(ui, gallery):
         outputs=[*ui.pages]
     )
 
-    def save_description(filenamex,images,filename,descrip):
+    def save_description(filename,descrip):
         filename = filename.split('.')[0]+".description.txt"
-        file1 = open(filename,'w')
-        print(file1)
-        file1.write(descrip)
-        file1.close()
+        try: 
+            f = open(filename,'w')
+        except OSError:
+            print ("Could not open file to write: " + filename)
+        with f:
+            f.write(descrip)
+            f.close()
         return [page.create_html(ui.tabname) for page in ui.stored_extra_pages]
     
     ui.button_save_description.click(
         fn=save_description,
-        _js="function(x,y,z){return selected_gallery_index(), y, z]}",
-        inputs=[ui.description_target_filename, gallery, ui.description_target_filename, ui.description_input],
+        _js="function(x,y){return [x,y]}",
+        inputs=[ui.description_target_filename, ui.description_input],
         outputs=[*ui.pages]
     )