Update app.py to commit 58d0dedd in GH repo

app.py

@@ -3,6 +3,11 @@ import streamlit_authenticator as stauth
 import sqlite3
 import yaml
 from yaml.loader import SafeLoader
+import os
+
+from free_speech_app.DataLoadDb import *
+from free_speech_app.FreeSpeechPromptsResponses import *
+from langchain.chat_models import ChatOpenAI
 
 # Connect to SQLite database
 conn = sqlite3.connect('user_data.db')
@@ -55,21 +60,26 @@ if authentication_status:
         st.title("Main Screen")
         
         # Input boxes
+        api_input = st.text_input("OpenAI API Token")
         original_post = st.text_input("Paste Original Post Here")
         background_info = st.text_input("Background information on original post (references, relevant information, best practices for responding)")
         
-        # Function to process inputs from box 1 and 2. You will need to replace this with your actual function.
-        def process_inputs(original_post, background_info, principles, writing_style, sources):
-            # Use principles, writing_style, and sources in your function
-            return f"Draft Response: {original_post}, {background_info}, {principles}, {writing_style}, {sources}"
-
-        if user_data is None:
-            draft_response = process_inputs(original_post, background_info, "", "", "")
-        else:
-            draft_response = process_inputs(original_post, background_info, user_data[1], user_data[2], user_data[3])
-        
+        chat_mdl = None
+        draft_response = ''
+
+        # Check if the "Submit" button is clicked
+        if st.button("Submit"):
+            if api_input:
+                os.environ["OPENAI_API_KEY"] = api_input
+                chat_mdl = ChatOpenAI(model_name = 'gpt-3.5-turbo-16k', temperature=0.1)
+    
+            if chat_mdl is not None:
+                if user_data is None:
+                    draft_response = generate_custom_response(original_post, chat_mdl, background_info, "", "", "").content
+                else:
+                    draft_response = generate_custom_response(original_post, chat_mdl, background_info, user_data[1], user_data[2], user_data[3]).content
+            
         # Output from function
-        draft_response = process_inputs(original_post, background_info, user_data[1], user_data[2], user_data[3])
         st.text_area(label="Draft Response. Please edit here or prompt suggestions in the box below.", value=draft_response, height=350)
 
         regenerate_prompt = st.text_input("Additional prompting for regenerating draft response")
@@ -99,4 +109,4 @@ elif authentication_status is None:
         st.error(e)
 
 with open('config.yaml', 'w') as file:
-    yaml.dump(config, file, default_flow_style=False)
+    yaml.dump(config, file, default_flow_style=False)