Update app.py

app.py

@@ -135,7 +135,12 @@ def inpaint(prompt, filename, mode):
         global COUNTER
         COUNTER += 1
         print(f'[I]: mode 1')
-        return f'current counter is: {COUNTER}\nis there a file called {COUNTER-1}.txt? {str(COUNTER-1)+".txt" in os.listdir()}', Oimg
+        try:
+            with open(f'{filename}.txt','w') as file:
+                file.write(prompt)
+            return f'Success', Oimg
+        except: pass   
+        return f'Failure to write', Oimg
     # End of the little test
         
     # Set up the images