Update app.py

app.py

@@ -3,6 +3,9 @@ from huggingface_hub import InferenceClient
 import os
 import pandas as pd
 from mitreattack.stix20 import MitreAttackData
+from format_input import detect_log_type
+from descriptions import descriptions
+from testing_input import log_input
 
 # Chemins des fichiers JSON
 ics_attack_path = 'ics-attack.json'
@@ -59,7 +62,7 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
-        gr.Textbox(value=f"""<s>[INST] Given these TTPs: {techniques_str}\n\nfigure out which technique is used in these logs and respond in bullets points and nothing else if there is no TTPs detected just say that there is not TTP detected: [/INST]""", label="System message"),
+        gr.Textbox(value=f"""<s>[INST] Given these TTPs: {techniques_str}\n\n and here are {descriptions}\n\nfigure out which technique is used in these logs and respond in bullets points and nothing else :\n{log_line}\n[/INST]""", label="System message"),
             gr.Slider(minimum=1, maximum=2048, value=1024, step=1, label="Max new tokens"),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.1, step=0.1, label="Temperature"),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),