Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
|
@@ -4,7 +4,7 @@ from huggingface_hub import hf_hub_download
|
|
| 4 |
import pandas as pd
|
| 5 |
import os
|
| 6 |
|
| 7 |
-
app = FastAPI(title="Telegram Database API", description="
|
| 8 |
|
| 9 |
API_KEY = "demo"
|
| 10 |
API_KEY_NAME = "x-api-key"
|
|
@@ -22,19 +22,25 @@ def load_data():
|
|
| 22 |
token = os.environ.get("HF_TOKEN")
|
| 23 |
csv_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME, repo_type="dataset", token=token)
|
| 24 |
|
| 25 |
-
print("Loading database with
|
| 26 |
-
#
|
| 27 |
-
|
| 28 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
df = df.astype(str).fillna("")
|
| 30 |
-
print("Database loaded successfully
|
| 31 |
except Exception as e:
|
| 32 |
print(f"Error loading database: {str(e)}")
|
| 33 |
|
| 34 |
@app.get("/")
|
| 35 |
def read_root():
|
| 36 |
return {
|
| 37 |
-
"message": "
|
| 38 |
"developer": "coder petro"
|
| 39 |
}
|
| 40 |
|
|
@@ -53,7 +59,7 @@ def search_data(
|
|
| 53 |
|
| 54 |
search_lower = query.lower()
|
| 55 |
|
| 56 |
-
# Fast row-filtering using
|
| 57 |
mask = df.apply(lambda row: row.str.lower().str.contains(search_lower)).any(axis=1)
|
| 58 |
results = df[mask].to_dict(orient="records")
|
| 59 |
|
|
|
|
| 4 |
import pandas as pd
|
| 5 |
import os
|
| 6 |
|
| 7 |
+
app = FastAPI(title="Telegram Database API", description="Python Engine by coder petro")
|
| 8 |
|
| 9 |
API_KEY = "demo"
|
| 10 |
API_KEY_NAME = "x-api-key"
|
|
|
|
| 22 |
token = os.environ.get("HF_TOKEN")
|
| 23 |
csv_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME, repo_type="dataset", token=token)
|
| 24 |
|
| 25 |
+
print("Loading database with Robust Python Engine...")
|
| 26 |
+
# Switched to engine='python' and added on_bad_lines='skip' to handle data formatting errors safely
|
| 27 |
+
df = pd.read_csv(
|
| 28 |
+
csv_path,
|
| 29 |
+
sep='|',
|
| 30 |
+
low_memory=False,
|
| 31 |
+
engine='python',
|
| 32 |
+
on_bad_lines='skip'
|
| 33 |
+
)
|
| 34 |
+
# Fill missing values and optimize string types
|
| 35 |
df = df.astype(str).fillna("")
|
| 36 |
+
print("Database loaded successfully without any blocks!")
|
| 37 |
except Exception as e:
|
| 38 |
print(f"Error loading database: {str(e)}")
|
| 39 |
|
| 40 |
@app.get("/")
|
| 41 |
def read_root():
|
| 42 |
return {
|
| 43 |
+
"message": "Robust Optimized API is running 24/7!",
|
| 44 |
"developer": "coder petro"
|
| 45 |
}
|
| 46 |
|
|
|
|
| 59 |
|
| 60 |
search_lower = query.lower()
|
| 61 |
|
| 62 |
+
# Fast row-filtering using pandas optimized mask
|
| 63 |
mask = df.apply(lambda row: row.str.lower().str.contains(search_lower)).any(axis=1)
|
| 64 |
results = df[mask].to_dict(orient="records")
|
| 65 |
|