Update main.py

main.py

@@ -4,9 +4,16 @@ import random
 
 from flask import Flask, request, jsonify, redirect, url_for
 from flask_cors import CORS
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
 
 client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1")
 
+connection_string = "postgresql://data_owner:PFAnX9oJp4wV@ep-green-heart-a78sxj65.ap-southeast-2.aws.neon.tech/figurecircle?sslmode=require"
+
+engine = create_engine(connection_string)
+Session = sessionmaker(bind=engine)
+
 app = Flask(__name__)
 CORS(app)
     
@@ -109,7 +116,19 @@ def get_mentor():
     user_stream = content.get('stream')
     #user_semester = content.get('semester')
     courses = content.get('courses')
-    mentors_data= content.get('mentors_data')
+
+    session = Session()
+
+    # Query verified mentors
+    verified_mentors = session.query(Mentor).filter_by(verified=True).all()
+
+    mentor_list = [{"id": mentor.id, "mentor_name": mentor.mentor_name, "skills": mentor.skills,
+                    "qualification": mentor.qualification, "experience": mentor.experience,
+                    "verified": mentor.verified} for mentor in verified_mentors]
+
+    session.close()
+    
+    mentors_data= mentor_list
 
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -134,7 +153,6 @@ def get_mentor():
     Based on above details recommend the mentor that realtes to above details 
     Note: Output should be list in below format:
     [mentor1,mentor2,mentor3,...]
-    Return only answer which is in  [] brackets not prompt and unnecessary stuff, also dont add any special characters or punctuation marks
     """
     formatted_prompt = format_prompt(prompt)