Spaces:
Sleeping
Sleeping
File size: 3,478 Bytes
31ad3d4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 |
import json
import os
import random
import re
import sys
from datetime import datetime
from pathlib import Path
from typing import List, Optional
from uuid import uuid4
import pandas as pd
import gradio as gr
import numpy as np
import requests
from datasets import load_dataset
from huggingface_hub import (
CommitScheduler,
HfApi,
InferenceClient,
login,
snapshot_download,
)
from PIL import Image
from glob import glob
import requests
import re
import time
cached_latest_posts_df = None
last_fetched = None
def get_latest_pots():
global cached_latest_posts_df
global last_fetched
# make sure we don't fetch data too often, limit to 1 request per 10 minutes
now_time = datetime.now()
if last_fetched is not None and (now_time - last_fetched).seconds < 600:
print("Using cached data")
return cached_latest_posts_df
last_fetched = now_time
url = "https://www.reddit.com/r/GamePhysics/.json"
headers = {"User-Agent": "Mozilla/5.0"}
response = requests.get(url, headers=headers)
if response.status_code != 200:
return []
data = response.json()
# Extract posts from the data
posts = data["data"]["children"]
for post in posts:
title = post["data"]["title"]
post_id = post["data"]["id"]
print(f"ID: {post_id}, Title: {title}")
# create [post_id, title] list
examples = [[post["data"]["id"], post["data"]["title"]] for post in posts]
# make a dataframe
examples = pd.DataFrame(examples, columns=["post_id", "title"])
cached_latest_posts_df = examples
return examples
def row_selected(evt: gr.SelectData):
global cached_latest_posts_df
row = evt.index[0]
post_id = cached_latest_posts_df.iloc[row]["post_id"]
return post_id
def load_video(url):
# Regular expression pattern for r/GamePhysics URLs and IDs
pattern = r"https://www\.reddit\.com/r/GamePhysics/comments/([0-9a-zA-Z]+).*|([0-9a-zA-Z]+)"
# Match the URL or ID against the pattern
match = re.match(pattern, url)
if match:
# Extract the post ID from the URL
post_id = match.group(1) or match.group(2)
print(f"Valid GamePhysics post ID: {post_id}")
else:
post_id = url
video_url = f"https://huggingface.co/datasets/asgaardlab/GamePhysicsDailyDump/resolve/main/data/videos/{post_id}.mp4?download=true"
# make sure file exists before returning, make a request without downloading the file
r = requests.head(video_url)
if r.status_code != 200 and r.status_code != 302:
raise gr.Error(
f"Video is not in the repo, please try another post. - {r.status_code }"
)
return video_url
with gr.Blocks() as demo:
gr.Markdown("## Preview GamePhysics")
with gr.Row():
reddit_id = gr.Textbox(lines=1, placeholder="Post url or id here", label="URL")
dummt_title = gr.Textbox(visible=False)
load_btn = gr.Button("Load")
video_player = gr.Video()
gr.Markdown("## Latest Posts")
with gr.Column():
get_latest_pots_btn = gr.Button("Show Latest Posts")
latest_post_dataframe = gr.Dataframe()
load_btn.click(load_video, inputs=[reddit_id], outputs=[video_player])
get_latest_pots_btn.click(get_latest_pots, outputs=[latest_post_dataframe])
# demo.load(get_latest_pots, outputs=[app_examples])
latest_post_dataframe.select(fn=row_selected, outputs=[reddit_id])
demo.launch()
|