Skip to content

Commit

Permalink
Remove example configuration file and refactor UI utility functions f…
Browse files Browse the repository at this point in the history
…or improved model selection and auto-moderation interface.
  • Loading branch information
sosacrazy126 committed Dec 23, 2024
1 parent 29338dd commit 1b051f6
Show file tree
Hide file tree
Showing 3 changed files with 26 additions and 41 deletions.
13 changes: 0 additions & 13 deletions AutoGroq/configs/config_local.py.example

This file was deleted.

3 changes: 3 additions & 0 deletions AutoGroq/tools/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
"""
This module contains various tools used by the AutoGroq system.
"""
51 changes: 23 additions & 28 deletions AutoGroq/utils/ui_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -755,11 +755,7 @@ def rephrase_prompt(user_request, model, max_tokens=None, llm_provider=None, pro

def select_model():
provider = st.session_state.get('provider', LLM_PROVIDER)

if 'available_models' not in st.session_state or not st.session_state.available_models:
fetch_available_models(provider)

provider_models = st.session_state.available_models
provider_models = get_provider_models(provider)

if not provider_models:
st.warning(f"No models available for {provider}. Please check your API key and connection.")
Expand Down Expand Up @@ -796,7 +792,8 @@ def select_provider():
update_api_url(selected_provider)

# Clear any existing warnings
st.session_state.warning_placeholder.empty()
if 'warning_placeholder' in st.session_state:
st.session_state.warning_placeholder.empty()

# Check for API key and prompt if not found
api_key = get_api_key(selected_provider)
Expand Down Expand Up @@ -846,28 +843,26 @@ def update_temperature(value):


def show_interfaces():
st.markdown('<div class="discussion-whiteboard">', unsafe_allow_html=True)
display_discussion_and_whiteboard()
st.markdown('</div>', unsafe_allow_html=True)

st.markdown('<div class="user-input">', unsafe_allow_html=True)
auto_moderate = st.checkbox("Auto-moderate (slow, eats tokens, but very cool)", key="auto_moderate", on_change=trigger_moderator_agent_if_checked)
if auto_moderate:
with st.spinner("Auto-moderating..."):
moderator_response = trigger_moderator_agent()
if moderator_response:
if st.session_state.next_agent:
st.session_state.user_input = f"To {st.session_state.next_agent}: {moderator_response}"
else:
st.session_state.user_input = moderator_response
st.success("Auto-moderation complete. New input has been generated.")
else:
st.warning("Auto-moderation failed due to rate limiting. Please wait a moment and try again, or proceed manually.")

user_input = st.text_area("Additional Input:", value=st.session_state.user_input, height=200, key="user_input_widget")
reference_url = st.text_input("URL:", key="reference_url_widget")

st.markdown('</div>', unsafe_allow_html=True)
with st.container():
col1, col2 = st.columns([3, 1])
with col1:
display_discussion_and_whiteboard()
with col2:
auto_moderate = st.checkbox("Auto-moderate (slow, eats tokens, but very cool)", key="auto_moderate", on_change=trigger_moderator_agent_if_checked)
if auto_moderate:
with st.spinner("Auto-moderating..."):
moderator_response = trigger_moderator_agent()
if moderator_response:
if st.session_state.next_agent:
st.session_state.user_input = f"To {st.session_state.next_agent}: {moderator_response}"
else:
st.session_state.user_input = moderator_response
st.success("Auto-moderation complete. New input has been generated.")
else:
st.warning("Auto-moderation failed due to rate limiting. Please wait a moment and try again, or proceed manually.")

user_input = st.text_area("Additional Input:", value=st.session_state.user_input, height=200, key="user_input_widget")
reference_url = st.text_input("URL:", key="reference_url_widget")

return user_input, reference_url

Expand Down

0 comments on commit 1b051f6

Please sign in to comment.