Update app.py

app.py

@@ -285,7 +285,15 @@ def user_submit_message(msg: str, conversation_state: List[ChatMessage]):
     return "", conversation_state
 
 def respond_wrapper(message: str, conversation_state: List[ChatMessage], max_tokens, temperature, top_p):
-    for updated_messages in stream_gemini_response(message, conversation_state):
+    # 마지막 사용자 메시지 가져오기
+    last_user_message = ""
+    for msg in reversed(conversation_state):
+        if msg.role == "user":
+            last_user_message = msg.content
+            break
+            
+    # 마지막 사용자 메시지로 응답 생성
+    for updated_messages in stream_gemini_response(last_user_message, conversation_state):
         yield "", convert_for_messages_format(updated_messages)
 
 def create_ui():
@@ -439,4 +447,4 @@ if __name__ == "__main__":
         print(f"Error in main: {str(e)}")
         print("Detailed error information:")
         print(traceback.format_exc())
-        raise
+        raise