Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import os
 import time
 import uuid
@@ -7,6 +6,60 @@ import libtorrent as lt
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import StreamingResponse
 import magic
+from fastapi import Request
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
+from bs4 import BeautifulSoup
+import requests
+
+# Add to top of app.py
+app.mount("/static", StaticFiles(directory="static"), name="static")
+templates = Jinja2Templates(directory=".")
+
+# Add search endpoint
+@app.get("/search")
+async def search_torrents(q: str):
+    try:
+        url = f"https://torrentgalaxy.to/torrents.php?search={q}"
+        headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3'}
+        
+        response = requests.get(url, headers=headers)
+        soup = BeautifulSoup(response.text, 'html.parser')
+        
+        torrents = []
+        for row in soup.select('div.tgxtablerow'):
+            try:
+                title = row.select_one('div:nth-child(4)').get_text(strip=True)
+                magnet = row.select_one('div:nth-child(5) a[href^="magnet:"]')['href']
+                size = row.select_one('div:nth-child(8)').get_text(strip=True)
+                seeds = row.select_one('div:nth-child(11) font').get_text(strip=True)
+                
+                torrents.append({
+                    'title': title,
+                    'magnet': magnet,
+                    'size': size,
+                    'seeds': seeds
+                })
+            except:
+                continue
+
+        return torrents[:10]  # Return top 10 results
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+# Add mime type endpoint
+@app.get("/stream/{torrent_id}/mime")
+async def get_mime_type(torrent_id: str):
+    if torrent_id not in active_sessions:
+        raise HTTPException(status_code=404, detail="Torrent session not found")
+    
+    file_path = active_sessions[torrent_id]["file_path"]
+    mime = magic.Magic(mime=True)
+    return {"mime_type": mime.from_file(file_path)}
+
+
+
 
 app = FastAPI()
 active_sessions = {}