Update app.py

app.py

@@ -65,7 +65,6 @@ def save_data(t,m,l,n,w,p):
     buffer.write(file_out.encode())
     buffer.seek(0)
     rand = random.randint(1000,9999)
-    "https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/sky.png"
     api.upload_file(
         path_or_fileobj=buffer,
         path_in_repo=f"{title}_{rand}.html",
@@ -73,6 +72,13 @@ def save_data(t,m,l,n,w,p):
         token=token,
         repo_type="space",
     )
+    api.upload_file(
+        path_or_fileobj=f"/images/{title}.png",
+        path_in_repo=f"/images/{title}.png",
+        repo_id=f'{user_name}/{repo_name}',
+        token=token,
+        repo_type="space",
+    )
     #return f"https://broadfield-a.hf.space/?pdfurl={t}&mod={m}&len={l}&nos={n}&wid={w}&pau={p}"
     return f"<div><a href='https://broadfield-a.static.hf.space/{title}_{rand}.html' target='_blank'>https://broadfield-a.static.hf.space/{title}_{rand}.html</a></div>"
 def load_html(url):