Update app.py

app.py

@@ -17,6 +17,9 @@ def get_user_media(username):
     base_url = "https://story.snapchat.com/@"
     mix = base_url + username
 
+    media_folder = "media"
+    os.makedirs(media_folder, exist_ok=True)
+
     r = requests.get(mix, headers=headers)
 
     if not r.ok:
@@ -37,10 +40,10 @@ def get_user_media(username):
             r = requests.get(file_url, stream=True, headers=headers)
 
             if "image" in r.headers['Content-Type']:
-                file_name = r.headers['ETag'].replace('"', '') + ".jpeg"
+                file_name = os.path.join(media_folder, r.headers['ETag'].replace('"', '') + ".jpeg")
 
             elif "video" in r.headers['Content-Type']:
-                file_name = r.headers['ETag'].replace('"', '') + ".mp4"
+                file_name = os.path.join(media_folder, r.headers['ETag'].replace('"', '') + ".mp4")
 
             if os.path.isfile(file_name):
                 continue
@@ -62,9 +65,10 @@ def get_user_media(username):
 def gr_interface(username):
     try:
         get_user_media(username)
-        return f"Successfully downloaded media for {username}."
+        return f"Successfully downloaded media for {username}. Check the 'media' folder."
     except Exception as e:
         return f"Error: {str(e)}"
 
 iface = gr.Interface(fn=gr_interface, inputs="text", outputs="text")
 iface.launch()
+