Update app.py

app.py

@@ -1,24 +1,26 @@
-from chatbot import Comsatsbot
 import uuid
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
-from langchain_groq import ChatGroq
-from langchain_community.embeddings import HuggingFaceBgeEmbeddings
-import os
 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'
@@ -26,20 +28,25 @@ 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, client, paths)
-
 
+chatbot = Comsatsbot(hf, llm, api_key, chats_collection, paths)
 
 # Endpoint for creating a new chat ID
-@app.post("/new_chat")
-def new_chat():
+@app.post("/get_new_chat")
+def create_new_chat():
     try:
         chat_id = str(uuid.uuid4())
-        message = chatbot.new_chat(id)
+        message = chatbot.new_chat(chat_id)
         return {"chat_id": chat_id, "message": "Successfully created new chat."}
-    except HTTPException as e:
-        raise e
+    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
@@ -49,28 +56,37 @@ class ChatRequest(BaseModel):
 @app.get("/get_chat/{chat_id}")
 def get_chat(chat_id: str):
     try:
-        chat_history = chatbot.load_chat(chat_id)
-        return {"chat_id": chat_id, "history": chat_history}
-    except HTTPException as e:
-        raise e  # Raise if chat ID is not found
+        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": 'Successfully deleted chat.'}
-    except HTTPException as e:
-        raise e  # Raise if chat ID is not found
+        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
-    answer = chatbot.response(question, chat_id)
 
     try:
+        answer = chatbot.response(question, chat_id)
+
         return {"answer": answer}
-    except HTTPException as e:
-        raise e
+    except KeyError:
+        raise HTTPException(status_code=404, detail="Chat ID not found")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))