Update app.py

app.py

@@ -59,22 +59,41 @@ def chat_with_distilgpt2(input_text, temperature, top_p, top_k):
     # Update the memory with the user input and model response
     memory.save_context({"input": input_text}, {"output": response})
     
-    return response
+    # Format the chat history for display
+    chat_history = conversation_history + f"\nYou: {input_text}\nBot: {response}\n"
+    
+    return chat_history
 
-# Set up the Gradio interface with additional sliders
-interface = gr.Interface(
-    fn=chat_with_distilgpt2,
-    inputs=[
-        gr.Textbox(label="Chat with DistilGPT-2"),  # User input text
-        gr.Slider(0.1, 1.0, step=0.1, value=1.0, label="Temperature"),  # Slider for temperature
-        gr.Slider(0.0, 1.0, step=0.1, value=1.0, label="Top-p"),       # Slider for top-p
-        gr.Slider(1, 100, step=1, value=50, label="Top-k")             # Slider for top-k
-    ],
-    outputs=gr.Textbox(label="DistilGPT-2's Response"),  # Model response
-    title="DistilGPT-2 Chatbot with Memory and Adjustable Parameters",
-    description="This is a simple chatbot powered by the DistilGPT-2 model with conversational memory, using LangChain. You can adjust temperature, top-p, and top-k using the sliders.",
-)
+# Set up the Gradio interface with the input box below the output box
+with gr.Blocks() as interface:
+    chatbot_output = gr.Textbox(label="Conversation", lines=15, placeholder="Chat history will appear here...", interactive=False)
+    
+    # Add the instruction message above the input box
+    gr.Markdown("**Instructions:** Press `Shift + Enter` to submit, and `Enter` for a new line.")
+    
+    # Input box for the user
+    user_input = gr.Textbox(label="Your Input", placeholder="Type your message here...", lines=2, show_label=True)
+    
+    # Sliders for temperature, top_p, and top_k
+    temperature_slider = gr.Slider(0.1, 1.0, step=0.1, value=1.0, label="Temperature")
+    top_p_slider = gr.Slider(0.0, 1.0, step=0.1, value=1.0, label="Top-p")
+    top_k_slider = gr.Slider(1, 100, step=1, value=50, label="Top-k")
+    
+    # Define the function to update the chat
+    def update_chat(input_text, chat_history, temperature, top_p, top_k):
+        updated_history = chat_with_distilgpt2(input_text, temperature, top_p, top_k)
+        return updated_history, ""
+    
+    # Submit when pressing Shift + Enter
+    user_input.submit(update_chat, 
+                      inputs=[user_input, chatbot_output, temperature_slider, top_p_slider, top_k_slider], 
+                      outputs=[chatbot_output, user_input])
 
+    # Layout for sliders and chatbot UI
+    gr.Row([temperature_slider, top_p_slider, top_k_slider])
+    
 # Launch the Gradio app
 interface.launch()
 
+    
+