Go-Filter-Bot / app.py
Thaimozhi's picture
Satisfy HF ZeroGPU checker with dummy @spaces.GPU decorator
b9b0071
Raw History Blame Contribute Delete
2.09 kB
import os
import shutil
import subprocess
import urllib.request
import tarfile
import gradio as gr
try:
import spaces
@spaces.GPU
def dummy_gpu_fn():
pass
except Exception:
pass
def ensure_go():
if shutil.which("go"):
print("Go compiler found in system PATH.")
return "go"
go_bin = os.path.abspath("go_dist/bin/go")
if os.path.exists(go_bin):
print(f"Go compiler found at {go_bin}")
return go_bin
print("Downloading Go compiler (go1.21.6)...")
url = "https://go.dev/dl/go1.21.6.linux-amd64.tar.gz"
tar_path = "/tmp/go.tar.gz"
urllib.request.urlretrieve(url, tar_path)
print("Extracting Go...")
os.makedirs("go_dist", exist_ok=True)
with tarfile.open(tar_path, "r:gz") as tar:
tar.extractall(path="go_dist_temp")
if os.path.exists("go_dist_temp/go"):
shutil.rmtree("go_dist", ignore_errors=True)
os.rename("go_dist_temp/go", "go_dist")
shutil.rmtree("go_dist_temp", ignore_errors=True)
if os.path.exists(tar_path):
os.remove(tar_path)
return go_bin
# Setup Go environment & PATH
go_cmd = ensure_go()
env = os.environ.copy()
if os.path.exists(os.path.abspath("go_dist/bin")):
env["PATH"] = os.path.abspath("go_dist/bin") + ":" + env.get("PATH", "")
# Set Go bot internal HTTP port to 8080 to prevent collision with Gradio (7860 & 7861)
env["PORT"] = "8080"
# Build Go Bot binary
print("Building Go-Filter-Bot binary...")
subprocess.run([go_cmd, "build", "-o", "Go-Filter-Bot", "."], check=True, env=env)
# Run Go Bot binary in background
print("Starting Go-Filter-Bot process...")
bot_process = subprocess.Popen(["./Go-Filter-Bot"], env=env)
# Create Gradio interface for HF Space supervisor
with gr.Blocks(title="Telegram Go Filter Bot") as demo:
gr.Markdown("""
# 🤖 Telegram Auto-Filter Bot
### Status: Online 24/7 🚀
The bot is active and processing updates in the background.
""")
if __name__ == "__main__":
demo.launch(server_name="0.0.0.0", server_port=7860, ssr_mode=False)