Update app.py

app.py

@@ -1,35 +1,19 @@
-import os
 import gradio as gr
 import pandas as pd
-from pydrive.auth import GoogleAuth
-from pydrive.drive import GoogleDrive
+from PIL import Image
+import os
 
-# Load credentials from environment variables
-gauth = GoogleAuth()
-gauth.LocalWebserverAuth()
+# Define the path to the CSV file in the current directory
+csv_file_path = "descriptions.csv"
 
-# Initialize Google Drive
-drive = GoogleDrive(gauth)
+# Initialize an empty DataFrame to store the data
+data = pd.DataFrame(columns=["Image", "Description 1", "Description 2", "Description 3"])
 
-# Function to handle the user's input and store it in a Google Drive file
+# Function to handle the user's input and store it in the DataFrame
 def describe_image(image, desc1, desc2, desc3):
-    data = pd.DataFrame(columns=["Image", "Description 1", "Description 2", "Description 3"])
+    global data
     data = data.append({"Image": image, "Description 1": desc1, "Description 2": desc2, "Description 3": desc3}, ignore_index=True)
-
-    # Create or get a reference to the file on Google Drive
-    file_list = drive.ListFile({'q': "'" + os.environ.get("GDRIVE_FOLDER_ID") + "' in parents and trashed=false"}).GetList()
-    for file in file_list:
-        if file['title'] == os.environ.get("CSV_FILENAME"):
-            gdrive_file = file
-            break
-    else:
-        gdrive_file = drive.CreateFile({'title': os.environ.get("CSV_FILENAME"), "parents": [{"kind": "drive#fileLink", "id": os.environ.get("GDRIVE_FOLDER_ID")}]})
-
-    # Write the data to the file and save it on Google Drive
-    data.to_csv('temp.csv', index=False)
-    gdrive_file.SetContentFile('temp.csv')
-    gdrive_file.Upload()
-
+    data.to_csv(csv_file_path, index=False)  # Save the data to a CSV file in the current directory
     return None  # Return None for no specific output
 
 iface = gr.Interface(
@@ -40,7 +24,7 @@ iface = gr.Interface(
         "text",
         "text"
     ],
-    outputs=gr.Info("Data saved successfully on Google Drive"),  # Display the success message
+    outputs=gr.Info("Data saved successfully"),  # Display the success message
     live=False,
     title="Image Description App",
     description="Upload an image and provide three descriptions for it.",
@@ -51,4 +35,4 @@ iface = gr.Interface(
     )
 )
 
-iface.launch()
+iface.launch(share=True)