merged pr #218 with the new streamlit

.gitignore

@@ -20,6 +20,8 @@ accounts.txt
 
 **/__pycache__/
 
+__pycache__/
+
 *.log
 
 cookie.json

gui/query_methods.py

@@ -53,6 +53,8 @@ def query_you(question: str) -> str:
         # Return error message if an exception occurs
         return f'An error occurred: {e}. Please make sure you are using a valid cloudflare clearance token and user agent.'
 
+
+def query(user_input: str, selected_method: str = "Random") -> str:
     # Define a dictionary containing all query methods
     avail_query_methods = {
         "Forefront": query_forefront,
@@ -65,8 +67,6 @@ avail_query_methods = {
         # "Ora": query_ora,
     }
 
-def query(user_input: str, selected_method: str = "Random") -> str:
-
     # If a specific query method is selected (not "Random") and the method is in the dictionary, try to call it
     if selected_method != "Random" and selected_method in avail_query_methods:
         try:
@@ -98,5 +98,3 @@ def query(user_input: str, selected_method: str = "Random") -> str:
 
     return result
 
-
-__all__ = ['query', 'avail_query_methods']

gui/streamlit_chat_app.py

@@ -5,10 +5,11 @@ sys.path.append(os.path.join(os.path.dirname(__file__), os.path.pardir))
 
 import streamlit as st
 from streamlit_chat import message
-from query_methods import query, avail_query_methods
+from query_methods import query
 import pickle
 import openai_rev
 
+
 conversations_file = "conversations.pkl"
 
 def load_conversations():
@@ -71,7 +72,6 @@ if st.sidebar.button("New Conversation"):
     st.session_state['current_conversation'] = {'user_inputs': [], 'generated_responses': []}
     st.session_state['input_field_key'] += 1
 
-print(openai_rev.Provider.__methods__.keys())
 st.session_state['query_method'] = st.sidebar.selectbox(
     "Select API:",
     options=openai_rev.Provider.__members__.keys(),