Update app.py

app.py

@@ -1,29 +1,30 @@
-from fastapi import FastAPI, Request
+from fastapi import FastAPI
 from pydantic import BaseModel
-import openai
-import os
-
-openai.api_key = os.getenv("OPENAI_API_KEY")
+import re
 
 app = FastAPI()
 
-class ChatRequest(BaseModel):
+class CommandInput(BaseModel):
     message: str
-    history: list = []
+    user_id: int  # the person who sent the command
+    chat_id: int  # the chat where the command was used
 
-@app.post("/chat")
-async def chat(data: ChatRequest):
-    messages = [{"role": "system", "content": "You are a helpful assistant."}]
-    messages += data.history
-    messages.append({"role": "user", "content": data.message})
+@app.post("/command")
+def handle_command(data: CommandInput):
+    message = data.message.strip().lower()
+    response = "Unknown command."
 
-    response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        messages=messages
-    )
+    if message.startswith("/ban"):
+        match = re.search(r"@?(\w+)", message)
+        if match:
+            target = match.group(1)
+            response = f"User @{target} has been banned in chat {data.chat_id}."
+    elif message.startswith("/mute"):
+        match = re.search(r"@?(\w+)", message)
+        if match:
+            target = match.group(1)
+            response = f"User @{target} has been muted in chat {data.chat_id}."
+    elif message.startswith("/help"):
+        response = "Available commands: /ban @user, /mute @user, /help"
 
-    reply = response['choices'][0]['message']['content']
-    return {
-        "reply": reply,
-        "history": messages + [{"role": "assistant", "content": reply}]
-    }
+    return {"response": response}