Update app.py

app.py

@@ -18,7 +18,7 @@ from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
 import chromedriver_autoinstaller
 import os
-import io
+import tempfile
 
 # Ensure necessary NLTK resources are downloaded
 nltk.download('punkt')
@@ -171,18 +171,23 @@ def search_wellness_professionals(location):
     # Check if data is found
     if google_places_data:
         df = pd.DataFrame(google_places_data, columns=["Name", "Address", "Website"])
-        # Create a CSV in memory
-        csv_file = io.StringIO()
-        df.to_csv(csv_file, index=False)
-        csv_file.seek(0)  # Reset the pointer to the beginning of the file
-        return csv_file  # Return the file-like object
+        
+        # Create a temporary file to store the CSV
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".csv")
+        df.to_csv(temp_file, index=False)
+        temp_file.close()  # Close the file so that Gradio can download it
+        
+        return temp_file.name  # Return the path to the temporary file
     else:
         # If no data found, return a dummy CSV with a message
         dummy_df = pd.DataFrame([["No data found.", "", ""]], columns=["Name", "Address", "Website"])
-        csv_file = io.StringIO()
-        dummy_df.to_csv(csv_file, index=False)
-        csv_file.seek(0)
-        return csv_file  # Return the dummy file
+        
+        # Create a temporary file for the dummy data
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".csv")
+        dummy_df.to_csv(temp_file, index=False)
+        temp_file.close()  # Close the file
+        
+        return temp_file.name  # Return the path to the dummy file
 
 # Gradio Interface
 def gradio_interface(message, location, state):