Update app.py

app.py

@@ -7,12 +7,12 @@ import pandas as pd
 os.environ['OPENAI_API_KEY'] = 'AIzaSyC8tIrUlJcCRnyksbykluUZVvER6ynHQeI'
 os.environ['SERPAPI_API_KEY'] = '533b5ecf4f322a3e1fb6c6341fc922267805add7e804f7783949a17175ed2f69'
 
-# Define the path for saving CSV files
-HUGGING_FACE_FOLDER = 'https://huggingface.co/spaces/Satyam0077/MultiAgentWorkflowAI/tree/main'  # Update this to your actual Hugging Face directory
-SAVE_DIR = os.path.join(HUGGING_FACE_FOLDER, 'https://huggingface.co/spaces/Satyam0077/MultiAgentWorkflowAI/resolve/main/saved_csv_files')
+# Define the directory name for saving CSV files
+SAVE_DIR = 'saved_csv_files'
 
 # Ensure the save directory exists
-os.makedirs(SAVE_DIR, exist_ok=True)
+if not os.path.exists(SAVE_DIR):
+    os.makedirs(SAVE_DIR)
 
 # Define the functions for your workflow
 def industry_research(company_name, industry_name):
@@ -42,7 +42,7 @@ def save_results_to_csv(results, company_name):
     df = pd.DataFrame(results)
     csv_file_path = os.path.join(SAVE_DIR, f"{company_name}_research_results.csv")
     df.to_csv(csv_file_path, index=False)
-    st.success(f"Results saved to {csv_file_path}")
+    return csv_file_path  # Return the file path for download
 
 def main():
     st.title("Multi-Agent Workflow for AI Use Cases")
@@ -73,8 +73,10 @@ def main():
                 for result in research_results:
                     st.write(f"- [Resource: {result.get('title')}]({result.get('link')})")
                 
-                # Save results to CSV
-                save_results_to_csv(research_results, company_name)
+                # Save results to CSV and provide a download link
+                csv_file_path = save_results_to_csv(research_results, company_name)
+                with open(csv_file_path, 'rb') as file:
+                    st.download_button(label="Download CSV", data=file, file_name=os.path.basename(csv_file_path), mime='text/csv')
 
             else:
                 st.error("Error: " + research_results['error'])
@@ -82,4 +84,4 @@ def main():
             st.warning("Please enter both a company name and an industry name.")
 
 if __name__ == "__main__":
-    main()
+    main()