Add JSON modif on unparsed files

app.py

@@ -1268,12 +1268,53 @@ def main():
                 st.dataframe(filtered_data)
             except Exception as e:
                 st.error(f"An error occurred while reading the file: {e}")
+            modify_json = st.checkbox('Custom JSON')
+            API_KEY = st.text_input('OpenAI API Key', API_KEY, type='password', help="Enter your OpenAI API key")
+            if not API_KEY:
+                st.warning("Please enter your API key to proceed.")
+            if modify_json:
+                FORMAT_LONG = st.text_area('Custom JSON', FORMAT_LONG, height=500)
+            
+            # If the DataFrame is successfully created, allow the user to select a column
+            col_unparsed = st.selectbox("Select column corresponding to text", data.columns)
+            
+            if st.button("Process Column"):
+                selected_column_data = filtered_data[col_unparsed].tolist()
+                st.write("Column Data:", selected_column_data)
+                st.session_state.result = selected_column_data
+            if 'parsed_responses' not in st.session_state: # Button to trigger parsing of descriptions
+                    with st.status(f"Parsing..", expanded=True) as status:
+                        try:
+                            st.write("Parsing descriptions...")
+                            parser = UAPParser(api_key=API_KEY, model='gpt-3.5-turbo-0125', col=st.session_state.result)
+                            #descriptions = unparsed['description'].tolist()
+                            descriptions = st.session_state.result
+                            format_long = FORMAT_LONG
+                            parser.process_descriptions(descriptions, format_long)
+                            parsed_responses = parser.parse_responses()
+                            try:
+                                responses_df = parser.responses_to_df('sightingDetails', parsed_responses)
+                            except Exception as e:
+                                status.update(label=f"Error parsing: {e}", state="error")
+                                responses_df = parser.responses_to_df(parsed_responses)
+                            st.dataframe(responses_df)
+                            st.session_state['parsed_responses'] = responses_df.copy()
+                            status.update(label="Parsing complete", expanded=False)
+                        except Exception as e:
+                            status.update(label=f"Parsing failed : {e}", state="error")
+            else:
+                # Prompt the user to upload a file if they haven't already
+                st.warning("Please upload a file to proceed.")
 
     # Parsed data
     parsed_tickbox = st.checkbox('Parsed Data')
     if parsed_tickbox:
-        parsed_responses = filter_dataframe(parsed)
-        st.session_state['parsed_responses'] = parsed_responses
+        if responses_df:
+            parsed_responses = filter_dataframe(responses_df)
+            st.session_state['parsed_responses'] = parsed_responses
+        else:
+            parsed_responses = filter_dataframe(parsed)
+            st.session_state['parsed_responses'] = parsed_responses
         col1, col2 = st.columns(2)
         st.dataframe(parsed_responses)
         with col1: