[ADDED] api with api key auth

.gitignore

@@ -0,0 +1,2 @@
+__pycache__/
+

Dockerfile

@@ -0,0 +1,9 @@
+FROM python:3.11.0
+
+COPY . .
+
+WORKDIR /
+
+RUN pip install --no-cache-dir --upgrade -r /requirements.txt
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

auth.py

@@ -0,0 +1,9 @@
+from fastapi.security.api_key import APIKeyHeader
+from fastapi import Security, HTTPException
+
+API_Keys = ["abc"]
+api_key_header = APIKeyHeader(name="x-api-key", auto_error=False)
+
+async def api_key_auth(api_key: str = Security(api_key_header)):
+    if api_key not in API_Keys:
+        raise HTTPException(status_code=401, detail="Missing or invalid API key")

client.py

@@ -0,0 +1,23 @@
+from openai import AzureOpenAI
+import constants
+
+client = AzureOpenAI(
+    api_key=constants.AZURE_OPENAI_API_KEY,  
+    api_version=constants.API_VERSION,
+    azure_endpoint = constants.AZURE_OPENAI_ENDPOINT
+    )
+
+prompt = 'Write a tagline for an ice cream shop.'
+
+response = client.completions.create(
+    model='gpt-35-turbo-instruct',
+    prompt=prompt,
+    max_tokens=50,
+    n=1,
+    stop=None,
+    temperature=0.7
+)
+
+result = response.choices[0].text.strip()
+
+print(result)

constants.py

@@ -0,0 +1,3 @@
+AZURE_OPENAI_API_KEY=''
+AZURE_OPENAI_ENDPOINT=''
+API_VERSION='2024-02-01'

requirements.txt

@@ -0,0 +1,3 @@
+fastapi==0.99.1
+uvicorn
+openai

server.py

@@ -0,0 +1,13 @@
+from fastapi import FastAPI, Depends
+import auth
+
+app = FastAPI()
+
+@app.get("/")
+async def welcome():
+    return "Hello, Welcome to AI space!"
+
+@app.get("/generate")
+async def generate(api_key = Depends(auth.api_key_auth)):
+    # add the logic to call gpt here
+    return "All good. You only get this message if you're authenticated"