ZeroGPU-Leader

Running

App Files Files Community

seawolf2357 commited on Mar 12

Commit

cf37bae

verified ·

1 Parent(s): 045c5a5

Update app.py

Browse files

Files changed (1) hide show

app.py +14 -82

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flask import Flask, render_template, request, redirect, url_for, jsonify, session, Response
 import requests
 import os
 from datetime import timedelta
@@ -30,6 +30,14 @@ HUGGINGFACE_URLS = [
     "https://huggingface.co/spaces/openfree/Article-Generator",
 ]
 # Extract title from the last part of URL
 def extract_title(url):
     parts = url.split("/")
@@ -58,83 +66,6 @@ def validate_token(token):
     return False, None
-# Create a redirect page for each URL
-def create_frame_page(url, title):
-    return f'''
-<!DOCTYPE html>
-<html lang="ko">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>{title}</title>
-    <style>
-        body, html {{
-            margin: 0;
-            padding: 0;
-            height: 100%;
-            overflow: hidden;
-        }}
-        .container {{
-            position: absolute;
-            width: 100%;
-            height: 100%;
-            display: flex;
-            flex-direction: column;
-        }}
-        .header {{
-            background: #f5f5f5;
-            padding: 10px;
-            display: flex;
-            justify-content: space-between;
-            align-items: center;
-            border-bottom: 1px solid #ddd;
-        }}
-        .header h3 {{
-            margin: 0;
-            font-size: 16px;
-        }}
-        .header a {{
-            color: #4CAF50;
-            text-decoration: none;
-        }}
-        .frame-container {{
-            flex-grow: 1;
-            position: relative;
-        }}
-        iframe {{
-            position: absolute;
-            top: 0;
-            left: 0;
-            width: 100%;
-            height: 100%;
-            border: none;
-        }}
-    </style>
-</head>
-<body>
-    <div class="container">
-        <div class="header">
-            <h3>{title}</h3>
-            <a href="{url}" target="_blank">새 창에서 열기</a>
-        </div>
-        <div class="frame-container">
-            <iframe src="{url}" allowfullscreen="true" frameborder="0" allow="accelerometer; camera; encrypted-media; geolocation; gyroscope; microphone; midi"></iframe>
-        </div>
-    </div>
-</body>
-</html>
-'''
-# Route to serve a framed page
-@app.route('/frame/<int:url_id>')
-def frame(url_id):
-    try:
-        url = HUGGINGFACE_URLS[url_id]
-        title = extract_title(url)
-        return create_frame_page(url, title)
-    except IndexError:
-        return "URL not found", 404
 # Homepage route
 @app.route('/')
 def home():
@@ -186,15 +117,16 @@ def get_urls():
     search_query = request.args.get('search', '').lower()
     results = []
-    for idx, url in enumerate(HUGGINGFACE_URLS):
         title = extract_title(url)
         if search_query and search_query not in url.lower() and search_query not in title.lower():
             continue
         results.append({
             'url': url,
-            'frameUrl': f'/frame/{idx}',
             'title': title
         })
@@ -646,7 +578,7 @@ if __name__ == '__main__':
       }
       urls.forEach(item => {
-        const { url, frameUrl, title } = item;
         // Create grid item
         const gridItem = document.createElement('div');
@@ -676,7 +608,7 @@ if __name__ == '__main__':
         // Create iframe to display the content
         const iframe = document.createElement('iframe');
-        iframe.src = frameUrl;
         iframe.title = title;
         iframe.allow = 'accelerometer; camera; encrypted-media; geolocation; gyroscope; microphone; midi';
         iframe.setAttribute('allowfullscreen', '');

+from flask import Flask, render_template, request, redirect, url_for, jsonify, session
 import requests
 import os
 from datetime import timedelta
     "https://huggingface.co/spaces/openfree/Article-Generator",
 ]
+# Transform Huggingface URL to direct space URL
+def transform_url(url):
+    prefix = "https://huggingface.co/spaces/"
+    if url.startswith(prefix):
+        rest = url[len(prefix):]
+        return f"https://{rest.replace('/', '-')}.hf.space"
+    return url
 # Extract title from the last part of URL
 def extract_title(url):
     parts = url.split("/")
     return False, None
 # Homepage route
 @app.route('/')
 def home():
     search_query = request.args.get('search', '').lower()
     results = []
+    for url in HUGGINGFACE_URLS:
         title = extract_title(url)
+        transformed_url = transform_url(url)
         if search_query and search_query not in url.lower() and search_query not in title.lower():
             continue
         results.append({
             'url': url,
+            'embedUrl': transformed_url,
             'title': title
         })
       }
       urls.forEach(item => {
+        const { url, embedUrl, title } = item;
         // Create grid item
         const gridItem = document.createElement('div');
         // Create iframe to display the content
         const iframe = document.createElement('iframe');
+        iframe.src = embedUrl;
         iframe.title = title;
         iframe.allow = 'accelerometer; camera; encrypted-media; geolocation; gyroscope; microphone; midi';
         iframe.setAttribute('allowfullscreen', '');