Spaces:
Sleeping
Sleeping
File size: 3,274 Bytes
a2b6bfb |
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 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
import uuid
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
from pymongo import MongoClient
import urllib.parse
import os
from langchain_groq import ChatGroq
from langchain_community.embeddings import HuggingFaceBgeEmbeddings
from chatbot import Comsatsbot
# FastAPI app setup
app = FastAPI()
model_name = "BAAI/bge-small-en"
model_kwargs = {"device": "cpu"}
encode_kwargs = {"normalize_embeddings": True}
hf = HuggingFaceBgeEmbeddings(
model_name=model_name, model_kwargs=model_kwargs, encode_kwargs=encode_kwargs
)
os.environ['GROQ_API_KEY'] = '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'
api_key = os.environ.get("GROQ_API_KEY")
llm = ChatGroq(temperature=0, groq_api_key=api_key, model_name="llama3-70b-8192")
# MongoDB setup
username = 'hasnainnaseer987'
password = 'Hasnain'
encoded_username = urllib.parse.quote_plus(username)
encoded_password = urllib.parse.quote_plus(password)
MONGODB_ATLAS_CLUSTER_URI = f'mongodb+srv://{encoded_username}:{encoded_password}@cluster0.jdfp3.mongodb.net/'
client = MongoClient(MONGODB_ATLAS_CLUSTER_URI)
db = client.get_database('chat_db') # Assume this is your database
chats_collection = db.get_collection('chats') # Collection to store chats
paths = ['/content/FYP Supervisor Feedback.csv', '/content/urdu_data.csv', '/content/english_data.csv']
chatbot = Comsatsbot(hf, llm, api_key, chats_collection, paths)
# Endpoint for creating a new chat ID
@app.post("/get_new_chat")
def create_new_chat():
try:
chat_id = str(uuid.uuid4())
message = chatbot.new_chat(chat_id)
return {"chat_id": chat_id, "message": "Successfully created new chat."}
except KeyError:
raise HTTPException(status_code=404, detail="Chat ID already exist try again plz...")
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
# Request model for response endpoint
class ChatRequest(BaseModel):
chat_id: str
question: str
# Endpoint for retrieving chat history by chat ID
@app.get("/get_chat/{chat_id}")
def get_chat(chat_id: str):
try:
history = chatbot.load_chat(chat_id)
return {"chat_id": chat_id, "history": history}
except KeyError:
raise HTTPException(status_code=404, detail="Chat ID not found")
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
# Endpoint for deleting a chat by chat ID
@app.delete("/delete_chat/{chat_id}")
def delete_chat(chat_id: str):
try:
message = chatbot.delete_chat(chat_id)
return {"message": f"Chat with ID {chat_id} has been deleted successfully."}
except KeyError:
raise HTTPException(status_code=404, detail="Chat ID not found")
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
# Endpoint for getting a response based on chat ID and question
@app.post("/response")
def response(request: ChatRequest):
chat_id = request.chat_id
question = request.question
try:
answer = chatbot.response(question, chat_id)
return {"answer": answer}
except KeyError:
raise HTTPException(status_code=404, detail="Chat ID not found")
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
|