Update app.py

app.py

@@ -110,7 +110,7 @@ def get_video_info(video_id):
     video_info = {
         'title': response['items'][0]['snippet']['title'],
         'channel_title': response['items'][0]['snippet']['channelTitle'],
-        'category': response['items'][0]['snippet']['categoryId'],
+        'category': category_dict.get(int(response['items'][0]['snippet']['categoryId']), "Unknown category"),
         'tags': '|'.join(response['items'][0]['snippet'].get('tags', [])),
         'views': response['items'][0]['statistics']['viewCount'],
         'likes': response['items'][0]['statistics'].get('likeCount', 0),
@@ -141,6 +141,39 @@ st.markdown("### Generate comments using video metadata!")
 
 st.image(Image.open("clown.jpg"), width=400)
 
+category_dict = {1: 'Film & Animation',
+2: 'Autos & Vehicles',
+10: 'Music',
+15: 'Pets & Animals',
+17: 'Sports',
+18: 'Short Movies',
+19: 'Travel & Events',
+20: 'Gaming',
+21: 'Videoblogging',
+22: 'People & Blogs',
+23: 'Comedy',
+24: 'Entertainment',
+25: 'News & Politics',
+26: 'Howto & Style',
+27: 'Education',
+28: 'Science & Technology',
+29: 'Nonprofits & Activism',
+30: 'Movies',
+31: 'Anime/Animation',
+32: 'Action/Adventure',
+33: 'Classics',
+34: 'Comedy',
+35: 'Documentary',
+36: 'Drama',
+37: 'Family',
+38: 'Foreign',
+39: 'Horror',
+40: 'Sci-Fi/Fantasy',
+41: 'Thriller',
+42: 'Shorts',
+43: 'Shows',
+44: 'Trailers'}
+
 tokenizer, model = load_model()
 
 link = st.text_input("Enter YouTube Link", placeholder="Paste URL here...")