File size: 3,232 Bytes
fa947da |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
from fastapi import FastAPI, HTTPException, UploadFile, File, Security, status
from fastapi.security.api_key import APIKeyHeader
from environs import Env
import json
# Importing modules from both scripts
from coresugg import ConversationPayload as ConversationPayloadSugg, create_conversation_starter_prompt, generate_conversation_starters, NUMBER_OF_MESSAGES_FOR_CONTEXT as NUMBER_OF_MESSAGES_FOR_CONTEXT_SUGG
from corechat import ConversationPayload as ConversationPayloadChat, get_conversation_suggestions, NUMBER_OF_MESSAGES_FOR_CONTEXT as NUMBER_OF_MESSAGES_FOR_CONTEXT_CHAT
# Load environment variables
env = Env()
API_KEY = env.str("API_KEY", "koottu123456abcDEF")
api_key_header = APIKeyHeader(name="X-API-KEY", auto_error=False)
app = FastAPI()
async def get_api_key(api_key_header: str = Security(api_key_header)):
if api_key_header == API_KEY:
return api_key_header
else:
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="Invalid API Key")
# Route from app.py
@app.post("/integration")
def integration(payload: ConversationPayloadSugg, api_key: str = Security(get_api_key)):
from_user_questions = payload.FromUserKavasQuestions[-NUMBER_OF_MESSAGES_FOR_CONTEXT_SUGG:]
to_user_questions = payload.ToUserKavasQuestions[-NUMBER_OF_MESSAGES_FOR_CONTEXT_SUGG:]
ai_prompt = create_conversation_starter_prompt(from_user_questions + to_user_questions, payload.Chatmood)
conversation_starters = generate_conversation_starters(ai_prompt)
return {"conversation_starters": conversation_starters}
# Route from chatt.py
@app.post("/chat_integration")
def chat_integration(payload: ConversationPayloadChat, api_key: str = Security(get_api_key)):
last_chat_messages = payload.LastChatMessages[-NUMBER_OF_MESSAGES_FOR_CONTEXT_CHAT:]
suggestions = get_conversation_suggestions(last_chat_messages)
return {"version": "1.0.0-alpha", "suggested_responses": suggestions}
# Combined upload endpoint
@app.post("/upload")
async def upload_file(file: UploadFile = File(...), api_key: str = Security(get_api_key)):
if file.content_type != 'application/json':
raise HTTPException(status_code=400, detail="Invalid file type. Please upload a JSON file.")
data = await file.read()
try:
json_data = json.loads(data)
except json.JSONDecodeError:
raise HTTPException(status_code=400, detail="Invalid JSON format.")
if "FromUserKavasQuestions" in json_data and "Chatmood" in json_data:
prompt = create_conversation_starter_prompt(
json_data["FromUserKavasQuestions"],
json_data["Chatmood"]
)
starter_suggestion = generate_conversation_starters(prompt)
return {"conversation_starter": starter_suggestion}
elif "LastChatMessages" in json_data:
last_chat_messages = json_data["LastChatMessages"][-NUMBER_OF_MESSAGES_FOR_CONTEXT_CHAT:]
response = {
"version": "1.0.0-alpha",
"suggested_responses": get_conversation_suggestions(last_chat_messages)
}
return response
else:
raise HTTPException(status_code=400, detail="Invalid JSON structure.")
#uvicorn app:app --reload
#uvicorn aisugg22:app --reload
|