Update app.py

app.py

@@ -1,6 +1,13 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
+# Load knowledge base from a .txt file
+def load_knowledge_base(file_path):
+    with open(file_path, "r") as file:
+        return file.read()
+
+knowledge_base = load_knowledge_base("manufacturing_info.txt")
+
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
 """
@@ -15,7 +22,15 @@ def respond(
     temperature,
     top_p,
 ):
-    messages = [{"role": "system", "content": system_message}]
+    # Combine system message with knowledge base
+    system_message_with_knowledge = (
+        system_message
+        + "\n\nKnowledge Base:\n"
+        + knowledge_base
+        + "\n\nRespond to the user's queries based on this knowledge."
+    )
+    
+    messages = [{"role": "system", "content": system_message_with_knowledge}]
 
     for val in history:
         if val[0]:
@@ -46,16 +61,13 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
-        gr.Textbox(value=" your smart assistant for Additive Manufacturing!
-I'm here to help you with all your queries about 3D printing, whether you're a beginner exploring the basics or an expert diving into advanced techniques.
-
-Here’s how I can assist you:
-
-Explain key concepts like FDM, SLA, SLS, or DMLS.
-Guide you on materials like PLA, ABS, resins, or metal powders.
-Help troubleshoot common issues like warping, layer shifting, or nozzle clogs.
-Share best practices for optimizing print quality and reducing waste.
-Discuss innovations, trends, and applications in additive manufacturing.", label="System message"),
+        gr.Textbox(
+            value=(
+                "You are AddiBot, an expert chatbot for additive manufacturing. "
+                "Assist users with all their 3D printing needs, using the knowledge base provided."
+            ),
+            label="System message",
+        ),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
         gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
         gr.Slider(
@@ -68,6 +80,5 @@ Discuss innovations, trends, and applications in additive manufacturing.", label
     ],
 )
 
-
 if __name__ == "__main__":
     demo.launch()