Update app.py

app.py

@@ -1,5 +1,6 @@
 from llama_cpp import Llama
 from huggingface_hub import hf_hub_download
+import gradio as gr
 
 hf_hub_download(repo_id="Dread2Poor/Killamanjaro-8B-Model_Stock-Q4_K_M-GGUF", filename="killamanjaro-8b-model_stock-q4_k_m.gguf", repo_type="model", local_dir=".")
 
@@ -31,6 +32,8 @@ def chat_with_model(user_input):
     """
     Engage in a chat with the model using the provided user input.
     """
+    global history
+
     # Append user input to history
     history.append({"role": "user", "content": user_input})
     
@@ -48,11 +51,27 @@ def chat_with_model(user_input):
     
     return assistant_reply
 
-# Example usage
-if __name__ == "__main__":
-    while True:
-        user_input = input("You: ")
-        if user_input.lower() in ["exit", "quit"]:
-            break
-        reply = chat_with_model(user_input)
-        print("Assistant:", reply)
+def reset_history():
+    """
+    Reset the chat history to the initial state.
+    """
+    global history
+    history = [{"role": "system", "content": system_prompt}]
+    return gr.update(value=""), gr.update(value=[])
+
+# Create Gradio interface
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    with gr.Row():
+        with gr.Column(scale=4):
+            user_input = gr.Textbox(show_label=False, placeholder="Type your message here...")
+        with gr.Column(scale=1):
+            submit_btn = gr.Button("Send")
+    with gr.Row():
+        clear_btn = gr.Button("Clear Chat")
+
+    submit_btn.click(chat_with_model, inputs=user_input, outputs=chatbot)
+    clear_btn.click(reset_history, inputs=[], outputs=[user_input, chatbot])
+
+# Launch the interface
+demo.launch()