initial commit

.gitattributes

@@ -25,7 +25,6 @@
 *.safetensors filter=lfs diff=lfs merge=lfs -text
 saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
 *.tflite filter=lfs diff=lfs merge=lfs -text
 *.tgz filter=lfs diff=lfs merge=lfs -text
 *.wasm filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1 @@
+__pycache__

Dockerfile

@@ -0,0 +1,32 @@
+# Use the official Python 3.9 image
+FROM python:3.9
+
+RUN apt update && apt-get install -y ffmpeg apt-utils build-essential gcc && rm -rf /var/lib/apt/lists/*
+ 
+# Set the working directory to /code
+WORKDIR /code
+ 
+# Copy the current directory contents into the container at /code
+COPY ./requirements.txt /code/requirements.txt
+ 
+# Install requirements.txt 
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+ 
+# Set up a new user named "user" with user ID 1000
+RUN useradd -m -u 1000 user
+# Switch to the "user" user
+USER user
+# Set home to the user's home directory
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+ 
+# Set the working directory to the user's home directory
+WORKDIR $HOME/app
+ 
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+COPY --chown=user . $HOME/app
+ 
+# Start the FastAPI app on port 7860, the default port expected by Spaces
+# CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+
+CMD gunicorn -k uvicorn.workers.UvicornWorker --workers 2 --threads=2 --max-requests 512 --bind 0.0.0.0:7860 app:app

README.md

@@ -1,10 +1,64 @@
 ---
-title: Transcription Api
+title: Chatapi
 emoji: 🔥
 colorFrom: purple
-colorTo: gray
+colorTo: purple
 sdk: docker
 pinned: false
+license: apache-2.0
+duplicated_from: Subhraj07/chatapi
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+## How to Use the API using python
+
+### Create a Token
+```
+import requests
+
+headers = {
+    'accept': 'application/json',
+    'Content-Type': 'application/json',
+}
+
+json_data = {
+    "ref_key": "F0eeQ419wvoCSPH7KBmsd9OiVkF0W0DxK1XE9T3BlbkFJ0",
+    "expiry_date": "2023-06-15"
+}
+
+response = requests.post('https://subhraj07-chatapi.hf.space/create', headers=headers, json=json_data)
+token = response.json()
+```
+### calling open API (text-davinci-002/gpt-3.5-turbo)
+```
+import requests
+
+headers = {
+    'accept': 'application/json',
+    'Content-Type': 'application/json',
+    'Authorization': 'Bearer TOKEN',
+}
+
+json_data = {
+    'data': []
+}
+
+response = requests.post('https://subhraj07-chatapi.hf.space/transcribe', headers=headers, json=json_data)
+if response.status_code == 200:
+    print(response.json())
+```
+
+### Check list of tokens
+```
+import requests
+
+headers = {
+    'accept': 'application/json',
+    'Content-Type': 'application/json',
+    'Authorization': 'Bearer TOKEN
+}
+
+response = requests.get('https://subhraj07-chatapi.hf.space/list', headers=headers)
+print(response.json())
+
+```
+

app.py

@@ -0,0 +1,96 @@
+import openai
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+import pydantic
+from fastapi import FastAPI, HTTPException, Depends, Request, Response
+from fastapi.security import OAuth2PasswordBearer
+from tinydb import TinyDB
+from tinydb import Query
+from datetime import datetime
+from utils import generate_token
+import whisper
+import numpy as np
+from typing import List
+
+model = whisper.load_model("medium")
+# model = whisper.load_model("base")
+
+query = Query()
+db = TinyDB(".token.json")
+
+app = FastAPI()
+
+origins = ["*"]
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
+
+REF_KEY = "F0eeQ419wvoCSPH7KBmsd9OiVkF0W0DxK1XE9T3BlbkFJ0"
+model_name = ""
+
+def verify_token(token: str = Depends(oauth2_scheme)):
+    expiry = -1
+    res = db.get(query.token == token)
+
+    if res:
+        expiry = (datetime.strptime(res["expiry_date"], '%Y-%m-%d') - datetime.strptime(res["created_at"], '%Y-%m-%d')).days
+    
+    if expiry < 0:
+        return {"message": "Token is not Valid"}
+    
+    return token
+
+
+class AudioInput(BaseModel):
+    data: List
+
+class RefToken(BaseModel):
+    expiry_date: str
+    ref_key: str
+
+@app.get("/")
+async def base_url():
+    try:
+        return {
+            "Please Check the documentation here": "https://huggingface.co/spaces/Subhraj07/chatapi/blob/main/README.md",
+            "Swagger UI" : "https://subhraj07-chatapi.hf.space/docs"
+            }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail="An error occurred while processing the request." + str(e))
+
+@app.post("/create")
+async def create(data: RefToken):
+    token = "Reference Key is incorrect"
+    try:
+        if data.ref_key == REF_KEY:
+            token = generate_token(data.expiry_date, db) 
+            return {"TOKEN": token}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail="An error occurred while processing the request." + str(e))
+
+@app.get("/list")
+async def list(token: str = Depends(verify_token)):
+    try:
+        data = db.all()
+        return {"data": data}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail="An error occurred while processing the request." + str(e))
+
+@app.post("/transcribe")
+async def chat(audio_input: AudioInput, token: str = Depends(verify_token)):
+    try:
+        transcription = model.transcribe(np.asarray(audio_input))
+
+        return {
+            "transcript": transcription["text"]
+            }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail="An error occurred while processing the request." + str(e))

requirements.txt

@@ -0,0 +1,8 @@
+fastapi==0.95.1
+uvicorn==0.22.0
+openai==0.27.6
+tinydb==4.7.1
+gunicorn==20.1.0
+uvloop==0.15.2
+httptools==0.2.0
+git+https://github.com/openai/whisper.git

utils.py

@@ -0,0 +1,22 @@
+from datetime import datetime, timedelta, date
+import secrets
+import random
+import string
+
+
+def generate_token(expiry_date, db):
+    alphabet = string.ascii_letters  # Get a string containing all the alphabets (both upper and lower case)
+    first_char = random.choice(alphabet)  # Randomly choose an alphabet as the first character
+    token = "md-" + first_char + secrets.token_hex(60)
+
+    creation_date = date.today().strftime("%Y-%m-%d")
+
+    token_details = {
+        "created_at": creation_date,
+        "expiry_date" : expiry_date,
+        "token" : token 
+    }
+
+    db.insert(token_details)
+
+    return token