columns fix

app.py

@@ -2,42 +2,37 @@ import gradio as gr
 import pandas as pd
 
 
-columns_embeddings_col1 = ['Indicator Name']
 
-# columns to use for embeddings on table 2
-columns_embeddings_col2 = ['Indicator name (leonardo)']
-
-def process_tables(table1_path, table2_path, columns_embeddings_col1, columns_embeddings_col2, harmonization):
-    # Assuming process_similarity_results is a function defined in the uploaded functions.py
-    # that we've adapted based on previous discussions.
-    import functions
+def process_tables(table1_path, table2_path, harmonization, columns_embeddings_col1="Indicator Name", columns_embeddings_col2="Indicator name (leonardo)"):
+    import functions  # Assuming functions.py has necessary imports and functions
     
     # Load tables
     table1 = pd.read_csv(table1_path)
     table2 = pd.read_csv(table2_path)
 
-    # You would include here the logic from your Jupyter notebook
-    # This should involve any preprocessing, calling the similarity function, etc.
-    result_final = functions.process_similarity_results(table1, table2, harmonization,columns_embeddings_col1,columns_embeddings_col2)
+    # Process the tables using a function from functions.py
+    result_final = functions.process_similarity_results(table1, table2, harmonization, columns_embeddings_col1, columns_embeddings_col2)
 
-    # Export to CSV
+    # Save the result to a CSV file
     result_path = "/mnt/data/result_final.csv"
     result_final.to_csv(result_path, index=False)
+
     return result_path
 
-# Correct the initialization of Textbox components
+# Define the Gradio interface
 iface = gr.Interface(
     fn=process_tables,
     inputs=[
         gr.File(label="Upload Table 1 (Client Indicators or Framework Table)"),
         gr.File(label="Upload Table 2 (Internal Indicator or Indicator Table)"),
-        gr.Checkbox(label="Harmonization Mode", value=True)
-    ],
-    outputs=[
-        gr.File(label="Download Processed Results")
+        gr.Checkbox(label="Harmonization Mode", value=True),
+        gr.Textbox(label="Columns for Embeddings in Table 1", placeholder="Enter column names separated by commas", default="Indicator Name"),
+        gr.Textbox(label="Columns for Embeddings in Table 2", placeholder="Enter column names separated by commas", default="Indicator name (leonardo)")
     ],
+    outputs="file",
     description="Upload two tables and process them based on the selected parameters. If harmonization, set Table 1 as the Framework Table and Table 2 as the Indicator Table. Otherwise, set Table 1 as the Client Indicators Table and Table 2 as the Internal Indicator Table."
 )
 
 iface.launch()
 
+