Update modules/app.py

modules/app.py

@@ -1,51 +1,32 @@
+from fastapi import FastAPI, Query, Request
 import os
-import requests
-import json
-from io import BytesIO
-
-from fastapi import FastAPI
-from fastapi.staticfiles import StaticFiles
-from fastapi.responses import FileResponse, StreamingResponse
-
-from modules.inference import infer_t5
-from modules.dataset import query_emotion
-
-# https://huggingface.co/settings/tokens
-# https://huggingface.co/spaces/{username}/{space}/settings
-API_TOKEN = os.getenv("BIG_GAN_TOKEN")
+from typing import Any, Optional, Dict
+from person import Person
 
-app = FastAPI(docs_url=None, redoc_url=None)
-
-app.mount("/static", StaticFiles(directory="static"), name="static")
+import requests
 
+app = FastAPI(
+    title="Test"
+)
 
 @app.head("/")
 @app.get("/")
-def index() -> FileResponse:
-    return FileResponse(path="static/index.html", media_type="text/html")
-
-
-@app.get("/infer_biggan")
-def biggan(input):
-    output = requests.request(
-        "POST",
-        "https://api-inference.huggingface.co/models/osanseviero/BigGAN-deep-128",
-        headers={"Authorization": f"Bearer {API_TOKEN}"},
-        data=json.dumps(input),
-    )
-
-    return StreamingResponse(BytesIO(output.content), media_type="image/png")
-
-
-@app.get("/infer_t5")
-def t5(input):
-    output = infer_t5(input)
-
-    return {"output": output}
-
-
-@app.get("/query_emotion")
-def emotion(start, end):
-    output = query_emotion(int(start), int(end))
-
-    return {"output": output}
+async def get_user_info(request: Request) -> Dict[str, str]:
+    client_host = request.headers.get('x-forwarded-for')
+    user_agent = request.headers.get('user-agent')
+
+    print(request.headers)
+    
+    return {
+        "IP Address": client_host,
+        "User Agent": user_agent,
+    }
+
+def push_noti_tele(message):
+    url = f"https://api.telegram.org/bot7210957168:AAEy0umCg4VTCilHTE3yICnsQT8cPrbeOr4/sendMessage"
+
+    payload = {
+        "chat_id": "-4159820605",
+        "text": message
+    }
+    response = requests.post(url, data=payload)