Update app.py

app.py

@@ -86,7 +86,7 @@ capy_proposal = read_pdf(file_path)
 # input = read_pdf(file_path)
 
 # Function to grade the essay using GPT-4
-def grade_essay(str(input)):
+def grade_essay(input):
     # Sample prompt for grading using GPT-4
     template = f"""
 
@@ -175,17 +175,39 @@ def main():
         # Grading button
         if st.button("Grade Proposal"):
             if new_file:
-                final_feedback = grade_essay(input = new_file)
-                final_feedback = final_feedback.replace("\n", "")
-                df = pd.read_json(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)
+                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)