Update app.py

app.py

@@ -65,6 +65,9 @@ def question_selected(question):
 @spaces.GPU
 def respond(message, history, system_message, max_tokens, temperature, top_p):
     try:
+        # Initialize chat history if None
+        history = history or []
+        
         model, tokenizer = initialize_model()
         
         # Get context from database
@@ -93,10 +96,14 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
             max_new_tokens=max_tokens
         )[0]['generated_text']
         
-        yield output.strip()
+        # Add the new exchange to history
+        history.append((message, output.strip()))
+        
+        return history
         
     except Exception as e:
-        yield f"An error occurred: {str(e)}"
+        history.append((message, f"An error occurred: {str(e)}"))
+        return history
 
 # Create the Gradio interface
 with gr.Blocks(title="ROS2 Expert Assistant") as demo: