ZeroGPU-Leader

Running

App Files Files Community

seawolf2357 commited on Mar 12

Commit

045c5a5

verified ·

1 Parent(s): 98d3d46

Update app.py

Browse files

Files changed (1) hide show

app.py +82 -14

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flask import Flask, render_template, request, redirect, url_for, jsonify, session
 import requests
 import os
 from datetime import timedelta
@@ -30,15 +30,6 @@ HUGGINGFACE_URLS = [
     "https://huggingface.co/spaces/openfree/Article-Generator",
 ]
-# Get direct iframe URL for Huggingface spaces
-def get_iframe_url(url):
-    parts = url.split('/')
-    if len(parts) >= 6 and parts[3] == 'spaces':
-        owner = parts[4]
-        repo = parts[5]
-        return f"https://{owner}-{repo}.hf.space"
-    return url
 # Extract title from the last part of URL
 def extract_title(url):
     parts = url.split("/")
@@ -67,6 +58,83 @@ def validate_token(token):
     return False, None
 # Homepage route
 @app.route('/')
 def home():
@@ -118,7 +186,7 @@ def get_urls():
     search_query = request.args.get('search', '').lower()
     results = []
-    for url in HUGGINGFACE_URLS:
         title = extract_title(url)
         if search_query and search_query not in url.lower() and search_query not in title.lower():
@@ -126,7 +194,7 @@ def get_urls():
         results.append({
             'url': url,
-            'iframeUrl': get_iframe_url(url),
             'title': title
         })
@@ -578,7 +646,7 @@ if __name__ == '__main__':
       }
       urls.forEach(item => {
-        const { url, iframeUrl, title } = item;
         // Create grid item
         const gridItem = document.createElement('div');
@@ -608,7 +676,7 @@ if __name__ == '__main__':
         // Create iframe to display the content
         const iframe = document.createElement('iframe');
-        iframe.src = iframeUrl;
         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, Response
 import requests
 import os
 from datetime import timedelta
     "https://huggingface.co/spaces/openfree/Article-Generator",
 ]
 # Extract title from the last part of URL
 def extract_title(url):
     parts = url.split("/")
     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():
     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():
         results.append({
             'url': url,
+            'frameUrl': f'/frame/{idx}',
             'title': title
         })
       }
       urls.forEach(item => {
+        const { url, frameUrl, title } = item;
         // Create grid item
         const gridItem = document.createElement('div');
         // 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', '');