Update app.py

app.py

@@ -1,79 +1,93 @@
+import gradio as gr
 import requests
 import csv
-import gradio as gr
-import os
+import re
+from datetime import timedelta
 
-def fetch_comments(api_key, video_id):
-    """
-    Fetch comments from a YouTube video using the YouTube Data API.
+# YouTube API key and required endpoint
+API_KEY = 'YOUR_YOUTUBE_API_KEY'  # Replace with your API key
+YOUTUBE_API_URL = 'https://www.googleapis.com/youtube/v3/videos'
 
-    Parameters:
-    - api_key: Google API key.
-    - video_id: YouTube video ID.
+def parse_duration(duration):
+    """
+    Parse ISO 8601 duration string to seconds.
+    Example: "PT1H10M5S" -> 4205 seconds
+    """
+    regex = re.match(r"PT(\d+H)?(\d+M)?(\d+S)?", duration)
+    hours = int(regex.group(1)[:-1] if regex.group(1) else 0)
+    minutes = int(regex.group(2)[:-1] if regex.group(2) else 0)
+    seconds = int(regex.group(3)[:-1] if regex.group(3) else 0)
+    return hours * 3600 + minutes * 60 + seconds
 
-    Returns:
-    - Path to the CSV file containing commenters' usernames and IDs.
+def fetch_video_data(video_ids):
+    """
+    Fetch video details (title, duration) from YouTube API.
     """
-    base_url = "https://www.googleapis.com/youtube/v3/commentThreads"
-    params = {
-        "part": "snippet",
-        "videoId": video_id,
-        "key": api_key,
-        "maxResults": 100
-    }
+    video_data = []
+    total_time_spent = 0  # Total time spent in seconds
 
-    response = requests.get(base_url, params=params)
-    if response.status_code == 200:
+    for video_id in video_ids:
+        params = {
+            'part': 'contentDetails,snippet',
+            'id': video_id,
+            'key': API_KEY
+        }
+        response = requests.get(YOUTUBE_API_URL, params=params)
         data = response.json()
-        items = data.get("items", [])
 
-        # Prepare CSV
-        filename = "comments.csv"
-        with open(filename, mode="w", newline="", encoding="utf-8") as file:
-            writer = csv.writer(file)
-            writer.writerow(["Username", "User ID", "Comment"])
-            
-            for item in items:
-                snippet = item.get("snippet", {}).get("topLevelComment", {}).get("snippet", {})
-                username = snippet.get("authorDisplayName", "N/A")
-                user_id = snippet.get("authorChannelId", {}).get("value", "N/A")
-                comment = snippet.get("textDisplay", "N/A")
-                writer.writerow([username, user_id, comment])
-        
-        return filename
-    else:
-        return f"Error: {response.status_code} - {response.text}"
+        if 'items' in data:
+            item = data['items'][0]
+            title = item['snippet']['title']
+            duration = item['contentDetails']['duration']
+            video_duration_seconds = parse_duration(duration)
+            total_time_spent += video_duration_seconds
 
-def gradio_interface(api_key, video_id):
+            video_data.append({
+                'Video ID': video_id,
+                'Title': title,
+                'Duration (seconds)': video_duration_seconds
+            })
+
+    # Convert total time spent to hours and minutes
+    total_time_spent_in_hours = str(timedelta(seconds=total_time_spent))
+    
+    # Save data to CSV
+    filename = "videos_time_spent.csv"
+    with open(filename, mode="w", newline="", encoding="utf-8") as file:
+        writer = csv.DictWriter(file, fieldnames=["Video ID", "Title", "Duration (seconds)"])
+        writer.writeheader()
+        writer.writerows(video_data)
+
+    return filename, total_time_spent_in_hours
+
+def gradio_interface(video_ids):
     """
-    Gradio interface function to fetch comments and create a downloadable CSV.
+    Gradio interface function to process video IDs and generate time statistics.
     """
-    csv_path = fetch_comments(api_key, video_id)
-    if csv_path.startswith("Error"):
-        return csv_path, None
-    return "CSV file created successfully. Click the link to download.", csv_path
+    video_ids_list = video_ids.split(",")  # Convert input string to list
+    csv_file, total_time = fetch_video_data(video_ids_list)
+    return f"Total time spent: {total_time} on these videos. Download the CSV below.", csv_file
 
-# Gradio App
+# Gradio App setup
 with gr.Blocks() as demo:
-    gr.Markdown("### YouTube Comment Exporter")
-    gr.Markdown("Fetch usernames and user IDs of commenters on a YouTube video.")
-
+    gr.Markdown("### Time Spent on YouTube Videos")
+    gr.Markdown("Enter a list of YouTube video IDs (comma separated) to calculate the time spent on each video.")
+    
     with gr.Row():
-        api_key_input = gr.Textbox(label="API Key", placeholder="Enter your Google API key", type="password")
-        video_id_input = gr.Textbox(label="Video ID", placeholder="Enter the YouTube video ID")
+        video_ids_input = gr.Textbox(label="Enter Video IDs", placeholder="e.g., dQw4w9WgXcQ, kJQP7kiw5Fk")
     
     with gr.Row():
-        fetch_button = gr.Button("Fetch Comments")
+        submit_button = gr.Button("Calculate Time Spent")
     
     with gr.Row():
-        message_output = gr.Textbox(label="Status", interactive=False)
+        message_output = gr.Textbox(label="Result", interactive=False)
         download_link = gr.File(label="Download CSV")
 
-    fetch_button.click(
+    submit_button.click(
         fn=gradio_interface,
-        inputs=[api_key_input, video_id_input],
+        inputs=[video_ids_input],
         outputs=[message_output, download_link]
     )
 
-# Launch the App
+# Launch the app
 demo.launch()