Update app.py

app.py

@@ -3,13 +3,20 @@ from huggingface_hub import InferenceClient
 from sentence_transformers import SentenceTransformer
 import torch
 
+def read_knowlege_file(knowlege_file) 
+
 # Load knowledge
-with open("knowledge.txt", "r", encoding="utf-8") as file:
-    knowledge = file.read()
+    with open(knowlege_file, "r", encoding="utf-8") as file:
+        knowledge = file.read()
+
+    cleaned_chunks = [chunk.strip() for chunk in knowledge.strip().split("\n") if chunk.strip()]
+    model = SentenceTransformer('all-MiniLM-L6-v2')
+    chunk_embeddings = model.encode(cleaned_chunks, convert_to_tensor=True)
+    return chunk_embeddings
+
+chunk_embeddings = read_knowlege_file("knowlege.txt")
 
-cleaned_chunks = [chunk.strip() for chunk in knowledge.strip().split("\n") if chunk.strip()]
-model = SentenceTransformer('all-MiniLM-L6-v2')
-chunk_embeddings = model.encode(cleaned_chunks, convert_to_tensor=True)
+print(chunk_embeddings)
 
 def get_top_chunks(query):
     query_embedding = model.encode(query, convert_to_tensor=True)
@@ -93,15 +100,16 @@ with gr.Blocks(theme=theme) as chatbot:
             
             
             character_name = gr.Textbox(label = "Character Name", placeholder="Type your name here…", info ="optional")
-            character_class = gr.Dropdown(['Barbarian', 'Bard', 'Cleric', 'Druid', 'Fighter', 'Monk', 'Paladin', 'Ranger', 'Rogue', 'Sorcerer', 'Warlock', 'Wizard'], label="Character Class", info="Choose one or more", multiselect=True)
+            character_class = gr.Dropdown(['Barbarian', 'Bard', 'Cleric', 'Druid', 'Fighter', 'Monk', 'Paladin', 'Ranger', 'Rogue', 'Sorcerer', 'Warlock', 'Wizard'], label="Character Class", multiselect=True)
             character_race = gr.Dropdown([
         "Aarakocra", "Aasimar", "Bugbear", "Centaur", "Changeling", "Dragonborn", "Dwarf", "Elf", 
         "Fairy", "Firbolg", "Genasi", "Githyanki", "Githzerai", "Gnome", "Goblin", "Goliath", 
         "Half-Elf", "Half-Orc", "Halfling", "Hobgoblin", "Human", "Kenku", "Kobold", "Leonin", 
         "Lizardfolk", "Minotaur", "Orc", "Owlin", "Satyr", "Shadar-kai", "Shifter", "Tabaxi", 
         "Tiefling", "Tortle", "Triton", "Warforged", "Yuan-ti"
-    ], multiselect=True, label="Character Race", info="You can choose multiple")
+    ], multiselect=True, label="Character Race")
             character_alignment = gr.Dropdown(["Lawful Good", "Neutral Good", "Chaotic Good", "Lawful Neutral", "True Neutral", "Chaotic Neutral", "Lawful Evil", "Neutral Evil", "Chaotic Evil"], label="Character Alignment", elem_classes="alignment_radio")
+            character_level=gr.Textbox(label = "Character Level")
 
             
         with gr.Column(scale=2):