Update app.py

app.py

@@ -7,8 +7,6 @@ import pandas as pd
 from io import BytesIO
 import base64
 import re
-from PIL import Image
-from io import BytesIO
 
 # Model yang digunakan sekarang hasbigani/indobertsentiment
 model_name = "hasbigani/indobertsentiment"
@@ -101,23 +99,15 @@ def generate_visualization(results):
     plt.close()
     return f"<img src='data:image/png;base64,{encoded}'/>"
 
-# Fungsi untuk mengambil thumbnail dari URL YouTube
-def get_thumbnail(url):
-    video_id = extract_video_id(url)
-    if video_id:
-        return f"https://img.youtube.com/vi/{video_id}/0.jpg"
-    return None
-
 # Fungsi utama untuk analisis sentimen
 def analyze_sentiment(url, jumlah):
     comments = get_youtube_comments(url, max_comments=jumlah)
     if not comments:
-        return pd.DataFrame(), "Tidak ada komentar ditemukan", None
+        return pd.DataFrame(), "Tidak ada komentar ditemukan"
     results = classify_sentiment(comments)
     df = pd.DataFrame(results, columns=["Komentar", "IndoBERT", "Confidence"])
     chart = generate_visualization(results)
-    thumbnail_url = get_thumbnail(url)
-    return df, chart, thumbnail_url
+    return df, chart
 
 gr.Interface(
     fn=analyze_sentiment,
@@ -127,10 +117,8 @@ gr.Interface(
     ],
     outputs=[
         gr.Dataframe(label="Preview Komentar dan Sentimen"),
-        gr.HTML(label="Visualisasi Sentimen"),
-        gr.Image(label="Thumbnail Video YouTube", type="url")
+        gr.HTML(label="Visualisasi Sentimen")
     ],
     title="Analisis Komentar YouTube 🇮🇩 dengan IndoBERT",
     description="Masukkan URL YouTube dan sistem akan menarik komentar dan menganalisisnya menggunakan model IndoBERT."
 ).launch()
-