Update app.py

app.py

@@ -13,10 +13,14 @@ Article = Article + " Originals go here: https://github.com/AaronCWacker/Yggdras
 
 def fake_gan():
     base_dir = os.getcwd()  # Get the current base directory
-    png_files = [file for file in os.listdir(base_dir) if file.endswith(".png")]  # List all files ending with ".png"
+    jpg_files = [file for file in os.listdir(base_dir) if file.lower().endswith((".jpg", ".jpeg"))]  # List all files ending with ".jpg" or ".jpeg"
+    images = [(random.choice(jpg_files), os.path.splitext(file)[0]) for file in jpg_files]
+    return images
     
+def fake_gan_old3():
+    base_dir = os.getcwd()  # Get the current base directory
+    png_files = [file for file in os.listdir(base_dir) if file.endswith(".jpg")]  # List all files ending with ".png"
     images = [(random.choice(png_files), os.path.splitext(file)[0]) for file in png_files]
-    
     return images
 
 def fake_gan_old2():