Update app.py

app.py

@@ -53,10 +53,12 @@ llm = HuggingFacePipeline.from_model_id(
 chat_model = ChatHuggingFace(llm=llm)
 
 # history of the messages
-messages = []
 
-def llm_run(prompt):
-    global messages
+def clear_memory(messages):
+    messages.clear()
+    return "Memory cleaned."
+    
+def llm_run(prompt, messages):
     print(f"question is {prompt}")
     lang = single_detection(prompt, api_key='4ab77f25578d450f0902fb42c66d5e11')
     if lang == 'en':
@@ -73,7 +75,22 @@ def llm_run(prompt):
 # def greet(prompt, m_type):
 #     return "hi"
 
-demo = gr.Interface(fn=llm_run, inputs=["text"], outputs="text")
+with gr.Blocks(theme=gr.themes.Default(primary_hue=gr.themes.colors.orange, secondary_hue=gr.themes.colors.pink)) as demo:
+    stored_message = gr.State([])
+    with gr.Row():
+        with gr.Column(scale=2):
+            text1 = gr.Textbox(lines=7, label="Prompt", scale=2)
+            with gr.Row():
+                btn1 = gr.Button("Submit", scale=1)
+                btn2 = gr.Button("Clear",  scale=1)
+                btn3 = gr.Button("Clean Memory",  scale=2)
+        with gr.Column(scale=2):
+            out_text = gr.Text(lines=15, label="Output", scale=2)
+    btn1.click(fn=llm_run, inputs=[text1, stored_message], outputs=out_text)
+    btn2.click(lambda: [None, None], outputs=[text1, out_text])
+    btn3.click(fn=clear_memory, inputs=[stored_message], outputs=[out_text])
+    
+# demo = gr.Interface(fn=llm_run, inputs=["text"], outputs="text")
 demo.launch(debug=True, share=True)