Update main.py

main.py

@@ -789,12 +789,11 @@ def roadmap():
             # Get roadmap data
             roadmap_data = roadmap_collection.find_one({"_id": ObjectId(data.get("_id")), "userId": user_id})
             if roadmap_data:
-                return jsonify({'success': True, 'roadmapData': {'data': roadmap_data['data']}, 'activeDays': roadmap_data['activeDays']})
+                return jsonify({'success': True, 'roadmapData': {'data': roadmap_data['data']}, 'activeDays': roadmap_data['activeDays'], 'practice': roadmap_data['practice'], 'title': roadmap_data['title']})
             else:
                 return jsonify({'success': False})
             
 
-
 @app.route('/roadmap/history', methods=['GET'])
 @auth_user
 def roadmapGetHistory():
@@ -812,7 +811,6 @@ def roadmapGetHistory():
         return jsonify({"error": "An error occurred"}), 500
 
 
-
 @app.route('/roadmapmodder', methods=['POST'])
 @auth_user
 def roadmapmodder():
@@ -959,6 +957,125 @@ def roadmapmodder():
         return jsonify({'success': True})
 
 
+@app.route('/problemgenerator', methods=['POST'])
+def problemgenerator():
+    if request.method == 'POST':
+        data = request.get_json()
+        user_id = data.get('user_id')
+        obj_id = data.get('_id')
+        week_day = int(data.get('weekDay'))
+
+        if data.get('type') == 1:
+            query = data.get('query') 
+
+            prompt = '''
+                Generate 5 Multiple Choice Questions with single correct answers and 4 options in the following format:
+                [{"question": "What is that?", "options": {"1": "Option1", "2": "Option2", "3": "Option3", "4": "Option4"}, "answer" : {"3": "Option3"}, "explaination": "This is explaination."}, ... , {"question": "What is that?", "options": {"0": "Option1", "1": "Option2", "3": "Option3", "4": "Option4"}, "answer" : {"3": "Option3"}, "explaination": "This is explaination."}]
+
+                - Topic for questions is : ''' + query + '''
+                - It is required that the response should be an list of objects. Don't include any other verbose explanations and don't include the markdown syntax anywhere.
+            '''
+
+            response = model.generate_content(prompt)
+
+            if response.text[0] == '[':
+                retro = json.loads(response.text)
+                for ele in retro:
+                    ele["solved"] = False
+                
+                roadmap_collection.update_one(
+                    {"_id": ObjectId(obj_id)},
+                    {"$set": { f"practice.{str(week_day)}": retro }}
+                )
+                
+                problems_data = roadmap_collection.find_one({"_id": ObjectId(data.get("_id")), "userId": user_id}, {"_id": 1, "practice": 1})
+                
+                return jsonify({"success": True, "practice": problems_data['practice'] }), 200
+            
+            elif response.text[0] == '`':
+                retro = json.loads(response.text[7:-3])
+                for ele in retro:
+                    ele["solved"] = False
+                
+                roadmap_collection.update_one(
+                    {"_id": ObjectId(obj_id)},
+                    {"$set": {  f"practice.{str(week_day)}": retro }}
+                )
+
+                problems_data = roadmap_collection.find_one({"_id": ObjectId(data.get("_id")), "userId": user_id}, {"_id": 1, "practice": 1})
+                
+                return jsonify({"success": True, "practice": problems_data['practice'] }), 200
+            
+            print(response.text)
+            return jsonify({"success": False}), 500
+
+        elif data.get('type') == 2:
+            query = data.get('query') 
+
+            prompt = '''
+                Generate 5 Multiple Choice Questions with single correct answers and 4 options in the following format:
+                [{"question": "What is that?", "options": {"1": "Option1", "2": "Option2", "3": "Option3", "4": "Option4"}, "answer" : {"3": "Option3"}, "explaination": "This is explaination."}, ... , {"question": "What is that?", "options": {"0": "Option1", "1": "Option2", "3": "Option3", "4": "Option4"}, "answer" : {"3": "Option3"}, "explaination": "This is explaination."}]
+
+                - Topic for questions is : ''' + query + '''
+                - It is required that the response should be an list of objects. Don't include any other verbose explanations and don't include the markdown syntax anywhere.
+            '''
+
+            response = model.generate_content(prompt)
+            
+            if response.text[0] == '[':
+                temp_data = roadmap_collection.find_one({"_id": ObjectId(obj_id)})
+                retro = json.loads(response.text)
+                for ele in retro:
+                    ele["solved"] = False
+                
+                temp_data['practice'][str(week_day)] += retro
+                roadmap_collection.update_one(
+                    {"_id": ObjectId(obj_id)},
+                    {"$set": { "practice": temp_data['practice'] }}
+                )
+
+                problems_data = roadmap_collection.find_one({"_id": ObjectId(data.get("_id")), "userId": user_id}, {"_id": 1, "practice": 1})
+                
+                return jsonify({"success": True, "practice": problems_data['practice'] }), 200
+            
+            elif response.text[0] == '`':
+                temp_data = roadmap_collection.find_one({"_id": ObjectId(obj_id)})
+                retro = json.loads(response.text[7:-3])
+                for ele in retro:
+                    ele["solved"] = False
+                
+                temp_data['practice'][str(week_day)] += retro
+                roadmap_collection.update_one(
+                    {"_id": ObjectId(obj_id)},
+                    {"$set": { "practice": temp_data['practice'] }}
+                )
+
+                problems_data = roadmap_collection.find_one({"_id": ObjectId(data.get("_id")), "userId": user_id}, {"_id": 1, "practice": 1})
+                
+                return jsonify({"success": True, "practice": problems_data['practice'] }), 200
+            
+            print(response.text)
+            return jsonify({"success": False}), 500
+
+
+@app.route('/problemhandler', methods=['POST'])
+def problemhandler():
+    if request.method == 'POST':
+        data = request.get_json()
+        obj_id = data.get('_id')
+        week_day = data.get('weekDay')
+        indexer = data.get('index')
+
+        temp = f"practice.{week_day}.{indexer}.solved"
+        print(temp)
+
+        test = roadmap_collection.update_one(
+            {"_id": ObjectId(obj_id)},
+            {"$set": { temp : True }}
+        )
+        print(test)
+        return jsonify({"success": True}), 200
+
 
 if __name__ == '__main__':
     app.run(debug=True, host='0.0.0.0')