Update app.py

app.py

@@ -87,16 +87,16 @@ def inference(prompt):
 #     return result
 
      
-# # def remember(prompt, result):
-#     global memory
-#     # Store the session as a dictionary
-#     session = {'prompt': prompt, 'result': result}
-#     memory.append(session)
-
-#     # Update previous_sessions for display
-#     session_display = [f"Q: {s['prompt']} \nA: {s['result']}" for s in memory]
+# def remember(prompt, result):
+    global memory
+    # Store the session as a dictionary
+    session = {'prompt': prompt, 'result': result}
+    memory.append(session)
+
+    # Update previous_sessions for display
+    session_display = [f"Q: {s['prompt']} \nA: {s['result']}" for s in memory]
     
-#     return "\n\n".join(session_display)  # Return formatted sessions as a string
+    return "\n\n".join(session_display)  # Return formatted sessions as a string
 
 
 
@@ -113,14 +113,14 @@ with gr.Blocks(theme='gradio/soft', css=custom_css) as demo:
     gr.Markdown("<center><h1>HydroFlow LLM Demo</h1></center>")
     
 
-    # with gr.Row():
+    with gr.Row():
         # with gr.Column(scale=1):
         #     gr.Markdown("## Previous Sessions")
         #     #session_list = gr.Textbox(label="Sessions", value="\n".join(previous_sessions), interactive=False, lines=4, max_lines=20)
         #     add_button = gr.Button("New Session")
         #     clear_session = gr.Button("Clear Session")
 
-        with gr.Column(scale=2):
+        with gr.Column(scale=1):
             output = gr.Textbox(label="Result", lines=5, max_lines=20)
             prompt = gr.Textbox(label="Enter your Prompt here", max_lines=20)