Update home/views.py

home/views.py

@@ -160,6 +160,27 @@ def uploader(request):
         thread.join()
         return redirect('/upload/')
 
+@csrf_exempt
+def uploaderAI(request):
+    global gauth
+    if gauth.access_token_expired:
+        # Refresh them if expired
+        gauth.Refresh()
+    if request.method == 'POST':
+        try:
+            file = request.FILES['file']
+            title = request.POST['title']
+            prompt = request.POST['prompt']
+            if title == '':
+                title = file.name
+            folder_id = "1Hxpc5aT4xl9Nz1QUQJirJcnaT9J1QrzR"
+            file1 = DRIVE.CreateFile({'title': title, 'description':prompt, 'parents': [{'id': folder_id}]})
+            file1.content = file
+            file1.Upload()
+            return HttpResponse(json.dumps({'status': 'success'}), content_type='application/json')
+        except Exception as e:
+            return HttpResponse(json.dumps({'status': 'error','error':str(e)}), content_type='application/json')
+
 
 @csrf_exempt
 @login_required