Create app.py

app.py

@@ -0,0 +1,64 @@
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+from typing import List, Tuple
+import requests
+from bs4 import BeautifulSoup
+
+app = FastAPI()
+
+class SearchQuery(BaseModel):
+    query: str
+
+class LinkResults(BaseModel):
+    four_links_click: List[str]
+    direct4_links: List[str]
+    magnet_links: List[str]
+
+def scrape_links(query: str) -> List[str]:
+    try:
+        url = f"https://www.full4movies.fyi/?s={query}"
+        response = requests.get(url)
+
+        if response.status_code == 200:
+            soup = BeautifulSoup(response.text, 'html.parser')
+            elements = soup.find_all('h2', class_='blog-entry-title entry-title')
+            links = [element.find('a')['href'] for element in elements]
+            return links
+        else:
+            raise HTTPException(status_code=response.status_code, detail="Failed to retrieve the webpage.")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+def find_links(url: str) -> Tuple[List[str], List[str], List[str]]:
+    try:
+        response = requests.get(url)
+        if response.status_code == 200:
+            soup = BeautifulSoup(response.text, 'html.parser')
+            links = soup.find_all('a', href=True)
+            
+            four_links_click = [link['href'] for link in links if '4links.click' in link['href']]
+            direct4_links = [link['href'] for link in links if 'direct4.link' in link['href']]
+            magnet_links = [link['href'] for link in links if link['href'].startswith('magnet:')]
+
+            return four_links_click, direct4_links, magnet_links
+        else:
+            raise HTTPException(status_code=response.status_code, detail="Failed to retrieve the selected webpage.")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"An error occurred while scraping the selected link: {str(e)}")
+
+@app.post("/search", response_model=List[str])
+def search_links(payload: SearchQuery):
+    links = scrape_links(payload.query)
+    if not links:
+        raise HTTPException(status_code=404, detail="No links found.")
+    return links
+
+@app.get("/details", response_model=LinkResults)
+def get_details(url: str):
+    four_links_click, direct4_links, magnet_links = find_links(url)
+
+    return LinkResults(
+        four_links_click=four_links_click,
+        direct4_links=direct4_links,
+        magnet_links=magnet_links
+    )