Update app.py

app.py

@@ -175,41 +175,17 @@ def main():
         # Grading button
         if st.button("Grade Proposal"):
             if new_file:
-                result = grade_essay(input=new_file)
-                
-                st.write("Result from grade_essay:", result)
-                st.write("Type of result:", type(result))
-                
-                if result:
-                    try:
-                        # Case 1: if it's a string, assume it's a JSON string
-                        if isinstance(result, str):
-                            df = pd.read_json(result)
-            
-                        # Case 2: if it's already a dict or list of dicts, use DataFrame directly
-                        elif isinstance(result, (dict, list)):
-                            df = pd.DataFrame(result)
-            
-                        else:
-                            st.error("Unexpected return type from grade_essay.")
-            
-                    except ValueError as e:
-                        st.error(f"Error parsing JSON: {e}")
-                else:
-                    st.error("No data returned from grade_essay.")
-                # final_feedback = grade_essay(input = new_file)
-                # final_feedback = final_feedback.replace("\n", "")
-                # df = pd.read_json(str(grade_essay(input = new_file)))
-                # df = df.T
-                # # Store results in session state
-                # st.session_state.results.append({
-                #     'Feedback': df
-                # })
-
-                # st.write("Feedback:")
-                # st.dataframe(df)
-
-                # st.dataframe(df)
+                new_file = json.loads(new_file)
+                final_feedback = grade_essay(input = new_file)
+                df = pd.read_json(str(grade_essay(input = new_file)))
+                df = df.T
+                # Store results in session state
+                st.session_state.results.append({
+                    'Feedback': df
+                })
+
+                st.write("Feedback:")
+                st.dataframe(df)
     
                 st.success("Hoàn thành chấm điểm.")
             else: