fixed f-string bugs

pages/llama_custom_demo.py

@@ -11,7 +11,7 @@ from models.llms import (
     llm_gpt_3_5_turbo,
     llm_gpt_3_5_turbo_0125,
     llm_gpt_4_0125,
-    llm_llama_13b_v2_replicate
+    llm_llama_13b_v2_replicate,
 )
 from models.embeddings import hf_embed_model, openai_embed_model
 from models.llamaCustom import LlamaCustom
@@ -45,6 +45,7 @@ llama_llms = {
     # "meta/llama-2-13b-chat": llm_llama_13b_v2_replicate,
 }
 
+
 def init_session_state():
     if "llama_messages" not in st.session_state:
         st.session_state.llama_messages = [
@@ -60,6 +61,7 @@ def init_session_state():
     if "llama_custom" not in st.session_state:
         st.session_state.llama_custom = None
 
+
 # @st.cache_resource
 def index_docs(
     filename: str,
@@ -73,7 +75,7 @@ def index_docs(
 
             # test the index
             index.as_query_engine().query("What is the capital of France?")
-            
+
         else:
             reader = SimpleDirectoryReader(input_files=[f"{SAVE_DIR}/{filename}"])
             docs = reader.load_data(show_progress=True)
@@ -81,7 +83,9 @@ def index_docs(
                 documents=docs,
                 show_progress=True,
             )
-            index.storage_context.persist(persist_dir=f"vectorStores/{filename.replace(".", '_')}")
+            index.storage_context.persist(
+                persist_dir=f"vectorStores/{filename.replace('.', '_')}"
+            )
 
     except Exception as e:
         print(f"Error: {e}")
@@ -92,6 +96,7 @@ def index_docs(
 def load_llm(model_name: str):
     return llama_llms[model_name]
 
+
 init_session_state()
 
 st.set_page_config(page_title="Llama", page_icon="🦙")
@@ -102,7 +107,9 @@ tab1, tab2 = st.tabs(["Config", "Chat"])
 
 with tab1:
     with st.form(key="llama_form"):
-        selected_llm_name = st.selectbox(label="Select a model:", options=llama_llms.keys())
+        selected_llm_name = st.selectbox(
+            label="Select a model:", options=llama_llms.keys()
+        )
 
         if selected_llm_name.startswith("openai"):
             # ask for the api key
@@ -140,7 +147,12 @@ with tab1:
 with tab2:
     messages_container = st.container(height=300)
     show_previous_messages(framework="llama", messages_container=messages_container)
-    show_chat_input(disabled=False, framework="llama", model=st.session_state.llama_custom, messages_container=messages_container)
+    show_chat_input(
+        disabled=False,
+        framework="llama",
+        model=st.session_state.llama_custom,
+        messages_container=messages_container,
+    )
 
     def clear_history():
         messages_container.empty()
@@ -155,4 +167,3 @@ with tab2:
     if st.button("Clear Chat History"):
         clear_history()
         st.rerun()
-