Update app.py

app.py

@@ -5,9 +5,24 @@ import io
 
 def predict_and_download(positive_csv_file, unlabelled_csv_file, n, text):
     selected_paper_info = oneclass.select_top_n_papers(n, positive_csv_file, unlabelled_csv_file)
-    csv_string = selected_paper_info.to_csv(index=False)
-    csv_filename = "selected_papers.csv"
-    return selected_paper_info, csv_filename
+    
+    # Create a StringIO object to store CSV data
+    csv_buffer = StringIO()
+    
+    # Write DataFrame to the StringIO buffer as CSV
+    selected_paper_info.to_csv(csv_buffer, index=False)
+    
+    # Get the CSV data from the buffer
+    csv_content = csv_buffer.getvalue()
+    
+    # Close the buffer
+    csv_buffer.close()
+    
+    # Save CSV content to a variable
+    csv_variable = csv_content
+    
+    # Return selected_paper_info and CSV content
+    return selected_paper_info, csv_variable
 
 # Create the interface
 iface = gr.Interface(