Update app.py

app.py

@@ -99,11 +99,11 @@ async def improve_prompt(prompt):
         return f"Error mejorando el prompt: {e}"
 
 def get_storage():
-    files = [{"name": str(file.resolve()), "size": file.stat().st_size,}
-        for file in DATA_PATH.glob("*.jpg") 
-        if file.is_file()]
+    files = [{"name": str(file.resolve()), "size": file.stat().st_size}
+             for file in DATA_PATH.glob("*.jpg") if file.is_file()]
+    
     usage = sum([f['size'] for f in files])
-    return [file["name"] for f in files], f"Uso total: {usage/(1024.0 ** 3):.3f}GB"
+    return [f["name"] for f in files], f"Uso total: {usage/(1024.0 ** 3):.3f}GB"
 
 def get_prompts():
     prompt_files = [file for file in DATA_PATH.glob("*.txt") if file.is_file()]
@@ -187,4 +187,4 @@ for idx, file in enumerate(files):
         if st.button(f"Borrar Imagen {idx+1}", key=f"delete_{idx}"):
             os.remove(file)
             if prompt_file:
-                os.remove(prompt_file)
+                os.remove(prompt_file)