Update main.py

main.py

@@ -2,6 +2,7 @@ import logging
 import time
 import os
 import requests
+import platform
 from pyrogram import *
 from pyrogram.types import *
 from pyrogram.errors import *
@@ -12,6 +13,7 @@ logging.basicConfig(
 )
 logging.getLogger("pyrogram").setLevel(logging.WARNING)
 
+
 client = Client(
     "Ryzenth",
     api_id=API_ID,
@@ -75,21 +77,13 @@ We’re excited to have you here! Here’s what you can enjoy:
 - **V1**: Free and unlimited access for all users.
 - **V2**: Premium features with a 30-day free trial.
 
-🚨 **Important Rules**:
-To ensure a safe and fair environment for everyone, please adhere to the following rules:
-1. **Prohibited Actions**:
-   - Exploiting or abusing the API for malicious purposes.
-   - Sharing API keys or credentials with unauthorized users.
-   - Violating any applicable laws or regulations.
-
-2. **Consequences**:
-   - Breaking the rules will result in immediate suspension or banning.
-   - Harassment or misuse of the API will not be tolerated.
-
 🔒 **Your Safety Matters**:
 If you encounter any issues or feel harassed, please contact our support team immediately.
 """
 
+def get_user_agent() -> str:
+    return f"Ryzenth/Python-{platform.python_version()}"
+
 @client.on_callback_query(filters.regex("^banyes_"))
 async def cb_banned_nothing(client, query):
     action = query.data.split("_", 1)[1]
@@ -192,7 +186,10 @@ async def cb_banned_and_key(client, query):
     response = requests.post(
         url,
         params=params,
-        headers={"x-api-key": os.environ.get("AKENOX_KEY")}
+        headers={
+            "User-Agent": get_user_agent(),
+            "x-api-key": os.environ.get("AKENOX_KEY")
+        }
     )
     if response.status_code != 200:
         return await query.answer("Error api try again", True)
@@ -225,7 +222,10 @@ async def cb_run_cohere(client, query):
     response = requests.post(
         url,
         params={"user_id": user_id},
-        headers={"x-api-key": os.environ.get("AKENOX_KEY")}
+        headers={
+            "User-Agent": get_user_agent(),
+            "x-api-key": os.environ.get("AKENOX_KEY")
+        }
     )
     if response.status_code != 200:
         return await query.answer("Error api try again", True)
@@ -234,7 +234,10 @@ async def cb_run_cohere(client, query):
     if data_json.get("apiKey", False):
         response_two = requests.get(
             url_two,
-            headers={"x-api-key": data_json["apiKey"]},
+            headers={
+                "User-Agent": get_user_agent(),
+                "x-api-key": data_json["apiKey"]
+            },
             params={"query": "hello world!"}
         ).json()
         if response_two.get("results") is None:
@@ -250,7 +253,10 @@ async def cb_new_cohere(client, query):
     response = requests.post(
         url,
         params={"user_id": user_id},
-        headers={"x-api-key": os.environ.get("AKENOX_KEY")}
+        headers={
+            "User-Agent": get_user_agent(),
+            "x-api-key": os.environ.get("AKENOX_KEY")
+        }
     )
     if response.status_code != 200:
         return await query.answer("Error api try again", True)
@@ -295,7 +301,10 @@ async def cb_new_revokekey(client, query):
                 "user_id": user_id,
                 "username": query.from_user.username
             },
-            headers={"x-api-key": os.environ.get("AKENOX_KEY")}
+            headers={
+                "User-Agent": get_user_agent(),
+                "x-api-key": os.environ.get("AKENOX_KEY")
+            }
         )
         response.raise_for_status()
         data_json = response.json()
@@ -365,7 +374,10 @@ async def cb_new_keyprem(client, query):
         response = requests.post(
             url,
             params={"user_id": user_id},
-            headers={"x-api-key": os.environ.get("AKENOX_KEY")}
+            headers={
+                "User-Agent": get_user_agent(),
+                "x-api-key": os.environ.get("AKENOX_KEY")
+            }
         )
         response.raise_for_status()
         data_json = response.json()
@@ -440,11 +452,17 @@ async def cb_new_key(client, query):
                 "user_id": user_id,
                 "username": query.from_user.username
             },
-            headers={"x-api-key": os.environ.get("AKENOX_KEY")}
+            headers={
+                "User-Agent": get_user_agent(),
+                "x-api-key": os.environ.get("AKENOX_KEY")
+            }
         )
         response_two = requests.get(
             url_two,
-            headers={"x-api-key": os.environ.get("AKENOX_KEY")}
+            headers={
+                "User-Agent": get_user_agent(),
+                "x-api-key": os.environ.get("AKENOX_KEY")
+            }
         )
         response.raise_for_status()
         response_two.raise_for_status()