Update main.py

main.py

@@ -11,10 +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, deleteMindmap, getMindmapByid
+from controllers.mindmap import saveMindmap, getMindmap, deleteMindmap, getMindmapByid, saveMindmapById
 import json
+from bson import ObjectId
 from dotenv import load_dotenv
 import os
+import re
+import requests
+
 
 load_dotenv()
 
@@ -33,11 +37,12 @@ restUri = os.getenv('REST_URI')
 
 uri = f'mongodb+srv://{username}:{password}{restUri}'
 
-client = MongoClient(uri, server_api=ServerApi('1'))
+client = MongoClient(uri)
 db = client.GenUpNexus
 users_collection = db["users"]
 interviews_collection = db["interviews"]
 savedMindmap = db["savedMindmap"]
+roadmap_collection = db['roadmaps']
 
 # Send a ping to confirm a successful connection
 try:
@@ -64,6 +69,8 @@ def index():
 def index2():
     return "routes checking..."
 
+
+#generative model routes
 @app.route('/tree', methods=["POST", "GET"])
 def tree():
     if request.method == 'POST':
@@ -157,7 +164,6 @@ def tree():
         return jsonify({'success': True, 'data': response.text})
         # return temp 
 
-
 @app.route('/tree/demo', methods=["POST"])
 def treeDemo():
     if request.method == 'POST':
@@ -166,54 +172,80 @@ def treeDemo():
         print(query)
         response = model.generate_content('''Generate a comprehensive knowledge map representing the user's query, suitable for ReactFlow visualization.
 
-        **Prompt:** {query}
-
-        **Structure:**
-
-        - Top-level node: Represent the user's query.
-        - Sub-nodes branching out based on the query's relevance:
-            - Leverage external knowledge sources (e.g., Wikipedia, knowledge graphs, domain-specific APIs) to identify relevant sub-concepts, related entities, and potential relationships.
-        - Consider including different categories of sub-nodes:
-            - **Concepts:** Core ideas or principles related to the query.
-            - **Subfields:** Specialized areas within the main topic.
-            - **Applications:** Practical uses of the concept or subfield.
-            - **Tools and Technologies:** Software or platforms used to implement the concepts.
-            - **Examples:** Illustrative instances or use cases.
-            - **Historical Context:** Milestones or key figures in the topic's development.
-            - **See Also:** Links to broader concepts or related areas for the further exploration.
-            
-                                                
-        **Content:**
-
-        - Each node should have a label describing the concept, entity, or tool.
-        - Optionally, include brief descriptions, definitions, or key points within the nodes or as tooltips.
-        - Consider using icons to visually represent different categories of nodes (e.g., light bulb for concepts, gear for tools, calendar for historical context, puzzle piece for subfields).
-        - there should be atmax 10 nodes in the knowledge map.
-        - Also follow the n-ary tree structure for better visualization.
-        - Ensure the knowledge map is visually appealing, well-organized, and easy to navigate.
-
-        **Desired Format:**
-
-        - JSON structure compatible with ReactFlow:
-            - nodes (list): id, position, data (label, description, icon(if required), category), type(input, output or custom), style (background, color).
-            - edges (list): id, source, target, label(if required), animated (true or false), style (stroke).
-        - keep the position of nodes spaced out for better visualization.
-        - always keep the top-level node at the center of the visualization.
-        - keep atleast 2 edges "animated":true.
-        - Strictly keep the first node with style having color property with value blue and background property with value #0FFFF0.
-        - Strictly keep the second node with type property value as custom.
-        - You can style the nodes with different colors and edges with different colors. 
-        - to edit edges add style with stroke property and a hexcode value to it.
-                                        
-        Topic is: ''' + query)
+**Prompt:** {query}
+
+**Structure:**
+
+- Top-level node: Represent the user's query.
+- Sub-nodes branching out based on the query's relevance:
+    - Leverage external knowledge sources (e.g., Wikipedia, knowledge graphs, domain-specific APIs) to identify relevant sub-concepts, related entities, and potential relationships.
+- Consider including different categories of sub-nodes:
+    - **Concepts:** Core ideas or principles related to the query.
+    - **Subfields:** Specialized areas within the main topic.
+    - **Applications:** Practical uses of the concept or subfield.
+    - **Tools and Technologies:** Software or platforms used to implement the concepts.
+    - **Examples:** Illustrative instances or use cases.
+    - **Historical Context:** Milestones or key figures in the topic's development.
+    - **See Also:** Links to broader concepts or related areas for the further exploration.
+    
+                                          
+**Content:**
+
+- Each node should have a label describing the concept, entity, or tool.
+- Include brief descriptions, definitions, or key points within the nodes or as tooltips.
+- Consider using icons to visually represent different categories of nodes (e.g.💡 for concepts, ⚙️ for tools, 📅 for historical context, 🧩 for subfields).
+- Also follow the n-ary tree structure for better visualization.
+- There should be atmax 10 nodes.
+- Ensure the knowledge map is visually appealing, well-organized, and easy to navigate.
+
+**Desired Format:**
+
+- JSON structure compatible with ReactFlow:
+    - nodes (list): id, position, type(custom), data(label, description, icon, category).
+    - edges (list): id, source, target, label(if required), animated (true or false), style(stroke).
+- The nodes should not overlap and have enough spacing for readability, therefore adjust it position accordingly.
+- keep atleast 2 edges "animated":true.
+- Strictly keep the all the nodes with type property value as custom. 
+- to edit edges add style with stroke property and a hexcode value to it(Only use this color: #e92a67, #a853ba, #2a8af6, #e92a67).
+                                   
+Topic is: ''' + query)
         
         # response.text(8,)
         print(response.text)
         json_data = response.text
         modified_json_data = json_data[8:-3]
-        return jsonify({'success': True, 'data': modified_json_data})
+
+        while True:
+            new_object_id = ObjectId()
+            if savedMindmap.find_one({'_id': new_object_id}) is None:
+                break 
+
+        return jsonify({'success': True, 'data': modified_json_data, 'objectId': str(new_object_id)})
         # return temp 
 
+@app.route('/mindmap/generate/data', methods=["POST"])
+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)                                 
+        print(response.text)
+        json_data = response.text
+        modified_json_data = json_data[8:-3]
+
+        return jsonify({'success': True, 'data': modified_json_data})                                  
 
 def res(user_id):
     avg_text = 0        
@@ -292,7 +324,7 @@ def interview():
                                                 "code_suggestions": "Any suggestions or optimizations to the code...in string format.",
                                              }
 
-                                             At the end of the interview if no Questions are required then respond in the following JSON format:
+                                             At the end of the interview if no Questions are required then respond in the following format:
                                              {
                                                 "text_correctness": "Test the correctness of text and return a range from 1 to 5 of correctness of text.",
                                                 "text_suggestions": "Some suggestions regarding the text_answer...."
@@ -487,7 +519,7 @@ def signup():
     expires = timedelta(days=7)
     access_token = create_access_token(identity={"email": email, "id": str(result.inserted_id)}, expires_delta=expires)
 
-    res = {"name": name, "email": email}
+    res = {"name": name, "email": email, "userId": str(result.inserted_id)}
     
     return jsonify({"result": res, "token": access_token}), 201
 
@@ -508,7 +540,7 @@ def signin():
     expires = timedelta(days=7)
     access_token = create_access_token(identity={"email": user['email'], "id": str(user['_id'])}, expires_delta=expires)
 
-    res = {"name": user['name'], "email": user['email'], "user_id": str(user['_id'])}
+    res = {"name": user['name'], "email": user['email'], "userId": str(user['_id'])}
 
     return jsonify({"result": res, "token": access_token}), 200
 
@@ -536,6 +568,13 @@ def mindmapSave():
     data = request.json
     return saveMindmap(data, userId, savedMindmap)
 
+@app.route('/mindmap/save/id', methods=['POST'])
+@auth_user
+def mindmapSaveId():
+    userId = request.userId
+    data = request.json
+    return saveMindmapById(data, userId, savedMindmap)
+
 @app.route('/mindmap/get', methods=['GET'])
 @auth_user
 def mindmapGet():
@@ -543,10 +582,8 @@ def mindmapGet():
     return getMindmap(userId, savedMindmap)
 
 @app.route('/mindmap/get/<id>', methods=['GET'])
-@auth_user
 def mindmapGetById(id):
-    userId = request.userId
-    return getMindmapByid(userId, id, savedMindmap)
+    return getMindmapByid(id, savedMindmap)
 
 @app.route('/mindmap/delete', methods=['POST'])
 @auth_user
@@ -563,5 +600,267 @@ def mindmapDemo():
     return get_initial_data(), 200
 
 
+
+@app.route('/roadmap', methods=['POST'])
+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:
+            # 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
+            
+            # Youtube Integration & links modification
+            print(response)
+            print(response.text)
+            roads_data = json.loads(response.text)
+
+            print(roads_data)
+            return jsonify({'success': True})
+            
+            API_KEY = os.environ('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
+                    }
+                
+                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
+
+                  # 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, 'activeDays': [day_data], 'user_id': user_id})
+
+            return jsonify({'success': True})
+        
+        elif data.get('type') == 2:
+            # 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']})
+            else:
+                return jsonify({'success': False})
+            # return jsonify({'success': True, 'roadmapData': roads_data_demo, 'activeDays': active_days})
+            
+    
+
+@app.route('/roadmapmodder', methods=['POST'])
+def roadmapmodder():
+    if request.method == 'POST':
+        data = request.get_json()
+        user_id = data.get('user_id')
+        obj_id = data.get('_id')
+        day_data = data.get('todayData')
+        week_num = data.get('weekNum')
+        day_num = data.get('dayNum')
+        week_field = f'week{week_num}'
+        day_field = f'day{day_num}'
+
+        if data.get('type') == 11:                  # Add Youtube Video
+            video_url = data.get('videoUrl')
+
+            video_id_match = re.search(r"(?:https?:\/\/)?(?:www\.)?(?:youtube\.com\/(?:[^\/\n\s]+\/\S+\/|(?:v|e(?:mbed)?)\/|\S*?[?&]v=)|youtu\.be\/)([a-zA-Z0-9_-]{11})", video_url)
+            video_id = video_id_match.group(1)
+            api_key = os.getenv('YOUTUBE_API_KEY')
+
+            api_url = f"https://www.googleapis.com/youtube/v3/videos?id={video_id}&key={api_key}&part=snippet"
+            response = requests.get(api_url)
+            temp = {}
+            if response.status_code == 200:
+                data = response.json()
+            
+            if 'items' in data and data['items']:
+                item = data['items'][0]
+                print(item)
+
+                if item['kind'] == 'youtube#video':
+                    temp = {
+                        "channelName": item['snippet']['channelTitle'],
+                        "thumbnail": item['snippet']['thumbnails']['high']['url'],
+                        "type": 'video',
+                        "videoId": item['id'],
+                        "videoTitle": item['snippet']['localized']['title'],
+                        "viewed": True
+                    }
+                elif item['kind'] == 'youtube#playlist':
+                    temp = {
+                        "channelName": item['snippet']['channelTitle'],
+                        "thumbnail": item['snippet']['thumbnails']['high']['url'],
+                        "type": 'playlist',
+                        "playlistId": item['id'],
+                        "videoTitle": item['snippet']['localized']['title'],
+                        "viewed": True
+                    }
+
+            else:
+                print("Error occurred:", response.status_code)
+            
+            temp_data = roadmap_collection.find_one({"_id": ObjectId(obj_id)})
+            temp_data['data'][week_num-1][week_field]['data'][day_field]['youtube'].append(temp)
+
+            roadmap_collection.update_one(
+                {"_id": ObjectId(obj_id)},
+                {"$set": { "data": temp_data['data'] }}
+            )
+            
+            if temp_data['activeDays'][len(temp_data['activeDays'])-1]['day'] != day_data['day'] or temp_data['activeDays'][len(temp_data['activeDays'])-1]['month'] != day_data['month'] or temp_data['activeDays'][len(temp_data['activeDays'])-1]['year'] != day_data['year']:
+                updateActiveStreak(obj_id, day_data)
+            
+            return jsonify({'success': True, 'youtubeUpdate': temp})
+            
+        elif data.get('type') == 12:                # Update Youtube Video View
+            video_index = data.get('videoIndex')
+
+            temp_data = roadmap_collection.find_one({"_id": ObjectId(obj_id)})
+            temp_data['data'][week_num-1][week_field]['data'][day_field]['youtube'][video_index]['viewed'] = True
+
+            roadmap_collection.update_one(
+                {"_id": ObjectId(obj_id)},
+                {"$set": { "data": temp_data['data'] }}
+            )
+            
+            if temp_data['activeDays'][len(temp_data['active_days'])-1]['day'] != day_data['day'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['month'] != day_data['month'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['year'] != day_data['year']:
+                updateActiveStreak(obj_id, day_data)
+
+            return jsonify({'success': True})
+
+        elif data.get('type') == 21:                # Add Reference Link
+            ref_link = data.get('refLink')
+
+            temp_data = roadmap_collection.find_one({"_id": ObjectId(obj_id)})
+            temp_data['data'][week_num-1][week_field]['data'][day_field]['links'].append({"link": ref_link, "visited": True})
+
+            roadmap_collection.update_one(
+                {"_id": ObjectId(obj_id)},
+                {"$set": { "data": temp_data['data'] }}
+            )
+            
+            if temp_data['activeDays'][len(temp_data['active_days'])-1]['day'] != day_data['day'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['month'] != day_data['month'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['year'] != day_data['year']:
+                updateActiveStreak(obj_id, day_data)
+
+        elif data.get('type') == 22:                # Update Reference link Visit
+            ref_link_index = data.get('refLinkIndex')
+
+            temp_data = roadmap_collection.find_one({"_id": ObjectId(obj_id)})
+            temp_data['data'][week_num-1][week_field]['data'][day_field]['links'][ref_link_index]["visited"] = True
+
+            roadmap_collection.update_one(
+                {"_id": ObjectId(obj_id)},
+                {"$set": { "data": temp_data['data'] }}
+            )
+
+            if temp_data['activeDays'][len(temp_data['active_days'])-1]['day'] != day_data['day'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['month'] != day_data['month'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['year'] != day_data['year']:
+                updateActiveStreak(obj_id, day_data)
+
+        # Add Mindmaps change in mindap generation fucntion such that if roadmapId, WeekNum, DayNum is sent while generation of mindmap then associate it directly with the roadmap.
+
+        elif data.get('type') == 31:                # Update isComplete
+            temp_data = roadmap_collection.find_one({"_id": ObjectId(obj_id)})
+            temp_data['data'][week_num-1][week_field]['data'][day_field]['isComplete'] = True
+
+            roadmap_collection.update_one(
+                {"_id": ObjectId(obj_id)},
+                {"$set": { "data": temp_data['data'] }}
+            )
+            
+            if temp_data['activeDays'][len(temp_data['active_days'])-1]['day'] != day_data['day'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['month'] != day_data['month'] or temp_data['activeDays'][len(temp_data['active_days'])-1]['year'] != day_data['year']:
+                updateActiveStreak(obj_id, day_data)
+
+        return jsonify({'success': True})
+    
+
+def updateActiveStreak(object_id, day_data):       # Update Active Streak
+    temp_data = roadmap_collection.find_one({"_id": ObjectId(object_id)})
+    temp_data['activeDays'].append(day_data)
+
+    roadmap_collection.update_one(
+        {"_id": ObjectId(object_id)},
+        {"$set": { "activeDays": temp_data['activeDays'] }}
+    )
+
+
+
 if __name__ == '__main__':
-    app.run(debug=True)
+    app.run(debug=True, host='0.0.0.0')