Update student_functions.py

student_functions.py

@@ -11,8 +11,11 @@ from youtube_transcript_api.formatters import JSONFormatter
 from urllib.parse import urlparse, parse_qs
 from pypdf import PdfReader
 from ai71 import AI71
+from mistralai import Mistral
 import os
-
+API_KEY = 'xQ2Zhfsp4cLar4lvBRDWZKljvp0Ej427'
+MODEL = "mistral-large-latest"
+client = Mistral(api_key=API_KEY)
 AI71_API_KEY = "api71-api-652e5c6c-8edf-41d0-9c34-28522b07bef9"
 
 
@@ -28,26 +31,24 @@ def extract_text_from_pdf_s(pdf_path):
 
 def generate_response_from_pdf(query, pdf_text):
     response = ''
-    for chunk in AI71(AI71_API_KEY).chat.completions.create(
-            model="tiiuae/falcon-180b-chat",
+    chat_response = client.chat.complete(
+            model=MODEL,
             messages=[
                 {"role": "system", "content": "You are a pdf questioning assistant."},
                 {"role": "user",
                  "content": f'''Answer the querry based on the given content.Content:{pdf_text},query:{query}'''},
-            ],
-            stream=True,
-    ):
-        if chunk.choices[0].delta.content:
-            response += chunk.choices[0].delta.content
-    return response.replace("###", '')
+            ]
+        )
+    response_content = chat_response.choices[0].message.content
+    return jsonify({"response": response_content})
+    
 
 
 def generate_quiz(subject, topic, count, difficult):
     quiz_output = ""
-
-    for chunk in AI71(AI71_API_KEY).chat.completions.create(
-            model="tiiuae/falcon-180b-chat",
-            messages=[
+    chat_response = client.chat.complete(
+            model=MODEL,
+            messages=messages=[
                 {"role": "system", "content": "You are a teaching assistant."},
                 {"role": "user",
                  "content": f'''Generate {count} multiple-choice questions in the subject of {subject} for the topic {topic} for students at a {difficult} level. Ensure the questions are well-diversified and cover various aspects of the topic. Format the questions as follows:
@@ -57,93 +58,86 @@ Question: [Question text] [specific concept in a question]
 <<o>> [Option3] 
 <<o>> [Option4], 
 Answer: [Option number]'''},
-            ],
-            stream=True,
-    ):
-        if chunk.choices[0].delta.content:
-            quiz_output += chunk.choices[0].delta.content
-    print("Quiz generated")
-    return quiz_output
+            ]
+        )
+    response_content = chat_response.choices[0].message.content
+    return jsonify({"response": response_content})
+    
 
 
 
 def generate_ai_response(query):
     ai_response = ''
-    for chunk in AI71(AI71_API_KEY).chat.completions.create(
-            model="tiiuae/falcon-180b-chat",
-            messages=[
+    chat_response = client.chat.complete(
+            model=MODEL,
+            messages=messages=[
                 {"role": "system", "content": "You are a teaching assistant."},
                 {"role": "user", "content": f'Assist the user clearly for his questions: {query}.'},
-            ],
-            stream=True,
-    ):
-        if chunk.choices[0].delta.content:
-            ai_response += chunk.choices[0].delta.content
-    return ai_response.replace('###', '')
+            ]
+        )
+    response_content = chat_response.choices[0].message.content
+    return jsonify({"response": response_content})
+    
 
 
 def generate_project_idea(subject, topic, overview):
     string = ''
-    for chunk in AI71(AI71_API_KEY).chat.completions.create(
-            model="tiiuae/falcon-180b-chat",
-            messages=[
+    chat_response = client.chat.complete(
+            model=MODEL,
+            messages=messages=[
                 {"role": "system", "content": "You are a project building assistant."},
                 {"role": "user",
                  "content": f'''Give the different project ideas to build project in {subject} specifically in {topic} for school students. {overview}.'''},
-            ],
-            stream=True,
-    ):
-        if chunk.choices[0].delta.content:
-            string += chunk.choices[0].delta.content
-    return string
+            ]
+        )
+    response_content = chat_response.choices[0].message.content
+    return jsonify({"response": response_content})
 
+   
 
 def generate_project_idea_questions(project_idea, query):
     project_idea_answer = ''
-    for chunk in AI71(AI71_API_KEY).chat.completions.create(
-            model="tiiuae/falcon-180b-chat",
+    chat_response = client.chat.complete(
+            model=MODEL,
             messages=[
                 {"role": "system", "content": "You are a project building assistant."},
                 {"role": "user",
                  "content": f'''Assist me clearly for the following question for the given idea. Idea: {project_idea}. Question: {query}'''},
-            ],
-            stream=True,
-    ):
-        if chunk.choices[0].delta.content:
-            project_idea_answer += chunk.choices[0].delta.content
-    return project_idea_answer
+            ]
+        )
+    response_content = chat_response.choices[0].message.content
+    return jsonify({"response": response_content})
+    
 
 
 def generate_step_by_step_explanation(query):
     explanation = ''
-    for chunk in AI71(AI71_API_KEY).chat.completions.create(
-            model="tiiuae/falcon-180b-chat",
+    chat_response = client.chat.complete(
+            model=MODEL,
             messages=[
                 {"role": "system", "content": "You are the best teaching assistant."},
                 {"role": "user",
                  "content": f'''Provide me the clear step by step explanation answer for the following question. Question: {query}'''},
-            ],
-            stream=True,
-    ):
-        if chunk.choices[0].delta.content:
-            explanation += chunk.choices[0].delta.content
-    return explanation.replace('###', '')
+            ]
+        )
+    response_content = chat_response.choices[0].message.content
+    return jsonify({"response": response_content})
+    
 
 
 def study_plan(subjects, hours, arealag, goal):
     plan = ''
-    for chunk in AI71(AI71_API_KEY).chat.completions.create(
-            model="tiiuae/falcon-180b-chat",
+    chat_response = client.chat.complete(
+            model=MODEL,
             messages=[
                 {"role": "system", "content": "You are the best teaching assistant."},
                 {"role": "user",
                  "content": f'''Provide me the clear personalised study plan for the subjects {subjects} i lag in areas like {arealag}, im available for {hours} hours per day and my study goal is to {goal}.Provide me like a timetable like day1,day2 for 5 days with concepts,also suggest some books'''},
-            ],
-            stream=True,
-    ):
-        if chunk.choices[0].delta.content:
-            plan += chunk.choices[0].delta.content
-    return plan.replace('\n', '<br>')
+            ]
+        )
+    response_content = chat_response.choices[0].message.content
+    return jsonify({"response": response_content.replace('\n', '<br>')})
+    
 
 
 class ConversationBufferMemory:
@@ -170,18 +164,19 @@ def spk_msg(user_input, memory):
          "content": f"Previous conversation:\n{chat_history}\n\nNew human question: {user_input}\nResponse:"}
     ]
 
-    try:
-        for chunk in AI71(AI71_API_KEY).chat.completions.create(
-                model="tiiuae/falcon-180b-chat",
-                messages=messages,
-                stream=True,
-        ):
-            if chunk.choices[0].delta.content:
-                msg += chunk.choices[0].delta.content
-    except Exception as e:
-        print(f"An error occurred: {e}")
-
-    return msg
+    if 1==1:
+        chat_response = client.chat.complete(
+            model=MODEL,
+            messages=[
+                {"role": "system", "content": "You are a pdf questioning assistant."},
+                {"role": "user",
+                 "content": f'''Answer the querry based on the given content.Content:{pdf_text},query:{query}'''},
+            ]
+        )
+        response_content = chat_response.choices[0].message.content
+        return jsonify({"response": response_content})
+        
+     
 
 
 def get_first_youtube_video_link(query):
@@ -204,17 +199,16 @@ def get_first_youtube_video_link(query):
 
 def content_translate(text):
     translated_content = ''
-    for chunk in AI71(AI71_API_KEY).chat.completions.create(
-            model="tiiuae/falcon-180b-chat",
-            messages=[
+    chat_response = client.chat.complete(
+            model=MODEL,
+            messages=messages=[
                 {"role": "system", "content": "You are the best teaching assistant."},
                 {"role": "user", "content": f'''Translate the text to hindi. Text: {text}'''},
-            ],
-            stream=True,
-    ):
-        if chunk.choices[0].delta.content:
-            translated_content += chunk.choices[0].delta.content
-    return translated_content
+            ]
+        )
+    response_content = chat_response.choices[0].message.content
+    return jsonify({"response": response_content})
+    
 
 
 def get_video_id(url):
@@ -272,18 +266,16 @@ def get_simplified_explanation(text):
     )
 
     response = ""
-    for chunk in AI71(AI71_API_KEY).chat.completions.create(
-            model="tiiuae/falcon-180b-chat",
+    chat_response = client.chat.complete(
+            model=MODEL,
             messages=[
                 {"role": "system", "content": "You are a helpful assistant."},
                 {"role": "user", "content": prompt},
-            ],
-            stream=True,
-    ):
-        if chunk.choices[0].delta.content:
-            response += chunk.choices[0].delta.content
-
-    return response
+            ]
+        )
+    response_content = chat_response.choices[0].message.content
+    return jsonify({"response": response_content})
+    
 
 
 def summarise_text(url):
@@ -316,17 +308,18 @@ def generate_speech_from_pdf(content):
         print(f"Directory {directory} does not exist.")
 
     speech = ''
-    for chunk in AI71(AI71_API_KEY).chat.completions.create(
-            model="tiiuae/falcon-180b-chat",
+    chat_response = client.chat.complete(
+            model=MODEL,
             messages=[
                 {"role": "system", "content": "You are a summarising assistant."},
                 {"role": "user",
                  "content": f'''Summarise the given content for each chapter for 1 sentence.Content={content}'''},
-            ],
-            stream=True,
-    ):
-        if chunk.choices[0].delta.content:
-            speech += chunk.choices[0].delta.content
+            ]
+        )
+    response_content = chat_response.choices[0].message.content
+    
+    if response_content:
+            speech += response_content
     speech = speech[:-6].replace("###", '')
     chapters = speech.split('\n\n')
     pdf_audio(chapters[:4])
@@ -343,14 +336,13 @@ def content_translate(text):
 
 
     translated_content = ''
-    for chunk in AI71(AI71_API_KEY).chat.completions.create(
-        model="tiiuae/falcon-180b-chat",
-        messages=[
+    chat_response = client.chat.complete(
+            model=MODEL,
+            messages=[
             {"role": "system", "content": "You are the best teaching assistant."},
             {"role": "user", "content": f'''Translate the text to hindi. Text: {text}'''},
-        ],
-        stream=True,
-    ):
-        if chunk.choices[0].delta.content:
-            translated_content += chunk.choices[0].delta.content
-    return translated_content
+        ]
+        )
+    response_content = chat_response.choices[0].message.content
+    return jsonify({"response": response_content})
+