Update app.py

app.py

@@ -1,8 +1,7 @@
-from flask import Flask, render_template, request, redirect, url_for, jsonify, session, Response
+from flask import Flask, render_template, request, redirect, url_for, jsonify, session
 import requests
 import os
 from datetime import timedelta
-from urllib.parse import urlparse, urljoin
 
 app = Flask(__name__)
 app.secret_key = os.urandom(24)  # Session encryption key
@@ -31,38 +30,14 @@ HUGGINGFACE_URLS = [
     "https://huggingface.co/spaces/openfree/Article-Generator",
 ]
 
-# Extract model/space info from URL
-def extract_model_info(url):
+# Get direct iframe URL for Huggingface spaces
+def get_iframe_url(url):
     parts = url.split('/')
-    if len(parts) < 6:
-        return None
-    
-    if parts[3] == 'spaces' or parts[3] == 'models':
-        return {
-            'type': parts[3],
-            'owner': parts[4],
-            'repo': parts[5],
-            'full_id': f"{parts[4]}/{parts[5]}"
-        }
-    elif len(parts) >= 5:
-        # Other URL format
-        return {
-            'type': 'models',  # Default
-            'owner': parts[3],
-            'repo': parts[4],
-            'full_id': f"{parts[3]}/{parts[4]}"
-        }
-    
-    return None
-
-# Extract direct embed URL
-def get_embed_url(url):
-    model_info = extract_model_info(url)
-    if not model_info or model_info['type'] != 'spaces':
-        return url
-    
-    # For spaces, use the embedded version
-    return f"https://huggingface.co/spaces/{model_info['owner']}/{model_info['repo']}/embed"
+    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):
@@ -145,19 +120,14 @@ def get_urls():
     results = []
     for url in HUGGINGFACE_URLS:
         title = extract_title(url)
-        model_info = extract_model_info(url)
-        
-        if not model_info:
-            continue
         
         if search_query and search_query not in url.lower() and search_query not in title.lower():
             continue
         
         results.append({
             'url': url,
-            'embedUrl': get_embed_url(url),
-            'title': title,
-            'model_info': model_info
+            'iframeUrl': get_iframe_url(url),
+            'title': title
         })
     
     return jsonify(results)
@@ -272,7 +242,7 @@ if __name__ == '__main__':
     
     .grid-container {
       display: grid;
-      grid-template-columns: repeat(auto-fill, minmax(480px, 1fr));
+      grid-template-columns: repeat(auto-fill, minmax(450px, 1fr));
       gap: 1.5rem;
     }
     
@@ -283,7 +253,7 @@ if __name__ == '__main__':
       box-shadow: 0 4px 15px rgba(0,0,0,0.1);
       display: flex;
       flex-direction: column;
-      height: 650px;
+      height: 600px;
     }
     
     .grid-header {
@@ -608,7 +578,7 @@ if __name__ == '__main__':
       }
       
       urls.forEach(item => {
-        const { url, embedUrl, title } = item;
+        const { url, iframeUrl, title } = item;
         
         // Create grid item
         const gridItem = document.createElement('div');
@@ -638,7 +608,7 @@ if __name__ == '__main__':
         
         // Create iframe to display the content
         const iframe = document.createElement('iframe');
-        iframe.src = embedUrl; // Use the embed URL directly
+        iframe.src = iframeUrl;
         iframe.title = title;
         iframe.allow = 'accelerometer; camera; encrypted-media; geolocation; gyroscope; microphone; midi';
         iframe.setAttribute('allowfullscreen', '');