Update app.py

app.py

@@ -1,19 +1,17 @@
 import gradio as gr
 import pandas as pd
-from PIL import Image
-import os
-
-# Define the path to the CSV file in the current directory
-csv_file_path = "csv_data/descriptions.csv"
-
-# 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 the DataFrame
-def describe_image(image, desc1, desc2, desc3):
-    global data
-    data = data.append({"Image": image, "Description 1": desc1, "Description 2": desc2, "Description 3": desc3}, ignore_index=True)
-    data.to_csv(csv_file_path, index=False)  # Save the data to a CSV file in the current directory
+def describe_image(image, desc1, desc2, desc3, csv_file_path):
+    # Check if the file exists, and if not, create it
+    if not pd.api.types.is_file_like(csv_file_path):
+        data = pd.DataFrame(columns=["Image", "Description 1", "Description 2", "Description 3"])
+    else:
+        data = pd.read_csv(csv_file_path)
+
+    data = data.append({"Image": image, "Description 1": desc1, "Description 2": desc2, "Description 3": desc3},
+                       ignore_index=True)
+    data.to_csv(csv_file_path, index=False)  # Save the data to the specified CSV file
     return None  # Return None for no specific output
 
 iface = gr.Interface(
@@ -22,7 +20,8 @@ iface = gr.Interface(
         gr.Image(type="pil", label="Upload an Image"),
         "text",
         "text",
-        "text"
+        "text",
+        "text"  # Add an input field for specifying the CSV file path
     ],
     outputs=gr.Info("Data saved successfully"),  # Display the success message
     live=False,
@@ -31,4 +30,4 @@ iface = gr.Interface(
     allow_flagging=False,
 )
 
-iface.launch()
+iface.launch()