Update app.py

add youtube links

app.py

@@ -90,8 +90,12 @@ def query_memes(user_input: str):
     raw_url = meme.link
     description = meme.description
 
-    file_id = raw_url.split("/file/d/")[1].split("/")[0]
-    embed_url = f"https://drive.google.com/file/d/{file_id}/preview"
+    if "drive.google" in raw_url:
+        file_id = raw_url.split("/file/d/")[1].split("/")[0]
+        embed_url = f"https://drive.google.com/file/d/{file_id}/preview"
+    else:
+        embed_url = raw_url
+    
     embed_html = f'''
     <div style="position: relative; padding-bottom: 56.25%; height: 0; overflow: hidden; border-radius: 8px; box-shadow: 0 4px 10px rgba(0,0,0,0.2); margin-top: 1em;">
       <iframe src="{embed_url}"
@@ -100,7 +104,6 @@ def query_memes(user_input: str):
       </iframe>
     </div>
     '''
-
     return embed_html, description