Update main.py

main.py

@@ -20,9 +20,15 @@ class Mentor(Base):
 
     id = Column(Integer, primary_key=True)
     mentor_name = Column(String)
-    skills = Column(String)
-    qualification = Column(String)
-    experience = Column(String)
+    profile_photo = Column(LargeBinary)  # Store mentor's profile photo as binary data
+    description = Column(String)
+    highest_degree = Column(String)
+    expertise = Column(String)
+    recent_project = Column(String)
+    meeting_time = Column(String)
+    fees = Column(String)
+    stream = Column(String)
+    country = Column(String)
     verified = Column(Boolean, default=False)
 
 engine = create_engine(connection_string)
@@ -130,9 +136,23 @@ def get_mentor():
     # 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]
+    mentor_list = []
+    for mentor in verified_mentors:
+        mentor_info = {
+            "id": mentor.id,
+            "mentor_name": mentor.mentor_name,
+            "profile_photo": mentor.profile_photo.decode('utf-8'),  # Decode binary photo to string
+            "description": mentor.description,
+            "highest_degree": mentor.highest_degree,
+            "expertise": mentor.expertise,
+            "recent_project": mentor.recent_project,
+            "meeting_time": mentor.meeting_time,
+            "fees": mentor.fees,
+            "stream": mentor.stream,
+            "country": mentor.country,
+            "verified": mentor.verified
+        }
+        mentor_list.append(mentor_info)
 
     session.close()
     
@@ -261,6 +281,37 @@ def get_certificate():
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False)
     return jsonify({"ans": stream})
 
+@app.route('/get_three_streams', methods=['POST'])
+def get_three_streams():
+    temperature = 0.9
+    max_new_tokens = 256
+    top_p = 0.95
+    repetition_penalty = 1.0
+
+    content = request.json
+    user_degree = content.get('degree')  # Uncomment this line
+    
+
+    generate_kwargs = dict(
+        temperature=temperature,
+        max_new_tokens=max_new_tokens,
+        top_p=top_p,
+        repetition_penalty=repetition_penalty,
+        do_sample=True,
+        seed=42,
+    )
+    prompt = f""" prompt: 
+    You need to act like as recommendation engine for stream recommendation for a student based on user degree. Below are details.
+    Degree: {user_degree}
+    Based on above degree details recommend only 3 the streams 
+    Note: Output should be list in below format:
+    [stream1, stream2, stream2]
+    Return only answer not prompt and unnecessary stuff, also dont add any special characters or punctuation marks
+    """
+    formatted_prompt = format_prompt(prompt)
+
+    stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False)
+    return jsonify({"ans": stream})
 
 @app.route('/get_competition', methods=['POST'])
 def get_competition():