Update main.py

main.py

@@ -240,13 +240,18 @@ def treeDemo():
 - 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).
+- 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.
                                    
 Topic is: ''' + query)
         
         # response.text(8,)
         print(response.text)
-        json_data = response.text
-        modified_json_data = json_data[8:-3]
+        modified_json_data = {}
+        if response.text[0] == '`':
+            json_data = response.text
+            modified_json_data = json_data[8:-3]
+        elif response.text[0] == '{':
+            modified_json_data = json.loads(response.text)
 
         while True:
             new_object_id = ObjectId()
@@ -381,7 +386,6 @@ def interview():
                                         
                                         - 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.''')
                 print(response.text)
-                print(response.text)
                 temp = json.loads(response.text)
                 user_chats[user_id]['qa'] = [{'question': temp['question']}]
                 return jsonify({'success': True, 'data': response.text})