tikendraw commited on
Commit
5f096f6
·
1 Parent(s): 6b79ba6
Files changed (2) hide show
  1. app/app.py +14 -1
  2. core/utils.py +0 -1
app/app.py CHANGED
@@ -1,14 +1,22 @@
 
1
  import time
2
  import streamlit as st
3
  from core.generation_utils import generate_answer, load_llm
4
  from core.types import ThoughtStepsDisplay, BigMessage
5
  from .app_config import InputConfig, ENV_FILE_PATH, CONFIG_FILE_PATH
6
  import litellm
 
 
7
 
8
-
 
9
 
10
 
11
  def config_sidebar(config:InputConfig) -> InputConfig:
 
 
 
 
12
  st.sidebar.header('Configuration')
13
  model_name = st.sidebar.text_input('Model Name: e.g. provider/model-name',value=config.model_name, placeholder='openai/gpt-3.5-turbo')
14
  model_api_key = st.sidebar.text_input('API Key: ',type='password',value=config.model_api_key, placeholder='sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx')
@@ -32,6 +40,11 @@ def config_sidebar(config:InputConfig) -> InputConfig:
32
  if st.sidebar.button('Save config'):
33
  config.save(env_file=ENV_FILE_PATH, config_file=CONFIG_FILE_PATH)
34
  st.sidebar.success('Config saved!')
 
 
 
 
 
35
 
36
  return config
37
 
 
1
+ from operator import call
2
  import time
3
  import streamlit as st
4
  from core.generation_utils import generate_answer, load_llm
5
  from core.types import ThoughtStepsDisplay, BigMessage
6
  from .app_config import InputConfig, ENV_FILE_PATH, CONFIG_FILE_PATH
7
  import litellm
8
+ import os
9
+ import webbrowser
10
 
11
+ def star_repo():
12
+ webbrowser.open("https://github.com/tikendraw/open-o1")
13
 
14
 
15
  def config_sidebar(config:InputConfig) -> InputConfig:
16
+
17
+ _ = st.button("Star the Repo!", callback=star_repo)
18
+
19
+
20
  st.sidebar.header('Configuration')
21
  model_name = st.sidebar.text_input('Model Name: e.g. provider/model-name',value=config.model_name, placeholder='openai/gpt-3.5-turbo')
22
  model_api_key = st.sidebar.text_input('API Key: ',type='password',value=config.model_api_key, placeholder='sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx')
 
40
  if st.sidebar.button('Save config'):
41
  config.save(env_file=ENV_FILE_PATH, config_file=CONFIG_FILE_PATH)
42
  st.sidebar.success('Config saved!')
43
+
44
+ if st.sidebar.button('Delete Key'):
45
+ os.remove(ENV_FILE_PATH)
46
+ config.model_api_key = ''
47
+ st.sidebar.success('Key deleted!')
48
 
49
  return config
50
 
core/utils.py CHANGED
@@ -3,7 +3,6 @@
3
  import re
4
  import json
5
  import ast
6
- from turtle import color
7
  from pydantic import ValidationError
8
  from termcolor import colored
9
 
 
3
  import re
4
  import json
5
  import ast
 
6
  from pydantic import ValidationError
7
  from termcolor import colored
8