update app

README.md

@@ -5,6 +5,6 @@ colorFrom: red
 colorTo: indigo
 sdk: gradio
 sdk_version: 4.42.0
-app_file: app_temp.py
+app_file: app_drive.py
 pinned: false
 ---

driveapi/drive.py

@@ -16,12 +16,16 @@ service = build('drive', 'v3', credentials=credentials)
 
 logs_id = os.environ.get('LOGS_ID')
 
-# Save Logs
 def upload_chat_to_drive(chat_history, file_name):
     # Convert chat history to CSV
     csv_output = io.StringIO()
-    writer = csv.writer(csv_output)
-    writer.writerows(chat_history)
+    writer = csv.writer(csv_output, quoting=csv.QUOTE_ALL)
+    
+    for row in chat_history:
+        # Ensure all items in the row are strings and handle potential encoding issues
+        encoded_row = [str(item).encode('utf-8', errors='replace').decode('utf-8') for item in row]
+        writer.writerow(encoded_row)
+    
     csv_output.seek(0)
 
     # File metadata
@@ -31,11 +35,16 @@ def upload_chat_to_drive(chat_history, file_name):
         'parents': [logs_id]
     }
 
-    # Upload file
-    media = MediaIoBaseUpload(csv_output, mimetype='text/csv')
-    file = service.files().create(body=file_metadata, media_body=media, fields='id').execute()
-
-
+    try:
+        # Upload file
+        media = MediaIoBaseUpload(csv_output, mimetype='text/csv', resumable=True)
+        file = service.files().create(body=file_metadata, media_body=media, fields='id').execute()
+        print(f'File ID: {file.get("id")}')
+        return file.get('id')
+    except Exception as e:
+        print(f"An error occurred while uploading to Google Drive: {e}")
+        return None
+    
 ## Read PDF files
 def download_file(file_id):
     service = build('drive', 'v3', credentials=credentials)

requirements.txt

@@ -1,5 +1,5 @@
 faiss-gpu==1.7.2
-gradio==4.20.0
+gradio==4.43.0
 pycryptodome==3.20.0
 tiktoken==0.5.1
 PyPDF2==3.0.1