Update app.py

app.py

@@ -18,6 +18,7 @@ from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
 import chromedriver_autoinstaller
 import os
+import io
 
 # Ensure necessary NLTK resources are downloaded
 nltk.download('punkt')
@@ -166,11 +167,22 @@ def search_wellness_professionals(location):
     query = "therapist OR counselor OR mental health professional"
     radius = 50000
     google_places_data = get_all_places(query, location, radius, api_key)
+    
+    # Check if data is found
     if google_places_data:
         df = pd.DataFrame(google_places_data, columns=["Name", "Address", "Website"])
-        return df.to_csv(index=False)
+        # 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
     else:
-        return "No data found."
+        # 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
 
 # Gradio Interface
 def gradio_interface(message, location, state):
@@ -216,4 +228,3 @@ iface = gr.Interface(
 # Launch Gradio interface
 if __name__ == "__main__":
     iface.launch(debug=True, share=True)  # Set share=True to create a public link
-