Update app.py

app.py

@@ -200,16 +200,17 @@ def main():
   #df = df3.groupby(['Id','Full_sentence','CauseOrEffect', 'Label_level1', 'Label_level2'])['Component'].apply(', '.join).reset_index()
   #st.write(df)
 
-  df = df3
-  df["CauseOrEffect"].replace({"C": "cause", "E": "effect"}, inplace=True)
-  df_final = df[df['CauseOrEffect'] != 'CT'] 
+  #df = df3
+  df3["CauseOrEffect"].replace({"C": "cause", "E": "effect"}, inplace=True)
+  df_final = df3[df3['CauseOrEffect'] != 'CT'] 
   df['New string'] = df_final['Component'].replace(r'[##]+', ' ', regex=True)
 
   df_final = df_final.drop("Component",axis=1)
-  df_final.insert(2, "Component", df['New string'], True)
+  df_final.insert(2, "Component", df3['New string'], True)
 
-  st.write(df_final[df_final['Component'].str.len() != 1])
-  df_final.to_csv('predictions.csv')
+  df_final1 = df_final[df_final['Component'].str.split().str.len().gt(1)]
+    #st.write(df_final[df_final['Component'].str.len() != 1])
+  #df_final.to_csv('predictions.csv')
   
 #   buffer = io.BytesIO()
 #   with pd.ExcelWriter(buffer, engine="xlsxwriter") as writer:
@@ -450,8 +451,8 @@ def main():
   
   buffer = io.BytesIO()
   with pd.ExcelWriter(buffer, engine="xlsxwriter") as writer:
-    df_tab.to_excel(writer,sheet_name="Sheet1",index=False)
-    df_final.to_excel(writer,sheet_name="Sheet2",index=False)
+    df_tab.to_excel(writer,sheet_name="count_result",index=False)
+    df_final1.to_excel(writer,sheet_name="Detailed_results",index=False)
     writer.close()
   df = pd.read_csv('final_data.csv', index_col=0)