[UPDATE] api

auth.py

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

client.py

@@ -1,23 +1,63 @@
 from openai import AzureOpenAI
 import os
+import model
 
-client = AzureOpenAI(
-    api_key=os.environ.get('OPENAI_API_KEY'),  
-    api_version=os.environ.get('MODEL_VERSION'),
-    azure_endpoint = os.environ.get('END_POINT')
+def create_client():
+    print("In create client")
+    client = AzureOpenAI(
+        api_key=os.environ.get('OPENAI_API_KEY'),  
+        api_version=os.environ.get('MODEL_VERSION'),
+        azure_endpoint = os.environ.get('END_POINT')
     )
+    return client
 
-prompt = 'Write a tagline for an ice cream shop.'
+def get_completion(ticket:model.Ticket):
+    print("In get completion")
+    client = create_client()
+    prompt = create_prompt(ticket.subject, ticket.description)
+    response = client.completions.create(
+        model=os.environ.get('MODEL_NAME'),
+        prompt=prompt,
+        max_tokens=100,
+        n=1,
+        stop=None,
+        temperature=0.7
+        )
+    result = response.choices[0].text.strip()
+    print(result)
+    return result
 
-response = client.completions.create(
-    model=os.environ.get('MODEL_NAME'),
-    prompt=prompt,
-    max_tokens=50,
-    n=1,
-    stop=None,
-    temperature=0.7
-)
+def create_prompt(subject:str, desc:str):
+    print("In create Prompt")
+    # add booking data here
+    data = ""
+    prompt = '''
+        You are an AI Customer Service Agent for booking.com. Please use the below instructions to answer.
+        What is the intent of this request ```
+        Subject: {{subject}} Description: {{desc}}``` in 1 single word? Use these options for intent : 
 
-result = response.choices[0].text.strip()
+        Complaint
+        Information Request
+        Service Request
+        Cancellation
+        Change
+        Refund Request
 
-print(result)
+        In the context of the previous request, 
+        If the intent is Information Request, use the ``` Booking Details: {{data}}``` to provide information as part of resolution response. 
+        otherwise Suggest a resolution response to address the customer query from the following actions list:
+
+        Issue Refund
+        Cancellation
+        Change in booking requested
+
+        Finally, Summarize the message: ```
+        Subject: {{subject}}Description:{{desc}}```  and generate a mail content for the customer about the whole conversation.
+
+        Give the final result like this: 
+        intent= {your intent}
+        suggested_response= {your response}
+        summary= {your summary}
+        mail={generated_mail}
+        '''
+    return prompt

model.py

@@ -0,0 +1,9 @@
+from pydantic import BaseModel
+
+class Ticket(BaseModel):
+    subject: str
+    description: str
+
+class TicketResponse(BaseModel):
+    subject: str
+    description: str

requirements.txt

@@ -1,3 +1,4 @@
 fastapi==0.99.1
 uvicorn
-openai
+openai
+pydantic

server.py

@@ -1,6 +1,8 @@
 from fastapi import FastAPI, Depends
 import os
 import auth
+import client
+import model
 
 app = FastAPI()
 
@@ -8,9 +10,11 @@ app = FastAPI()
 async def welcome():
     return "Hello, Welcome to AI space!"
 
-@app.get("/generate")
+@app.get("/test")
 async def generate(api_key = Depends(auth.api_key_auth)):
-    # add the logic to call gpt here
-    os.environ.get('test')
-    response = os.environ.get('test') 
-    return "All good. You only get this message if you're authenticated "+ response
+    return "All good. You only get this message if you're authenticated"
+
+@app.post("/generate")
+async def generate(ticket: model.Ticket, api_key = Depends(auth.api_key_auth)):
+    response = client.get_completion(ticket)
+    return response