env

.gitignore

@@ -2,3 +2,5 @@
 .venv
 # pycache
 __pycache__
+# .env
+.env

app.py

@@ -1,3 +1,4 @@
+import os
 import requests
 from flask import Flask, request, jsonify
 from flask_cors import CORS
@@ -6,6 +7,7 @@ from tempfile import NamedTemporaryFile
 
 app = Flask(__name__)
 CORS(app)
+LOAD_BALANCER_URL = os.getenv("LOAD_BALANCER_URL")
 
 # Function to download data from API and save to a temporary file
 def download_and_save_data(url: str) -> str:
@@ -17,8 +19,8 @@ def download_and_save_data(url: str) -> str:
         return tmp_file.name
 
 # URLs for the data
-films_url = 'https://unicone-studio-load-balancer.hf.space/api/film/all'
-tv_series_url = 'https://unicone-studio-load-balancer.hf.space/api/tv/all'
+films_url = f'{LOAD_BALANCER_URL}/api/film/all'
+tv_series_url = f'{LOAD_BALANCER_URL}/api/tv/all'
 
 # Download and save data
 films_file = download_and_save_data(films_url)

test_api.py

@@ -2,7 +2,7 @@ import requests
 import json
 
 # Define the API endpoint
-url = "http://127.0.0.1:5000/api/search"
+url = "https://unicone-studio-search.hf.space/api/search"
 
 # Define the search queries you want to test
 test_queries = [