Create assistant.py

assistant.py

@@ -0,0 +1,20 @@
+from command_classifier import classify_command
+from executor import execute_command
+
+def get_assistant_response(pipe, command: str, execute: bool) -> str:
+    command_type = classify_command(command)
+    if execute:
+        result = execute_command(command_type, command)
+        if result:
+            return result
+
+    prompt = f"<|user|>\n{command}\n<|assistant|>\n"
+    try:
+        output = pipe(prompt)[0]["generated_text"]
+        if "<|assistant|>" in output:
+            response = output.split("<|assistant|>")[-1].strip()
+        else:
+            response = output.strip()
+        return f"🤖 ({command_type}) {response}"
+    except Exception as e:
+        return f"⚠️ Error: {e}"