Update app.py

app.py

@@ -1,10 +1,16 @@
-import oneclass
 import gradio as gr
 import pandas as pd
 from io import StringIO
 
-def predict_and_download(positive_csv_file, unlabelled_csv_file, n,Hyperparameter_nu):
-    selected_paper_info = oneclass.select_top_n_papers(n, positive_csv_file, unlabelled_csv_file,Hyperparameter_nu)
+# Import the OneClass class
+from oneclass import OneClass
+
+def predict_and_download(positive_csv_file, unlabelled_csv_file, n, Hyperparameter_nu):
+    # Create an instance of the OneClass class
+    oc = OneClass()
+    
+    # Call the select_top_n_papers method
+    selected_paper_info = oc.select_top_n_papers(n, positive_csv_file, unlabelled_csv_file, Hyperparameter_nu)
     
     # Create a StringIO object to store CSV data
     csv_buffer = StringIO()
@@ -22,10 +28,20 @@ def predict_and_download(positive_csv_file, unlabelled_csv_file, n,Hyperparamete
 # Create the interface
 iface = gr.Interface(
     fn=predict_and_download,
-    inputs=["file", "file", "number","number"],
-    outputs=[gr.DataFrame(label="Selected Papers"), gr.DownloadButton(label="Download CSV")],
+    inputs=[
+        gr.inputs.File(type="csv", label="Positive CSV File"),
+        gr.inputs.File(type="csv", label="Unlabelled CSV File"),
+        gr.inputs.Number(label="Number of Papers to Select", default=10),
+        gr.inputs.Number(label="Hyperparameter nu", default=0.5)
+    ],
+    outputs=[
+        gr.outputs.Dataframe(label="Selected Papers", formats=["csv", "json"]),
+        gr.outputs.DownloadButton(label="Download CSV")
+    ],
     title="Paper Prediction",
-    description="Enter text and upload CSV files for labelled and unlabelled data.",
+    description="Enter the number of papers to select and upload CSV files for labelled and unlabelled data.",
+    article="This interface uses the OneClass algorithm to select the top N papers based on the input CSV files. The Hyperparameter nu controls the sensitivity of the algorithm.",
+    theme="default",
     allow_flagging='never'  # Disable flagging feature
 )