Update main.py

main.py

@@ -11,14 +11,14 @@ from flask_jwt_extended import JWTManager, create_access_token
 from middleware.authUser import auth_user
 from datetime import timedelta 
 from controllers.demo import get_initial_data
-from controllers.mindmap import saveMindmap, getMindmap, getMindmapHistory, deleteMindmap, getMindmapByid, saveMindmapById
+from controllers.mindmap import saveMindmap, getMindmap, getMindmapHistory, deleteMindmap, getMindmapByid, saveMindmapById, saveGeneratedData
 import json
-from bson import ObjectId
+from bson import ObjectId, json_util
 from dotenv import load_dotenv
 import os
 import re
 import requests
-
+import time
 
 load_dotenv()
 
@@ -26,9 +26,10 @@ app = Flask(__name__)
 
 bcrypt = Bcrypt(app)
 jwt = JWTManager(app)
-CORS(app)
+
 
 app.config['JWT_SECRET_KEY'] = os.getenv('JWT_SECRET')
+CORS(app,resources={r"/*":{"origins":"*"}})
 
 # MongoDB configuration
 username = urllib.parse.quote_plus(os.getenv('MONGO_USERNAME'))
@@ -55,7 +56,36 @@ except Exception as e:
 GOOGLE_API_KEY=os.getenv('GOOGLE_API_KEY')
 
 genai.configure(api_key=GOOGLE_API_KEY)
-model = genai.GenerativeModel('gemini-pro')
+
+generation_config = {
+  "temperature": 0.8,
+  "top_p": 1,
+  "top_k": 40,
+  "max_output_tokens": 2048,
+}
+
+safety_settings = [
+  {
+    "category": "HARM_CATEGORY_HARASSMENT",
+    "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+  },
+  {
+    "category": "HARM_CATEGORY_HATE_SPEECH",
+    "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+  },
+  {
+    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+    "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+  },
+  {
+    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+    "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+  },
+]
+
+model = genai.GenerativeModel('gemini-1.0-pro',
+                              generation_config=generation_config,
+                              safety_settings=safety_settings)
 
 # Caches to reduce no of queries to MongoDB...
 user_id_ping = {'current': 0}
@@ -69,7 +99,6 @@ def index():
 def index2():
     return "routes checking..."
 
-
 #generative model routes
 @app.route('/tree', methods=["POST", "GET"])
 def tree():
@@ -165,6 +194,7 @@ def tree():
         # return temp 
 
 @app.route('/tree/demo', methods=["POST"])
+@auth_user
 def treeDemo():
     if request.method == 'POST':
         data = request.get_json()
@@ -224,28 +254,36 @@ Topic is: ''' + query)
         # return temp 
 
 @app.route('/mindmap/generate/data', methods=["POST"])
+# @auth_user
 def generateData():
     if request.method == 'POST':
         data = request.get_json()
         topic = data.get('topic')
         description = data.get('description')
         category = data.get('category')
-        print(topic)
-        print(description)
-        print(category)
-        response = model.generate_content('''I will provide you a topic. You need to generate its detailed explanation also provide its links to the related topics and subtopics and respond with html format with tailiwindcss styling, wrapped under a single<div></div>.
-                            
-        - keep the background tranparent.
-        - Title should be in text-2xl font-bold.
-        - Subtitle should be in text-lg font-semibold.
-        - make sure the content is well structured and easy to read.
-        - make sure the link text style is in text-aqua.    
-        Topic: ''' + topic)                                 
+        nodeId = data.get('nodeId')
+        mapId = data.get('mapId')
+        # userId = request.userId
+
+        # try: 
+        #     mindMap = savedMindmap.find_one({"_id": ObjectId(mapId)})
+            
+        #     if mindMap["userId"] != userId:
+        #         return jsonify({"message": "Unauthorized"}), 401
+            
+        # except Exception as e:
+        #     print(e) 
+        #     return jsonify({"error": "An error occurred"}), 500
+
+        response = model.generate_content('''I will provide you a topic & related description. You need to generate its detailed explanation, also provide its links to the related topics and respond in markdown. 
+        - make sure the content is well structured and easy to read.  
+        Topic: ''' + topic + '\n\n' + '''Description: ''' + description)                                 
         print(response.text)
         json_data = response.text
-        modified_json_data = json_data[8:-3]
 
-        return jsonify({'success': True, 'data': modified_json_data})                                  
+        saveGeneratedData(mapId, nodeId, json_data, savedMindmap)
+
+        return jsonify({'success': True, 'data': json_data})                                  
 
 def res(user_id):
     avg_text = 0        
@@ -274,7 +312,6 @@ def res(user_id):
 
     return True
 
-
 @app.route('/interview', methods=["POST", "GET"])
 def interview():
     if request.method == 'POST':
@@ -598,126 +635,155 @@ def mindmapDelete():
     data = request.json
     return deleteMindmap(userId, data, savedMindmap)
 
-
 @app.route('/mindmap/demo', methods=['POST'])
 def mindmapDemo():
     data = request.json
     print(data)
     return get_initial_data(), 200
 
+
 @app.route('/roadmap', methods=['POST'])
+# @auth_user
 def roadmap():
     if request.method == 'POST':
         data = request.get_json()
         user_id = data.get('user_id')
-        day_data = data.get('todayData')
 
         if data.get('type') == 1:
+            day_data = data.get('todayData')
+            position = data.get('position')
+            time_road = data.get('time')
+            company = data.get('company')
+
             # Generate New Roadmap
-            response = model.generate_content('''Generate a roadmap for Web developement in 2 months  in following JSON format:
-[
- {
-  "week1": {
-   "title": "...",
-   "description": "What will i learn in this week...",
-   "data": {
-    "day1": {
-     "heading": "....",
-     "description": "What will i learn on this day...",
-     "links": [
-      "link1",
-      "link2",
-      ....
-     ]
-    },
-    "day2": {
-     "description": "...",
-     "heading": "....",
-     "links": [
-      "link1",
-      "link2",
-      ....
-     ]
-    },
-    ...
-   },
-   {
-    "week2": {
-     "title": "...",
-     "description": "What will i learn in this week...",
-     "data": {
-      "day1": {
-       "heading": "....",
-       "description": "What will i learn on this day...",
-       "links": [
-        "link1",
-        "link2",
-        ....
-       ]
-      },
-      "day2": {
-       "description": "...",
-       "heading": "....",
-       "links": [
-        "link1",
-        "link2",
-        ....
-       ]
-      },
-      ...
-    continue for other weeks...
-  ]
-Generate topics for all 7 days in a week and reference links only links(no header) accordingly.
-''')			# Prompt
+            chat_starter = '''Generate a week-wise roadmap for '''+ position + ''' in English in the following JSON format:
+            {
+                "week1": {
+                    "title": "...",
+                    "description": "What will i learn in this week...",
+                    "data": {
+                        "day1": {
+                            "description": "...",
+                            "heading": "What will I do on this day...",
+                            "links": [ "link1", "link2", .... ]
+                        },
+                        "day2": {
+                            "description": "...",
+                            "heading": "What will I do on this day...",
+                            "links": [ "link1", "link2", .... ]
+                        },
+                        "day3": {
+                            "description": "...",
+                            "heading": "What will I do on this day...",
+                            "links": [ "link1", "link2", .... ]
+                        },
+                        "day4": {
+                            "description": "...",
+                            "heading": "What will I do on this day...",
+                            "links": [ "link1", "link2", .... ]
+                        },
+                        "day5": {
+                            "description": "...",
+                            "heading": "What will I do on this day...",
+                            "links": [ "link1", "link2", .... ]
+                        },
+                        "day6": {
+                            "description": "...",
+                            "heading": "What will I do on this day...",
+                            "links": [ "link1", "link2", .... ]
+                        },
+                        "day7": {
+                            "description": "...",
+                            "heading": "What will I do on this day...",
+                            "links": [ "link1", "link2", .... ]
+                        }
+                    }
+                }
+            }
+            - I am preparing to get '''+ position + ''' at ''' + company + '''company.
+            - I will ask you one by one to generate for next weeks. Generate only links, do not add anything to links.
+            - It is required that the response should be an object. Don't include any other verbose explanations and don't include the markdown syntax anywhere.
+            - Roadmap length is of total ''' + time_road + ''' weeks. So adjust the content among the weeks accordingly.'''			# Prompt
             
-            # Youtube Integration & links modification
-            print(response)
+            roadmap_list = []
+
+            chat = model.start_chat(history=[])
+            response = chat.send_message(chat_starter)
+
             print(response.text)
-            roads_data = json.loads(response.text)
+
+            if response.text[0] == '{':
+                roadmap_list.append(json.loads(response.text))
+            elif response.text[0] == '`':
+                roadmap_list.append(json.loads(response.text[7:-3]))
+
+            for i in range(1,int(time_road)):
+                new_response = chat.send_message(f"Now generate for next week{i+1}")
+                print(new_response.text)
+                if new_response.text[0] == '{':
+                    roadmap_list.append(json.loads(new_response.text))
+                elif new_response.text[0] == '`':
+                    roadmap_list.append(json.loads(new_response.text[7:-3]))
+
+            roads_data = roadmap_list
 
             print(roads_data)
-            return jsonify({'success': True})
-            
-            API_KEY = os.environ('YOUTUBE_API_KEY')
+
+            time.sleep(60)
+
+            # Youtube Integration & links modification
+            API_KEY = os.getenv('YOUTUBE_API_KEY')
             url = "https://youtube.googleapis.com/youtube/v3/search"
             
             for index, ele in enumerate(roads_data):
                 for i in range(0,7):
                     params = {
                       "part": "snippet",
-                      "q": ele[f"week{index + 1}"]['data'][f"day{i+1}"]['heading'],
-                      "key": API_KEY
+                      "q": ele[f"week{index + 1}"]['data'][f"day{i+1}"]['heading'] + ele[f"week{index + 1}"]['title'],
+                      "key": API_KEY,
+                      "maxResults" : 3
                     }
                 
-                response = requests.get(url, params=params)
-                
-                if response.status_code == 200:
-                    print(response.json())
-                    video_data = response.json()
-
-                    temp1 = []
-                    for vid in video_data['items']:
-                        if vid['id']['kind']=='youtube#playlist':
-                            temp1.append({'viewed': False, 'type': 'playlist', 'playlistId': vid['id']['playlistId'], 'videoTitle': vid['snippet']['title'], 'channelName': vid['snippet']['channelTitle'], 'thumbnail': vid['snippet']['thumbnails']['high']['url']})
-                        elif vid['id']['kind']=='youtube#video':
-                            temp1.append({'viewed': False, 'type': 'video', 'videoId': vid['id']['videoId'], 'videoTitle': vid['snippet']['title'], 'channelName': vid['snippet']['channelTitle'], 'thumbnail': vid['snippet']['thumbnails']['high']['url']})
-
-                    ele[f"week{index + 1}"]['data'][f"day{i+1}"]['youtube'] = temp1
+                    response = requests.get(url, params=params)
+                    
+                    if response.status_code == 200:
+                        video_data = response.json()
+
+                        temp1 = []
+                        for vid in video_data['items']:
+                            if vid['id']['kind']=='youtube#playlist':
+                                temp1.append({'viewed': False, 'type': 'playlist', 'playlistId': vid['id']['playlistId'], 'videoTitle': vid['snippet']['title'], 'channelName': vid['snippet']['channelTitle'], 'thumbnail': vid['snippet']['thumbnails']['high']['url']})
+                            elif vid['id']['kind']=='youtube#video':
+                                temp1.append({'viewed': False, 'type': 'video', 'videoId': vid['id']['videoId'], 'videoTitle': vid['snippet']['title'], 'channelName': vid['snippet']['channelTitle'], 'thumbnail': vid['snippet']['thumbnails']['high']['url']})
+
+                        ele[f"week{index + 1}"]['data'][f"day{i+1}"]['youtube'] = temp1
+
+                    # Reference Links
+                        temp2 = []
+                        for link in ele[f"week{index + 1}"]['data'][f"day{i+1}"]['links']:
+                            temp2.append({'link': link, 'visited': False})
+
+                        ele[f"week{index + 1}"]['data'][f"day{i+1}"]['links'] = temp2
+                    
+                    else:
+                        print("Error:", response.json())
+                      
+            # Save to MongoDB
+            roadmap_collection.insert_one({'data': roads_data, 'title': position, 'activeDays': [day_data], 'userId': user_id})
 
-                  # Reference Links
-                    temp2 = []
-                    for link in ele[f"week{index + 1}"]['data'][f"day{i+1}"]['links']:
-                        temp2.append({'link': link, 'visited': False})
+            try:
+                results = roadmap_collection.find({"userId": user_id}, {"_id": 1, "title": 1, "activeDays": 1}).limit(5)
+                roadmaps = [{"_id": str(result["_id"]), "title": result["title"], "activeDays": result["activeDays"]} for result in results]
 
-                    ele[f"week{index + 1}"]['data'][f"day{i+1}"]['links'] = temp2
-                
+                if roadmaps:
+                    return json_util.dumps({"data": roadmaps}), 200
                 else:
-                    print("Error:", response.json())
-                      
-            # Save to MongoDB
-            roadmap_collection.insert_one({'data': roads_data, 'activeDays': [day_data], 'user_id': user_id})
+                    return jsonify({"msg": "No Mindmap stored"}), 404
+            except Exception as e:
+                print(e)  
+                return jsonify({"error": "An error occurred"}), 500
 
-            return jsonify({'success': True})
+            # return jsonify({'success': True})
         
         elif data.get('type') == 2:
             # Get roadmap data
@@ -726,10 +792,29 @@ Generate topics for all 7 days in a week and reference links only links(no heade
                 return jsonify({'success': True, 'roadmapData': {'data': roadmap_data['data']}, 'activeDays': roadmap_data['activeDays']})
             else:
                 return jsonify({'success': False})
-            # return jsonify({'success': True, 'roadmapData': roads_data_demo, 'activeDays': active_days})
             
-    
+
+
+@app.route('/roadmap/history', methods=['GET'])
+@auth_user
+def roadmapGetHistory():
+    userId = request.userId
+    try:
+        results = roadmap_collection.find({"userId": userId}, {"_id": 1, "title": 1, "activeDays": 1}).limit(5)
+        roadmaps = [{"_id": str(result["_id"]), "title": result["title"], "activeDays": result["activeDays"]} for result in results]
+
+        if roadmaps:
+            return json_util.dumps({"data": roadmaps}), 200
+        else:
+            return jsonify({"msg": "No Mindmap stored"}), 404
+    except Exception as e:
+        print(e)  
+        return jsonify({"error": "An error occurred"}), 500
+
+
+
 @app.route('/roadmapmodder', methods=['POST'])
+@auth_user
 def roadmapmodder():
     if request.method == 'POST':
         data = request.get_json()
@@ -874,5 +959,6 @@ def roadmapmodder():
         return jsonify({'success': True})
 
 
+
 if __name__ == '__main__':
     app.run(debug=True, host='0.0.0.0')