update set api keys

app.py

@@ -1237,8 +1237,6 @@ def render_expense_report_summary():
         )
         st.plotly_chart(cost_pie_chart, use_container_width=True)
 
-
-
         st.subheader('Proportion of Total Cost by API Version')
         cost_labels = []
         cost_proportions = []
@@ -1288,31 +1286,6 @@ def sidebar_content():
     except:
         st.header('Expense Report Summary')
         st.write('Available after first run...')
-        
-    # # Check if the expense summary is available in the session state
-    # if 'expense' not in st.session_state or st.session_state.expense is None:
-    #     st.sidebar.write('No expense report data available.')
-    #     return
-    
-    # # Retrieve the expense report summary
-    # expense_summary = st.session_state.expense
-
-    # # Display the expense report summary
-    # st.sidebar.markdown('**Run Count**: ' + str(expense_summary['run_count']))
-
-    # # API version usage percentages
-    # st.sidebar.markdown('**API Version Usage**:')
-    # for version, percentage in expense_summary['api_version_percentages'].items():
-    #     st.sidebar.markdown(f'- {version}: {percentage:.2f}%')
-
-    # # Summary of costs and tokens
-    # st.sidebar.markdown('**Total Cost**: $' + str(round(expense_summary['total_cost_sum'], 4)))
-    # st.sidebar.markdown('**Tokens In**: ' + str(expense_summary['tokens_in_sum']))
-    # st.sidebar.markdown('**Tokens Out**: ' + str(expense_summary['tokens_out_sum']))
-    # # st.sidebar.markdown('**Rate In**: $' + str(round(expense_summary['rate_in_sum'], 2)) + ' per 1000 tokens')
-    # # st.sidebar.markdown('**Rate Out**: $' + str(round(expense_summary['rate_out_sum'], 2)) + ' per 1000 tokens')
-    # st.sidebar.markdown('**Cost In**: $' + str(round(expense_summary['cost_in_sum'], 4)))
-    # st.sidebar.markdown('**Cost Out**: $' + str(round(expense_summary['cost_out_sum'], 4)))
 
 def main():
     with st.sidebar:
@@ -1320,8 +1293,8 @@ def main():
     # Main App
     content_header()
 
-    # tab_settings, tab_prompt, tab_domain, tab_component, tab_processing, tab_private, tab_delete = st.tabs(["Project Settings", "Prompt Builder", "Domain Knowledge","Component Detector", "Processing Options", "API Keys", "Space-Saver"])
-    tab_settings, tab_prompt, tab_domain, tab_component, tab_processing, tab_delete = st.tabs(["Project Settings", "Prompt Builder", "Domain Knowledge","Component Detector", "Processing Options", "Space-Saver"])
+    tab_settings, tab_prompt, tab_domain, tab_component, tab_processing, tab_private, tab_delete = st.tabs(["Project Settings", "Prompt Builder", "Domain Knowledge","Component Detector", "Processing Options", "API Keys", "Space-Saver"])
+    # tab_settings, tab_prompt, tab_domain, tab_component, tab_processing, tab_delete = st.tabs(["Project Settings", "Prompt Builder", "Domain Knowledge","Component Detector", "Processing Options", "Space-Saver"])
 
     with tab_settings:
         content_tab_settings()
@@ -1340,10 +1313,10 @@ def main():
     with tab_processing:
         content_tab_processing()
 
-    # with tab_private:
-    #     if st.button("Edit API Keys"):
-    #         st.session_state.proceed_to_private = True
-    #         st.rerun()
+    with tab_private:
+        if st.button("Edit API Keys"):
+            st.session_state.proceed_to_private = True
+            st.rerun()
 
     with tab_delete:
         create_space_saver()
@@ -1381,7 +1354,7 @@ if 'zip_filepath' not in st.session_state:
 #     create_private_file()
 if st.session_state.proceed_to_build_llm_prompt:
     build_LLM_prompt_config()
-# elif st.session_state.proceed_to_private:
-#     create_private_file()
+elif st.session_state.proceed_to_private:
+    create_private_file()
 elif st.session_state.proceed_to_main:
     main()